_TCPFileTransfer.au3

SUBMITTED BY: Guest

DATE: May 28, 2013, 8:22 a.m.

FORMAT: Text only

SIZE: 5.1 kB

Raw Download

HITS: 1023

Report
  1. #include <WinAPI.au3>
  2. #cs ----------------------------------------------------------------------------
  4. AutoIt Version: 3.3.6.0
  5. Author: DELmE
  7. Function:
  8. _TCPFileRecv
  10. Description:
  11. Receives a file being sent over a TCP connection
  13. Parameters:
  14. sock The TCP socket to be used
  15. rFile The name of the file to save to
  16. offset = 0 The offset to start writing data to the file at (to be used if the transfer is somehow interrupted) default value is 0 (no offset)
  17. $DataSize = 1024 The number of bytes of data to be received per packet, default value is 1024
  19. Return value:
  20. Success _TCPFileRecv returns the number of bytes written to the file
  21. Failure _TCPFileRecv returns 0 if the function failed and sets @error to the following:
  22. 1 = Error opening the file
  23. 2 = Failed to set offset
  24. 3 = Failed to receive data on socket (invalid socket)
  25. 4 = Unable to write data to the file
  26. Also sets @extended to the number of bytes of data that was successfully written to the file
  28. #ce ----------------------------------------------------------------------------
  29. Func _TCPFileRecv($sock, $rFile, $offset = 0, $DataSize = 1024)
  30. Local $i, $x, $lpszBuff, $nBytes, $rBuff, $hFile
  31. $nBytes = 0
  32. $lpszBuff = DllStructCreate("byte["&$DataSize&"]")
  33. $hFile = _WinAPI_CreateFile($rFile,3,4,4)
  34. If $hFile = 0 Then
  35. _WinAPI_CloseHandle($hFile)
  36. SetError(1)
  37. SetExtended($nBytes)
  38. Return 0
  39. EndIf
  40. While 1
  41. If _WinAPI_SetFilePointer($hFile, $offset, 0) = -1 Then
  42. _WinAPI_CloseHandle($hFile)
  43. SetError(2)
  44. SetExtended($nBytes)
  45. Return 0
  46. EndIf
  47. $rBuff = ""
  48. $rBuff = TCPRecv($sock, $DataSize)
  49. If $rBuff <> "" Then
  50. If @error = -1 Then
  51. _WinAPI_CloseHandle($hFile)
  52. SetError(3)
  53. SetExtended($nBytes)
  54. Return 0
  55. EndIf
  56. If $rBuff = @CRLF&@CRLF Then
  57. ExitLoop
  58. Else
  59. DllStructSetData($lpszBuff, 1, $rBuff)
  60. $x = StringLen(BinaryToString($rBuff))
  61. If _WinAPI_WriteFile($hFile, DllStructGetPtr($lpszBuff), $x, $i) = False Then
  62. _WinAPI_CloseHandle($hFile)
  63. SetError(4)
  64. SetExtended($nBytes)
  65. Return 0
  66. Else
  67. $nBytes += $i
  68. $offset += $i
  69. EndIf
  70. EndIf
  71. EndIf
  72. WEnd
  73. _WinAPI_CloseHandle($hFile)
  74. SetExtended($nBytes)
  75. Return $nBytes
  76. EndFunc ;==> _TCPFileRecv
  78. #cs ----------------------------------------------------------------------------
  80. AutoIt Version: 3.3.6.0
  81. Author: DELmE
  83. Function:
  84. _TCPFileSend
  86. Description:
  87. Sends a file over a TCP connection
  89. Parameters:
  90. sock The tcp socket to be used
  91. sFile The file to be sent
  92. offset = 0 The offset to start reading data from the file at (to be used if the transfer is somehow interrupted) default value is 0 (no offset)
  93. $DataSize = 1024 The number of bytes of data to be sent per packet, default value is 1024
  95. Return value:
  96. Success _TCPFileSend returns the number of bytes sent
  97. Failure _TCPFileSend returns 0 if an error occurred and sets @error to the following:
  98. 1 = The file does not exist
  99. 2 = Failed to set offset
  100. 3 = Failed to read data from file
  101. 4 = Failed to send data through socket
  102. Also sets @extended to the number of bytes of data that was successfully sent
  104. #ce ----------------------------------------------------------------------------
  105. Func _TCPFileSend($sock, $sFile, $offset = 0, $DataSize = 1024)
  106. Local $i, $lpszBuff, $nBytes, $sBuff, $hFile, $nFileSize
  107. $nFileSize = FileGetSize($sFile)
  108. $nFileSize -= $offset
  109. If $nFileSize < $DataSize Then $DataSize = $nFileSize
  110. $nBytes = 0
  111. $lpszBuff = DllStructCreate("byte["&$DataSize&"]")
  112. $hFile = _WinAPI_CreateFile($sFile,2,2,2)
  113. If $hFile = 0 Then
  114. _WinAPI_CloseHandle($hFile)
  115. SetError(1)
  116. SetExtended($nBytes)
  117. Return 0
  118. EndIf
  119. While $nFileSize > 0
  120. If _WinAPI_SetFilePointer($hFile, $offset, 0) = -1 Then
  121. _WinAPI_CloseHandle($hFile)
  122. SetError(2)
  123. SetExtended($nBytes)
  124. Return 0
  125. EndIf
  126. DllStructSetData($lpszBuff,1,"")
  127. If _WinAPI_ReadFile($hFile, DllStructGetPtr($lpszBuff), $DataSize, $i) = False Then
  128. _WinAPI_CloseHandle($hFile)
  129. SetError(3)
  130. Return 0
  131. Else
  132. $sBuff = DllStructGetData($lpszBuff, 1)
  133. $i = TCPSend($sock, $sBuff)
  134. If $i = 0 Then
  135. _WinAPI_CloseHandle($hFile)
  136. SetError(4)
  137. SetExtended($nBytes)
  138. Return 0
  139. Else
  140. $nBytes += $i
  141. $offset += $i
  142. $nFileSize -= $i
  143. EndIf
  144. EndIf
  145. WEnd
  146. _WinAPI_CloseHandle($hFile)
  147. TCPSend($sock, @CRLF&@CRLF)
  148. SetExtended($nBytes)
  149. Return $nBytes
  150. EndFunc ;==> _TCPFileSend
comments powered by Disqus