Ticket #1499: _FTP_Listtoarray.au3

File _FTP_Listtoarray.au3, 7.9 KB (added by Beege, on Feb 26, 2010 at 9:47:53 PM)
Line 
1#include <FTPEx.au3>
2;new
3$Open = _FTP_Open('MyFTP Control')
4$Connect = _FTP_Connect($Open, 'ftp.mozilla.org', '', '')
5If Not @error Then ConsoleWrite('Connection Successful.' & @CRLF)
6_FTP_DirSetCurrent($Connect, '/pub/addons')
7If Not @error Then ConsoleWrite('Current Directory = ' & _FTP_DirGetCurrent($Connect) & @CRLF)
8$Time = TimerInit()
9$Filelist = _FTP_ListToArrayEx2($Connect)
10ConsoleWrite($Filelist[0][0] & ' Dirs' & @CRLF & 'Time = ' & TimerDiff($Time) & @CRLF & @CRLF)
11_FTP_Close($Connect)
12
13;old
14$Open = _FTP_Open('MyFTP Control')
15$Connect = _FTP_Connect($Open, 'ftp.mozilla.org', '', '')
16If Not @error Then ConsoleWrite('Connection Successful.' & @CRLF)
17_FTP_DirSetCurrent($Connect, '/pub/addons')
18If Not @error Then ConsoleWrite('Current Directory = ' & _FTP_DirGetCurrent($Connect) & @CRLF)
19$Time = TimerInit()
20$Filelist = _FTP_ListToArrayEx($Connect)
21ConsoleWrite($Filelist[0][0] & ' Dirs' & @CRLF & 'Time = ' & TimerDiff($Time) & @CRLF & @CRLF)
22_FTP_Close($Connect)
23
24
25Func _FTP_ListToArrayEx2($l_FTPSession, $Return_Type = 0, $l_Flags = 0, $b_Fmt = 1, $l_Context = 0)
26 If $__ghWinInet_FTP = -1 Then Return SetError(-2, 0, 0)
27 Return __FTP_ListToArray2($l_FTPSession, $Return_Type, $l_Flags, $b_Fmt, 6, $l_Context)
28EndFunc ;==>_FTP_ListToArrayEx
29
30Func __FTP_ListToArray2($l_FTPSession, $Return_Type = 0, $l_Flags = 0, $bFmt = 1, $ArrayCount = 6, $l_Context = 0)
31 Local $tWIN32_FIND_DATA, $tFileTime, $IsDir, $callFindNext
32 Local $DirectoryIndex = 0, $FileIndex = 0
33
34 If $ArrayCount = 1 Then
35 Local $FileArray[1], $DirectoryArray[1]
36 Else
37 Local $FileArray[1][$ArrayCount], $DirectoryArray[1][$ArrayCount]
38 EndIf
39
40 If $Return_Type < 0 Or $Return_Type > 2 Then Return SetError(1, 0, $FileArray)
41
42;~ Global Const $tagWIN32_FIND_DATA = "DWORD dwFileAttributes; dword ftCreationTime[2]; dword ftLastAccessTime[2]; dword ftLastWriteTime[2]; DWORD nFileSizeHigh; DWORD nFileSizeLow; dword dwReserved0; dword dwReserved1; WCHAR cFileName[260]; WCHAR cAlternateFileName[14];"
43 $tWIN32_FIND_DATA = DllStructCreate($tagWIN32_FIND_DATA)
44 Local $callFindFirst = DllCall($__ghWinInet_FTP, 'handle', 'FtpFindFirstFileW', 'handle', $l_FTPSession, 'wstr', "", 'ptr', DllStructGetPtr($tWIN32_FIND_DATA), 'dword', $l_Flags, 'dword_ptr', $l_Context)
45 If @error Or Not $callFindFirst[0] Then Return SetError(1, _WinAPI_GetLastError(), 0)
46
47 Do
48 $IsDir = BitAND(DllStructGetData($tWIN32_FIND_DATA, "dwFileAttributes"), $FILE_ATTRIBUTE_DIRECTORY) = $FILE_ATTRIBUTE_DIRECTORY
49 If $IsDir And ($Return_Type <> 2) Then
50 $DirectoryIndex += 1
51 If $ArrayCount = 1 Then
52 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
53 ;Orig: ReDim $DirectoryArray[$DirectoryIndex + 1]
54 If UBound($DirectoryArray) < $DirectoryIndex+1 Then ReDim $DirectoryArray[$DirectoryIndex*2]
55 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
56 $DirectoryArray[$DirectoryIndex] = DllStructGetData($tWIN32_FIND_DATA, "cFileName")
57 Else
58 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
59 ;Orig: ReDim $DirectoryArray[$DirectoryIndex + 1][$ArrayCount]
60 If UBound($DirectoryArray) < $DirectoryIndex+1 Then ReDim $DirectoryArray[$DirectoryIndex*2][$ArrayCount]
61 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
62 $DirectoryArray[$DirectoryIndex][0] = DllStructGetData($tWIN32_FIND_DATA, "cFileName")
63
64 $DirectoryArray[$DirectoryIndex][1] = _WinAPI_MakeQWord(DllStructGetData($tWIN32_FIND_DATA, "nFileSizeLow"), DllStructGetData($tWIN32_FIND_DATA, "nFileSizeHigh"))
65 If $ArrayCount = 6 Then
66 $DirectoryArray[$DirectoryIndex][2] = DllStructGetData($tWIN32_FIND_DATA, "dwFileAttributes")
67 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftLastWriteTime"))
68 $DirectoryArray[$DirectoryIndex][3] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
69 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftCreationTime"))
70 $DirectoryArray[$DirectoryIndex][4] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
71 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftLastAccessTime"))
72 $DirectoryArray[$DirectoryIndex][5] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
73 EndIf
74 EndIf
75 ElseIf Not $IsDir And $Return_Type <> 1 Then
76 $FileIndex += 1
77 If $ArrayCount = 1 Then
78 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
79 ;Orig: ReDim $FileArray[$FileIndex + 1]
80 If UBound($FileArray) < $FileIndex+1 Then ReDim $FileArray[$FileIndex*2]
81 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
82 $FileArray[$FileIndex] = DllStructGetData($tWIN32_FIND_DATA, "cFileName")
83 Else
84 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
85 ;Orig: ReDim $FileArray[$FileIndex + 1][$ArrayCount]
86 If UBound($FileArray) < $FileIndex+1 Then ReDim $FileArray[$FileIndex*2][$ArrayCount]
87 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
88 $FileArray[$FileIndex][0] = DllStructGetData($tWIN32_FIND_DATA, "cFileName")
89
90 $FileArray[$FileIndex][1] = _WinAPI_MakeQWord(DllStructGetData($tWIN32_FIND_DATA, "nFileSizeLow"), DllStructGetData($tWIN32_FIND_DATA, "nFileSizeHigh"))
91 If $ArrayCount = 6 Then
92 $FileArray[$FileIndex][2] = DllStructGetData($tWIN32_FIND_DATA, "dwFileAttributes")
93
94 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftLastWriteTime"))
95 $FileArray[$FileIndex][3] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
96 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftCreationTime"))
97 $FileArray[$FileIndex][4] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
98 $tFileTime = DllStructCreate($tagFILETIME, DllStructGetPtr($tWIN32_FIND_DATA, "ftLastAccessTime"))
99 $FileArray[$FileIndex][5] = _Date_Time_FileTimeToStr( $tFileTime ,$bFmt)
100 EndIf
101 EndIf
102 EndIf
103
104 $callFindNext = DllCall($__ghWinInet_FTP, 'bool', 'InternetFindNextFileW', 'handle', $callFindFirst[0], 'ptr', DllStructGetPtr($tWIN32_FIND_DATA))
105 If @error Then Return SetError(2, _WinAPI_GetLastError(), 0)
106 Until Not $callFindNext[0]
107
108 DllCall($__ghWinInet_FTP, 'bool', 'InternetCloseHandle', 'handle', $callFindFirst[0])
109 ; No need to test @error.
110
111 If $ArrayCount = 1 Then
112 $DirectoryArray[0] = $DirectoryIndex
113 $FileArray[0] = $FileIndex
114 Else
115 $DirectoryArray[0][0] = $DirectoryIndex
116 $FileArray[0][0] = $FileIndex
117 EndIf
118
119 Switch $Return_Type
120 Case 0
121 If $ArrayCount = 1 Then
122 ReDim $DirectoryArray[$DirectoryArray[0] + $FileArray[0] + 1]
123 For $i = 1 To $FileIndex
124 $DirectoryArray[$DirectoryArray[0] + $i] = $FileArray[$i]
125 Next
126 $DirectoryArray[0] += $FileArray[0]
127 Else
128 ReDim $DirectoryArray[$DirectoryArray[0][0] + $FileArray[0][0] + 1][$ArrayCount]
129 For $i = 1 To $FileIndex
130 For $j = 0 To $ArrayCount-1
131 $DirectoryArray[$DirectoryArray[0][0] + $i][$j] = $FileArray[$i][$j]
132 Next
133 Next
134 $DirectoryArray[0][0] += $FileArray[0][0]
135 EndIf
136 Return $DirectoryArray
137 Case 1
138 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
139 ReDim $DirectoryArray[$DirectoryIndex+1]
140 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
141 Return $DirectoryArray
142 Case 2
143 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
144 ReDim $FileArray[$FileIndex +1]
145 ;XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
146 Return $FileArray
147 EndSwitch
148EndFunc ;==>__FTP_ListToArray