Code666 (代码贴、代码片段)

[Go] golang自定义的ftp客户端库 →→→→→进入此内容的聊天室

来自 , 2019-02-08, 写在 Go, 查看 138 次.
URL http://www.code666.cn/view/3677481d
  1. // FTP Client for Google Go language.
  2. // Author: smallfish <smallfish.xy@gmail.com>
  3.  
  4. package ftp
  5.  
  6. import (
  7. "fmt"
  8. "os"
  9. "net"
  10. "strconv"
  11. "strings"
  12. )
  13.  
  14. type FTP struct {
  15. host string
  16. port int
  17. user string
  18. passwd string
  19. pasv int
  20. cmd string
  21. Code int
  22. Message string
  23. Debug bool
  24. stream []byte
  26. Error os.Error
  27. }
  28.  
  29. func (ftp *FTP) debugInfo(s string) {
  30. if ftp.Debug {
  31. fmt.Println(s)
  32. }
  33. }
  34.  
  35. func (ftp *FTP) Connect(host string, port int) {
  36. addr := fmt.Sprintf("%s:%d", host, port)
  37. ftp.conn, ftp.Error = net.Dial("tcp", addr)
  38. ftp.Response()
  39. ftp.host = host
  40. ftp.port = port
  41. }
  42.  
  43. func (ftp *FTP) Login(user, passwd string) {
  44. ftp.Request("USER " + user)
  45. ftp.Request("PASS " + passwd)
  46. ftp.user = user
  47. ftp.passwd = passwd
  48. }
  49.  
  50. func (ftp *FTP) Response() (code int, message string) {
  51. ret := make([]byte, 1024)
  52. n, _ := ftp.conn.Read(ret)
  53. msg := string(ret[:n])
  54. code, _ = strconv.Atoi(msg[:3])
  55. message = msg[4 : len(msg)-2]
  56. ftp.debugInfo("<*cmd*> " + ftp.cmd)
  57. ftp.debugInfo(fmt.Sprintf("<*code*> %d", code))
  58. ftp.debugInfo("<*message*> " + message)
  59. return
  60. }
  61.  
  62. func (ftp *FTP) Request(cmd string) {
  63. ftp.conn.Write([]byte(cmd + "\r\n"))
  64. ftp.cmd = cmd
  65. ftp.Code, ftp.Message = ftp.Response()
  66. if cmd == "PASV" {
  67. start, end := strings.Index(ftp.Message, "("), strings.Index(ftp.Message, ")")
  68. s := strings.Split(ftp.Message[start:end], ",", -1)
  69. l1, _ := strconv.Atoi(s[len(s)-2])
  70. l2, _ := strconv.Atoi(s[len(s)-1])
  71. ftp.pasv = l1*256 + l2
  72. }
  73. if (cmd != "PASV") && (ftp.pasv > 0) {
  74. ftp.Message = newRequest(ftp.host, ftp.pasv, ftp.stream)
  75. ftp.debugInfo("<*response*> " + ftp.Message)
  76. ftp.pasv = 0
  77. ftp.stream = nil
  78. ftp.Code, _ = ftp.Response()
  79. }
  80. }
  81.  
  82. func (ftp *FTP) Pasv() {
  83. ftp.Request("PASV")
  84. }
  85.  
  86. func (ftp *FTP) Pwd() {
  87. ftp.Request("PWD")
  88. }
  89.  
  90. func (ftp *FTP) Cwd(path string) {
  91. ftp.Request("CWD " + path)
  92. }
  93.  
  94. func (ftp *FTP) Mkd(path string) {
  95. ftp.Request("MKD " + path)
  96. }
  97.  
  98. func (ftp *FTP) Size(path string) (size int) {
  99. ftp.Request("SIZE " + path)
  100. size, _ = strconv.Atoi(ftp.Message)
  101. return
  102. }
  103.  
  104. func (ftp *FTP) List() {
  105. ftp.Pasv()
  106. ftp.Request("LIST")
  107. }
  108.  
  109. func (ftp *FTP) Stor(file string, data []byte) {
  110. ftp.Pasv()
  111. if data != nil {
  112. ftp.stream = data
  113. }
  114. ftp.Request("STOR " + file)
  115. }
  116.  
  117. func (ftp *FTP) Quit() {
  118. ftp.Request("QUIT")
  119. ftp.conn.Close()
  120. }
  121.  
  122. // new connect to FTP pasv port, return data
  123. func newRequest(host string, port int, b []byte) string {
  124. conn, _ := net.Dial("tcp", fmt.Sprintf("%s:%d", host, port))
  125. defer conn.Close()
  126. if b != nil {
  127. conn.Write(b)
  128. return "OK"
  129. }
  130. ret := make([]byte, 4096)
  131. n, _ := conn.Read(ret)
  132. return string(ret[:n])
  133. }
  134.  
  135.  
  136. //go/8663

回复 "golang自定义的ftp客户端库"

这儿你可以回复上面这条便签

captcha

贴代码片段 - Code666