Golang g9pc.Client类代码示例

func lsone(c *g9pc.Client, s string, long bool) {
	st, err := c.FStat(normpath(s))
	if err != nil {
		fmt.Fprintf(os.Stderr, "error stat: %s\n", err)
		return
	}
	if st.Mode&g9p.DMDIR != 0 {
		file, err := c.FOpen(s, g9p.OREAD)
		if err != nil {
			fmt.Fprintf(os.Stderr, "error opening dir: %s\n", err)
			return
		}
		defer file.Close()
		for {
			d, err := file.Readdir(0)
			if err != nil {
				fmt.Fprintf(os.Stderr, "error reading dir: %s\n", err)
			}
			if d == nil || len(d) == 0 {
				break
			}
			for _, dir := range d {
				if long {
					fmt.Fprintf(os.Stdout, "%s\n", dirtostr(dir))
				} else {
					os.Stdout.WriteString(dir.Name + "\n")
				}
			}
		}
	} else {
		fmt.Fprintf(os.Stdout, "%s\n", dirtostr(st))
	}
}

func interactive(c *g9pc.Client) {
	reader, ok := bufio.NewReaderSize(os.Stdin, 8192)
	if ok != nil {
		fmt.Fprintf(os.Stderr, "can't create reader buffer: %s\n", ok)
	}
	done := make(chan error)
	go func() {
		e := c.Wait()
		fmt.Printf("wait finished (%v)\n", e)
		done <- e
	}()
loop:
	for {
		select {
		case e := <-done:
			fmt.Fprintf(os.Stderr, "server: %v\n", e)
			break loop
		default:
		}
		fmt.Print(*prompt)
		line, ok := reader.ReadSlice('\n')
		if ok != nil {
			fmt.Fprintf(os.Stderr, "exiting...\n")
			break
		}
		str := strings.TrimSpace(string(line))
		// TODO: handle larger input lines by doubling buffer
		in := strings.Split(str, "\n")
		for i := range in {
			if len(in[i]) > 0 {
				cmd(c, in[i])
			}
		}
	}
}

// Remove f from remote server
func rmone(c *g9pc.Client, f string) {
	fname := normpath(f)

	err := c.FRemove(fname)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error in stat %s: %s\n", fname, err)
		return
	}
}

// Create a single directory on remote server
func mkone(c *g9pc.Client, s string) {
	fname := normpath(s)
	file, err := c.FCreate(fname, 0777|g9p.DMDIR, g9p.OWRITE)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error creating directory %s: %s\n", fname, err)
		return
	}
	file.Close()
}

func walkone(c *g9pc.Client, s string) {
	ncwd := normpath(s)
	_, err := c.FWalk(ncwd)
	if err != nil {
		fmt.Fprintf(os.Stderr, "walk error: %s\n", err)
		return
	}
	cwd = ncwd
}

// Stat the remote file f
func statone(c *g9pc.Client, f string) {
	fname := normpath(f)

	stat, err := c.FStat(fname)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error in stat %s: %s\n", fname, err)
		return
	}
	fmt.Fprintf(os.Stdout, "%s\n", stat)
}

// Copy a local file to remote server
func cmdput(c *g9pc.Client, s []string) {
	var from, to string
	switch len(s) {
	case 1:
		_, to = path.Split(s[0])
		to = normpath(to)
		from = s[0]
	case 2:
		from, to = s[0], normpath(s[1])
	default:
		fmt.Fprintf(os.Stderr, "incorrect arguments; usage: put local [remote]\n")
	}

	fromfile, err := os.Open(from)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error opening %s for reading: %s\n", from, err)
		return
	}
	defer fromfile.Close()

	file, ferr := c.FOpen(to, g9p.OWRITE|g9p.OTRUNC)
	if ferr != nil {
		file, ferr = c.FCreate(to, 0666, g9p.OWRITE)
		if ferr != nil {
			fmt.Fprintf(os.Stderr, "error opening %s for writing: %s\n", to, err)
			return
		}
	}
	defer file.Close()

	buf := make([]byte, 8192)
	for {
		n, oserr := fromfile.Read(buf)
		if oserr != nil && oserr != io.EOF {
			fmt.Fprintf(os.Stderr, "error reading %s: %s\n", from, oserr)
			return
		}

		if n == 0 {
			break
		}

		m, err := file.Write(buf[0:n])
		if err != nil {
			fmt.Fprintf(os.Stderr, "error writing %s: %s\n", to, err)
			return
		}

		if m != n {
			fmt.Fprintf(os.Stderr, "short write %s\n", to)
			return
		}
	}
}

// Copy a remote file to local filesystem
func cmdget(c *g9pc.Client, s []string) {
	var from, to string
	switch len(s) {
	case 1:
		from = normpath(s[0])
		_, to = path.Split(s[0])
	case 2:
		from, to = normpath(s[0]), s[1]
	default:
		fmt.Fprintf(os.Stderr, "from arguments; usage: get from to\n")
	}

	tofile, err := os.Create(to)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error opening %s for writing: %s\n", to, err)
		return
	}
	defer tofile.Close()

	file, ferr := c.FOpen(from, g9p.OREAD)
	if ferr != nil {
		fmt.Fprintf(os.Stderr, "error opening %s for writing: %s\n", to, err)
		return
	}
	defer file.Close()

	buf := make([]byte, 8192)
	for {
		n, oserr := file.Read(buf)
		if oserr != nil {
			fmt.Fprintf(os.Stderr, "error reading %s: %s\n", from, oserr)
			return
		}
		if n == 0 {
			break
		}

		m, err := tofile.Write(buf[0:n])
		if err != nil {
			fmt.Fprintf(os.Stderr, "error writing %s: %s\n", to, err)
			return
		}

		if m != n {
			fmt.Fprintf(os.Stderr, "short write %s\n", to)
			return
		}
	}
}

func main() {
	var user g9p.User
	var err error
	var c *g9pc.Client
	var file *g9pc.File
	var d []*g9p.Dir

	flag.Parse()
	user = g9p.OsUsers.Uid2User(os.Geteuid())
	c, err = g9pc.Mount("tcp", *addr, "", user, nil)
	if err != nil {
		goto error
	}

	if flag.NArg() != 1 {
		log.Println("invalid arguments")
		return
	}

	file, err = c.FOpen(flag.Arg(0), g9p.OREAD)
	if err != nil {
		goto error
	}

	for {
		d, err = file.Readdir(0)
		if err != nil {
			goto error
		}

		if d == nil || len(d) == 0 {
			break
		}

		for i := 0; i < len(d); i++ {
			os.Stdout.WriteString(d[i].Name + "\n")
		}
	}

	file.Close()
	return

error:
	log.Println(fmt.Sprintf("Error: %v", err))
}

// Print the contents of f
func cmdcat(c *g9pc.Client, s []string) {
	buf := make([]byte, 8192)
	for _, f := range s {
		fname := normpath(f)
		file, err := c.FOpen(fname, g9p.OREAD)
		if err != nil {
			fmt.Fprintf(os.Stderr, "error opening %s: %s\n", f, err)
			continue
		}
		defer file.Close()
		for {
			n, err := file.Read(buf)
			if err != nil {
				fmt.Fprintf(os.Stderr, "error reading %s: %s\n", f, err)
			}
			if n == 0 {
				break
			}
			os.Stdout.Write(buf[0:n])
		}
	}
}

// Write the string s to remote file f. Create f if it doesn't exist
func writeone(c *g9pc.Client, f, s string) {
	fname := normpath(f)
	file, err := c.FOpen(fname, g9p.OWRITE|g9p.OTRUNC)
	if err != nil {
		file, err = c.FCreate(fname, 0666, g9p.OWRITE)
		if err != nil {
			fmt.Fprintf(os.Stderr, "error opening %s: %v\n", fname, err)
			return
		}
	}
	defer file.Close()

	m, err := file.Write([]byte(s))
	if err != nil {
		fmt.Fprintf(os.Stderr, "error writing to %s: %s\n", fname, err)
		return
	}

	if m != len(s) {
		fmt.Fprintf(os.Stderr, "short write %s\n", fname)
		return
	}
}