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
}
}