xiwh / zmodem Goto Github PK

		lof, err := os.Open(loPath)
		if err != nil {
			return nil
		}
		stat, err := lof.Stat()
		if err != nil {
			return nil
		}
		zf, writer := zmodem.NewZModemFile(stat.Name(), int(stat.Size()))
		go func() {
			io.Copy(writer, lof)
		}()

		//zf, _ := zmodem.NewZModemLocalFile(loPath)

		return zf
	},
	OnCheckDownload: func(file *zmodem.ZModemFile) {},
	OnDownload: func(file *zmodem.ZModemFile, reader io.ReadCloser) (err error) {
		dir := "/Users/chenbo36/Downloads/test/aaa/"

		loPath := filepath.Join(dir, file.Filename)
		lof, err := osutil.CreateFile(loPath)
		if err != nil {
			return err
		}

		io.Copy(lof, reader)

		return err
	},
	Writer:     zmIn,
	EchoWriter: zmOut,
})

go func() {
	time.Sleep(1 * time.Second)
	zmIn.Write([]byte("cd /root/test\n"))
	time.Sleep(1 * time.Second)
	zmIn.Write([]byte("rz\n"))
}()

for {
	data := make([]byte, 8192, 8194)
	n, err := zmOut.Read(data)
	if err != nil {
		return
	}

	if n > 0 {
		fmt.Println(string(data[:n]))
	}

}