hirochachacha / go-smb2

SMB2/3 client library written in Go.
BSD 2-Clause "Simplified" License
350 stars 94 forks source link
go golang smb smb2

smb2

Build Status Go Reference

Description

SMB2/3 client implementation.

Installation

go get github.com/hirochachacha/go-smb2

Documentation

http://godoc.org/github.com/hirochachacha/go-smb2

Examples

List share names

package main

import (
    "fmt"
    "net"

    "github.com/hirochachacha/go-smb2"
)

func main() {
    conn, err := net.Dial("tcp", "SERVERNAME:445")
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    d := &smb2.Dialer{
        Initiator: &smb2.NTLMInitiator{
            User:     "USERNAME",
            Password: "PASSWORD",
        },
    }

    s, err := d.Dial(conn)
    if err != nil {
        panic(err)
    }
    defer s.Logoff()

    names, err := s.ListSharenames()
    if err != nil {
        panic(err)
    }

    for _, name := range names {
        fmt.Println(name)
    }
}

File manipulation

package main

import (
    "io"
    "io/ioutil"
    "net"

    "github.com/hirochachacha/go-smb2"
)

func main() {
    conn, err := net.Dial("tcp", "SERVERNAME:445")
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    d := &smb2.Dialer{
        Initiator: &smb2.NTLMInitiator{
            User:     "USERNAME",
            Password: "PASSWORD",
        },
    }

    s, err := d.Dial(conn)
    if err != nil {
        panic(err)
    }
    defer s.Logoff()

    fs, err := s.Mount("SHARENAME")
    if err != nil {
        panic(err)
    }
    defer fs.Umount()

    f, err := fs.Create("hello.txt")
    if err != nil {
        panic(err)
    }
    defer fs.Remove("hello.txt")
    defer f.Close()

    _, err = f.Write([]byte("Hello world!"))
    if err != nil {
        panic(err)
    }

    _, err = f.Seek(0, io.SeekStart)
    if err != nil {
        panic(err)
    }

    bs, err := ioutil.ReadAll(f)
    if err != nil {
        panic(err)
    }

    fmt.Println(string(bs))
}

Check error types

package main

import (
    "context"
    "fmt"
    "net"
    "os"

    "github.com/hirochachacha/go-smb2"
)

func main() {
    conn, err := net.Dial("tcp", "SERVERNAME:445")
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    d := &smb2.Dialer{
        Initiator: &smb2.NTLMInitiator{
            User:     "USERNAME",
            Password: "PASSWORD",
        },
    }

    s, err := d.Dial(conn)
    if err != nil {
        panic(err)
    }
    defer s.Logoff()

    fs, err := s.Mount("SHARENAME")
    if err != nil {
        panic(err)
    }
    defer fs.Umount()

    _, err = fs.Open("notExist.txt")

    fmt.Println(os.IsNotExist(err)) // true
    fmt.Println(os.IsExist(err))    // false

    fs.WriteFile("hello2.txt", []byte("test"), 0444)
    err = fs.WriteFile("hello2.txt", []byte("test2"), 0444)
    fmt.Println(os.IsPermission(err)) // true

    ctx, cancel := context.WithTimeout(context.Background(), 0)
    defer cancel()

    _, err = fs.WithContext(ctx).Open("hello.txt")

    fmt.Println(os.IsTimeout(err)) // true
}

Glob and WalkDir through FS interface

package main

import (
    "fmt"
    "net"
    iofs "io/fs"

    "github.com/hirochachacha/go-smb2"
)

func main() {
    conn, err := net.Dial("tcp", "SERVERNAME:445")
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    d := &smb2.Dialer{
        Initiator: &smb2.NTLMInitiator{
            User:     "USERNAME",
            Password: "PASSWORD",
        },
    }

    s, err := d.Dial(conn)
    if err != nil {
        panic(err)
    }
    defer s.Logoff()

    fs, err := s.Mount("SHARENAME")
    if err != nil {
        panic(err)
    }
    defer fs.Umount()

    matches, err := iofs.Glob(fs.DirFS("."), "*")
    if err != nil {
        panic(err)
    }
    for _, match := range matches {
        fmt.Println(match)
    }

    err = iofs.WalkDir(fs.DirFS("."), ".", func(path string, d iofs.DirEntry, err error) error {
        fmt.Println(path, d, err)

        return nil
    })
    if err != nil {
        panic(err)
    }
}