Ticket #1499: _FTP_Listtoarray.au3

File _FTP_Listtoarray.au3, 7.9 KB (added by Beege, 15 years ago)
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