comctl32/listview: Remove superfluous forward declarations.
[wine] / programs / winetest / send.c
1 /*
2  * HTTP handling functions.
3  *
4  * Copyright 2003 Ferenc Wagner
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <winsock.h>
22 #include <stdio.h>
23 #include <errno.h>
24
25 #include "winetest.h"
26
27 static SOCKET
28 open_http (const char *server)
29 {
30     WSADATA wsad;
31     struct sockaddr_in sa;
32     SOCKET s;
33
34     report (R_STATUS, "Opening HTTP connection to %s", server);
35     if (WSAStartup (MAKEWORD (2,2), &wsad)) return INVALID_SOCKET;
36
37     sa.sin_family = AF_INET;
38     sa.sin_port = htons (80);
39     sa.sin_addr.s_addr = inet_addr (server);
40     if (sa.sin_addr.s_addr == INADDR_NONE) {
41         struct hostent *host = gethostbyname (server);
42         if (!host) {
43             report (R_ERROR, "Hostname lookup failed for %s", server);
44             goto failure;
45         }
46         sa.sin_addr.s_addr = ((struct in_addr *)host->h_addr)->s_addr;
47     }
48     s = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
49     if (s == INVALID_SOCKET) {
50         report (R_ERROR, "Can't open network socket: %d",
51                 WSAGetLastError ());
52         goto failure;
53     }
54     if (!connect (s, (struct sockaddr*)&sa, sizeof (struct sockaddr_in)))
55         return s;
56
57     report (R_ERROR, "Can't connect: %d", WSAGetLastError ());
58     closesocket (s);
59  failure:
60     WSACleanup ();
61     return INVALID_SOCKET;
62 }
63
64 static int
65 close_http (SOCKET s)
66 {
67     int ret;
68
69     ret = closesocket (s);
70     return (WSACleanup () || ret);
71 }
72
73 static int
74 send_buf (SOCKET s, const char *buf, size_t length)
75 {
76     int sent;
77
78     while (length > 0) {
79         sent = send (s, buf, length, 0);
80         if (sent == SOCKET_ERROR) return 1;
81         buf += sent;
82         length -= sent;
83     }
84     return 0;
85 }
86
87 static int
88 send_str (SOCKET s, ...)
89 {
90     va_list ap;
91     char *p;
92     int ret;
93     size_t len;
94
95     va_start (ap, s);
96     p = vstrmake (&len, ap);
97     va_end (ap);
98     if (!p) return 1;
99     ret = send_buf (s, p, len);
100     free (p);
101     return ret;
102 }
103
104 int
105 send_file (const char *name)
106 {
107     SOCKET s;
108     HANDLE file;
109 #define BUFLEN 8192
110     char buffer[BUFLEN+1];
111     DWORD bytes_read, filesize;
112     size_t total, count;
113     char *str;
114     int ret;
115
116     /* RFC 2616 */
117 #define SEP "--8<--cut-here--8<--"
118     static const char head[] = "POST /submit HTTP/1.0\r\n"
119         "Host: test.winehq.org\r\n"
120         "User-Agent: Winetest Shell\r\n"
121         "Content-Type: multipart/form-data; boundary=\"" SEP "\"\r\n"
122         "Content-Length: %u\r\n\r\n";
123     static const char body1[] = "--" SEP "\r\n"
124         "Content-Disposition: form-data; name=\"reportfile\"; filename=\"%s\"\r\n"
125         "Content-Type: application/octet-stream\r\n\r\n";
126     static const char body2[] = "\r\n--" SEP "\r\n"
127         "Content-Disposition: form-data; name=\"submit\"\r\n\r\n"
128         "Upload File\r\n"
129         "--" SEP "--\r\n";
130
131     s = open_http ("test.winehq.org");
132     if (s == INVALID_SOCKET) return 1;
133
134     file = CreateFileA( name, GENERIC_READ,
135                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
136                         NULL, OPEN_EXISTING, 0, NULL );
137
138     if ((file == INVALID_HANDLE_VALUE) &&
139         (GetLastError() == ERROR_INVALID_PARAMETER)) {
140         /* FILE_SHARE_DELETE not supported on win9x */
141         file = CreateFileA( name, GENERIC_READ,
142                             FILE_SHARE_READ | FILE_SHARE_WRITE,
143                             NULL, OPEN_EXISTING, 0, NULL );
144     }
145     if (file == INVALID_HANDLE_VALUE)
146     {
147         report (R_WARNING, "Can't open file '%s': %u", name, GetLastError());
148         goto abort1;
149     }
150     filesize = GetFileSize( file, NULL );
151     if (filesize > 1.5*1024*1024) {
152         report (R_WARNING,
153                 "File too big (%.1f MB > 1.5 MB); submitting partial report.",
154                 filesize/1024.0/1024);
155         filesize = 1.5*1024*1024;
156     }
157
158     report (R_STATUS, "Sending header");
159     str = strmake (&total, body1, name);
160     ret = send_str (s, head, filesize + total + sizeof body2 - 1) ||
161         send_buf (s, str, total);
162     free (str);
163     if (ret) {
164         report (R_WARNING, "Error sending header: %d, %d",
165                 errno, WSAGetLastError ());
166         goto abort2;
167     }
168
169     report (R_STATUS, "Sending %u bytes of data", filesize);
170     report (R_PROGRESS, 2, filesize);
171     total = 0;
172     while (total < filesize && ReadFile( file, buffer, BUFLEN/2, &bytes_read, NULL )) {
173         if (!bytes_read) break;
174         total += bytes_read;
175         if (total > filesize) bytes_read -= total - filesize;
176         if (send_buf (s, buffer, bytes_read)) {
177             report (R_WARNING, "Error sending body: %d, %d",
178                     errno, WSAGetLastError ());
179             goto abort2;
180         }
181         report (R_DELTA, bytes_read, "Network transfer: In progress");
182     }
183     CloseHandle( file );
184
185     if (send_buf (s, body2, sizeof body2 - 1)) {
186         report (R_WARNING, "Error sending trailer: %d, %d",
187                 errno, WSAGetLastError ());
188         goto abort1;
189     }
190     report (R_DELTA, 0, "Network transfer: Done");
191
192     total = 0;
193     while ((bytes_read = recv (s, buffer+total, BUFLEN-total, 0))) {
194         if ((signed)bytes_read == SOCKET_ERROR) {
195             report (R_WARNING, "Error receiving reply: %d, %d",
196                     errno, WSAGetLastError ());
197             goto abort1;
198         }
199         total += bytes_read;
200         if (total == BUFLEN) {
201             report (R_WARNING, "Buffer overflow");
202             goto abort1;
203         }
204     }
205     if (close_http (s)) {
206         report (R_WARNING, "Error closing connection: %d, %d",
207                 errno, WSAGetLastError ());
208         return 1;
209     }
210
211     str = strmake (&count, "Received %s (%d bytes).\n",
212                    name, filesize);
213     ret = memcmp (str, buffer + total - count, count);
214     free (str);
215     if (ret) {
216         buffer[total] = 0;
217         str = strstr (buffer, "\r\n\r\n");
218         if (!str) str = buffer;
219         else str = str + 4;
220         report (R_ERROR, "Can't submit logfile '%s'. "
221                 "Server response: %s", name, str);
222     }
223     return ret;
224
225  abort2:
226     CloseHandle( file );
227  abort1:
228     close_http (s);
229     return 1;
230 }