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 <wininet.h>
  23 #include <stdio.h>
  24 #include <errno.h>
  25
  26 #include "winetest.h"
  27
  28 #define USER_AGENT  "Winetest Shell"
  29 #define SERVER_NAME "test.winehq.org"
  30 #define URL_PATH "/submit"
  31 #define SEP "--8<--cut-here--8<--"
  32 #define CONTENT_HEADERS "Content-Type: multipart/form-data; boundary=\"" SEP "\"\r\n" \
  33                         "Content-Length: %u\r\n\r\n"
  34 static const char body1[] = "--" SEP "\r\n"
  35     "Content-Disposition: form-data; name=\"reportfile\"; filename=\"%s\"\r\n"
  36     "Content-Type: application/octet-stream\r\n\r\n";
  37 static const char body2[] = "\r\n--" SEP "\r\n"
  38     "Content-Disposition: form-data; name=\"submit\"\r\n\r\n"
  39     "Upload File\r\n"
  40     "--" SEP "--\r\n";
  41
  42 static SOCKET
  43 open_http (const char *server)
  44 {
  45     WSADATA wsad;
  46     struct sockaddr_in sa;
  47     SOCKET s;
  48
  49     report (R_STATUS, "Opening HTTP connection to %s", server);
  50     if (WSAStartup (MAKEWORD (2,2), &wsad)) return INVALID_SOCKET;
  51
  52     sa.sin_family = AF_INET;
  53     sa.sin_port = htons (80);
  54     sa.sin_addr.s_addr = inet_addr (server);
  55     if (sa.sin_addr.s_addr == INADDR_NONE) {
  56         struct hostent *host = gethostbyname (server);
  57         if (!host) {
  58             report (R_ERROR, "Hostname lookup failed for %s", server);
  59             goto failure;
  60         }
  61         sa.sin_addr.s_addr = ((struct in_addr *)host->h_addr)->s_addr;
  62     }
  63     s = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
  64     if (s == INVALID_SOCKET) {
  65         report (R_ERROR, "Can't open network socket: %d",
  66                 WSAGetLastError ());
  67         goto failure;
  68     }
  69     if (!connect (s, (struct sockaddr*)&sa, sizeof (struct sockaddr_in)))
  70         return s;
  71
  72     report (R_ERROR, "Can't connect: %d", WSAGetLastError ());
  73     closesocket (s);
  74  failure:
  75     WSACleanup ();
  76     return INVALID_SOCKET;
  77 }
  78
  79 static int
  80 close_http (SOCKET s)
  81 {
  82     int ret;
  83
  84     ret = closesocket (s);
  85     return (WSACleanup () || ret);
  86 }
  87
  88 static int
  89 send_buf (SOCKET s, const char *buf, size_t length)
  90 {
  91     int sent;
  92
  93     while (length > 0) {
  94         sent = send (s, buf, length, 0);
  95         if (sent == SOCKET_ERROR) return 1;
  96         buf += sent;
  97         length -= sent;
  98     }
  99     return 0;
 100 }
 101
 102 static int
 103 send_str (SOCKET s, ...)
 104 {
 105     va_list ap;
 106     char *p;
 107     int ret;
 108     size_t len;
 109
 110     va_start (ap, s);
 111     p = vstrmake (&len, ap);
 112     va_end (ap);
 113     if (!p) return 1;
 114     ret = send_buf (s, p, len);
 115     heap_free (p);
 116     return ret;
 117 }
 118
 119 static int
 120 send_file_direct (const char *name)
 121 {
 122     SOCKET s;
 123     HANDLE file;
 124 #define BUFLEN 8192
 125     char buffer[BUFLEN+1];
 126     DWORD bytes_read, filesize;
 127     size_t total, count;
 128     char *str;
 129     int ret;
 130
 131     /* RFC 2616 */
 132     static const char head[] = "POST " URL_PATH " HTTP/1.0\r\n"
 133         "Host: " SERVER_NAME "\r\n"
 134         "User-Agent: " USER_AGENT "\r\n"
 135         CONTENT_HEADERS
 136         "\r\n";
 137
 138     s = open_http (SERVER_NAME);
 139     if (s == INVALID_SOCKET) return 1;
 140
 141     file = CreateFileA( name, GENERIC_READ,
 142                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
 143                         NULL, OPEN_EXISTING, 0, NULL );
 144
 145     if ((file == INVALID_HANDLE_VALUE) &&
 146         (GetLastError() == ERROR_INVALID_PARAMETER)) {
 147         /* FILE_SHARE_DELETE not supported on win9x */
 148         file = CreateFileA( name, GENERIC_READ,
 149                             FILE_SHARE_READ | FILE_SHARE_WRITE,
 150                             NULL, OPEN_EXISTING, 0, NULL );
 151     }
 152     if (file == INVALID_HANDLE_VALUE)
 153     {
 154         report (R_WARNING, "Can't open file '%s': %u", name, GetLastError());
 155         goto abort1;
 156     }
 157     filesize = GetFileSize( file, NULL );
 158     if (filesize > 1.5*1024*1024) {
 159         report (R_WARNING,
 160                 "File too big (%.1f MB > 1.5 MB); submitting partial report.",
 161                 filesize/1024.0/1024);
 162         filesize = (DWORD) 1.5*1024*1024;
 163     }
 164
 165     report (R_STATUS, "Sending header");
 166     str = strmake (&total, body1, name);
 167     ret = send_str (s, head, filesize + total + sizeof body2 - 1) ||
 168         send_buf (s, str, total);
 169     heap_free (str);
 170     if (ret) {
 171         report (R_WARNING, "Error sending header: %d, %d",
 172                 errno, WSAGetLastError ());
 173         goto abort2;
 174     }
 175
 176     report (R_STATUS, "Sending %u bytes of data", filesize);
 177     report (R_PROGRESS, 2, filesize);
 178     total = 0;
 179     while (total < filesize && ReadFile( file, buffer, BUFLEN/2, &bytes_read, NULL )) {
 180         if (aborting) goto abort2;
 181         if (!bytes_read) break;
 182         total += bytes_read;
 183         if (total > filesize) bytes_read -= total - filesize;
 184         if (send_buf (s, buffer, bytes_read)) {
 185             report (R_WARNING, "Error sending body: %d, %d",
 186                     errno, WSAGetLastError ());
 187             goto abort2;
 188         }
 189         report (R_DELTA, bytes_read, "Network transfer: In progress");
 190     }
 191     CloseHandle( file );
 192
 193     if (send_buf (s, body2, sizeof body2 - 1)) {
 194         report (R_WARNING, "Error sending trailer: %d, %d",
 195                 errno, WSAGetLastError ());
 196         goto abort1;
 197     }
 198     report (R_DELTA, 0, "Network transfer: Done");
 199
 200     total = 0;
 201     while ((bytes_read = recv (s, buffer+total, BUFLEN-total, 0))) {
 202         if ((signed)bytes_read == SOCKET_ERROR) {
 203             report (R_WARNING, "Error receiving reply: %d, %d",
 204                     errno, WSAGetLastError ());
 205             goto abort1;
 206         }
 207         total += bytes_read;
 208         if (total == BUFLEN) {
 209             report (R_WARNING, "Buffer overflow");
 210             goto abort1;
 211         }
 212     }
 213     if (close_http (s)) {
 214         report (R_WARNING, "Error closing connection: %d, %d",
 215                 errno, WSAGetLastError ());
 216         return 1;
 217     }
 218
 219     str = strmake (&count, "Received %s (%d bytes).\n",
 220                    name, filesize);
 221     ret = total < count || memcmp (str, buffer + total - count, count) != 0;
 222     heap_free (str);
 223     if (ret) {
 224         buffer[total] = 0;
 225         str = strstr (buffer, "\r\n\r\n");
 226         if (!str) str = buffer;
 227         else str = str + 4;
 228         report (R_ERROR, "Can't submit logfile '%s'. "
 229                 "Server response: %s", name, str);
 230     }
 231     return ret;
 232
 233  abort2:
 234     CloseHandle( file );
 235  abort1:
 236     close_http (s);
 237     return 1;
 238 }
 239
 240 static int
 241 send_file_wininet (const char *name)
 242 {
 243     int ret = 0;
 244     HMODULE wininet_mod = NULL;
 245     HINTERNET (WINAPI *pInternetOpen)(LPCSTR agent, DWORD access_type, LPCSTR proxy_name, LPCSTR proxy_bypass, DWORD flags);
 246     HINTERNET (WINAPI *pInternetConnect)(HINTERNET session, LPCSTR server_name, INTERNET_PORT server_port, LPCSTR username, LPCSTR password, DWORD service, DWORD flags, DWORD_PTR *context);
 247     HINTERNET (WINAPI *pHttpOpenRequest)(HINTERNET connection, LPCSTR verb, LPCSTR object_name, LPCSTR version, LPCSTR referer, LPCSTR *accept_types, DWORD flags, DWORD_PTR context);
 248     BOOL (WINAPI *pHttpSendRequestEx)(HINTERNET request, LPINTERNET_BUFFERSA buffers_in, LPINTERNET_BUFFERSA buffers_out, DWORD flags, DWORD_PTR context);
 249     BOOL (WINAPI *pInternetWriteFile)(HINTERNET file, LPCVOID buffer, DWORD number_of_bytes_to_write, LPDWORD number_of_bytes_written);
 250     BOOL (WINAPI *pHttpEndRequest)(HINTERNET request, LPINTERNET_BUFFERSA buffers_out, DWORD flags, DWORD_PTR context);
 251     BOOL (WINAPI *pInternetReadFile)(HINTERNET file, LPCVOID buffer, DWORD number_of_bytes_to_read, LPDWORD number_of_bytes_read);
 252     BOOL (WINAPI *pInternetCloseHandle)(HINTERNET Handle) = NULL;
 253     HANDLE file = INVALID_HANDLE_VALUE;
 254     DWORD filesize, bytes_read, bytes_written;
 255     size_t total, count;
 256     char *str = NULL;
 257     HINTERNET session = NULL;
 258     HINTERNET connection = NULL;
 259     HINTERNET request = NULL;
 260     INTERNET_BUFFERSA buffers_in = { 0 };
 261     char buffer[BUFLEN+1];
 262
 263     static const char extra_headers[] =
 264         CONTENT_HEADERS;
 265
 266     wininet_mod = LoadLibrary ("wininet.dll");
 267     if (wininet_mod == NULL)
 268         goto done;
 269     pInternetOpen = (void *)GetProcAddress(wininet_mod, "InternetOpenA");
 270     pInternetConnect = (void *)GetProcAddress(wininet_mod, "InternetConnectA");
 271     pHttpOpenRequest = (void *)GetProcAddress(wininet_mod, "HttpOpenRequestA");
 272     pHttpSendRequestEx = (void *)GetProcAddress(wininet_mod, "HttpSendRequestExA");
 273     pInternetWriteFile = (void *)GetProcAddress(wininet_mod, "InternetWriteFile");
 274     pHttpEndRequest = (void *)GetProcAddress(wininet_mod, "HttpEndRequestA");
 275     pInternetReadFile = (void *)GetProcAddress(wininet_mod, "InternetReadFile");
 276     pInternetCloseHandle = (void *)GetProcAddress(wininet_mod, "InternetCloseHandle");
 277     if (pInternetOpen == NULL || pInternetConnect == NULL || pHttpOpenRequest == NULL || pHttpSendRequestEx == NULL || pHttpEndRequest == NULL ||
 278         pInternetWriteFile == NULL || pInternetReadFile == NULL || pInternetCloseHandle == NULL) {
 279         goto done;
 280     }
 281
 282     ret = 1;
 283
 284     file = CreateFileA( name, GENERIC_READ,
 285                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
 286                         NULL, OPEN_EXISTING, 0, NULL );
 287
 288     if ((file == INVALID_HANDLE_VALUE) &&
 289         (GetLastError() == ERROR_INVALID_PARAMETER)) {
 290         /* FILE_SHARE_DELETE not supported on win9x */
 291         file = CreateFileA( name, GENERIC_READ,
 292                             FILE_SHARE_READ | FILE_SHARE_WRITE,
 293                             NULL, OPEN_EXISTING, 0, NULL );
 294     }
 295     if (file == INVALID_HANDLE_VALUE) {
 296         report (R_WARNING, "Can't open file '%s': %u", name, GetLastError());
 297         goto done;
 298     }
 299
 300     filesize = GetFileSize( file, NULL );
 301     if (filesize > 1.5*1024*1024) {
 302         report (R_WARNING,
 303                 "File too big (%.1f MB > 1.5 MB); submitting partial report.",
 304                 filesize/1024.0/1024);
 305         filesize = 1.5*1024*1024;
 306     }
 307
 308     report (R_STATUS, "Opening HTTP connection to " SERVER_NAME);
 309     session = pInternetOpen (USER_AGENT, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
 310     if (session == NULL) {
 311         report (R_WARNING, "Unable to open connection, error %u", GetLastError());
 312         goto done;
 313     }
 314     connection = pInternetConnect (session, SERVER_NAME, INTERNET_DEFAULT_HTTP_PORT, "", "", INTERNET_SERVICE_HTTP, 0, 0);
 315     if (connection == NULL) {
 316         report (R_WARNING, "Unable to connect, error %u", GetLastError());
 317         goto done;
 318     }
 319     request = pHttpOpenRequest (connection, "POST", URL_PATH, NULL, NULL, NULL,
 320                                 INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_NO_COOKIES | INTERNET_FLAG_NO_UI |
 321                                 INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_RELOAD, 0);
 322     if (request == NULL) {
 323         report (R_WARNING, "Unable to open request, error %u", GetLastError());
 324         goto done;
 325     }
 326
 327     report (R_STATUS, "Sending request");
 328     str = strmake (&total, body1, name);
 329     memset(&buffers_in, 0, sizeof(INTERNET_BUFFERSA));
 330     buffers_in.dwStructSize = sizeof(INTERNET_BUFFERSA);
 331     buffers_in.dwBufferTotal = filesize + total + sizeof body2 - 1;
 332     buffers_in.lpcszHeader = strmake (&count, extra_headers, buffers_in.dwBufferTotal);
 333     buffers_in.dwHeadersLength = count;
 334     if (! pHttpSendRequestEx(request, &buffers_in, NULL, 0, 0)) {
 335         report (R_WARNING, "Unable to send request, error %u", GetLastError());
 336         goto done;
 337     }
 338
 339     if (! pInternetWriteFile(request, str, total, &bytes_written) || bytes_written != total) {
 340         report (R_WARNING, "Unable to write body data, error %u", GetLastError());
 341         goto done;
 342     }
 343
 344     report (R_STATUS, "Sending %u bytes of data", filesize);
 345     report (R_PROGRESS, 2, filesize);
 346     total = 0;
 347     while (total < filesize && ReadFile( file, buffer, BUFLEN/2, &bytes_read, NULL )) {
 348         if (aborting) goto done;
 349         if (!bytes_read) break;
 350         total += bytes_read;
 351         if (total > filesize) bytes_read -= total - filesize;
 352         if (! pInternetWriteFile (request, buffer, bytes_read, &bytes_written) || bytes_written != bytes_read) {
 353             report (R_WARNING, "Error sending body: %u", GetLastError ());
 354             goto done;
 355         }
 356         report (R_DELTA, bytes_read, "Network transfer: In progress");
 357     }
 358
 359     if (! pInternetWriteFile(request, body2, sizeof body2 - 1, &bytes_written) || bytes_written != sizeof body2 - 1) {
 360         report (R_WARNING, "Unable to write final body data, error %u", GetLastError());
 361         goto done;
 362     }
 363     if (! pHttpEndRequest(request, NULL, 0, 0)) {
 364         report (R_WARNING, "Unable to end request, error %u", GetLastError());
 365         goto done;
 366     }
 367     report (R_DELTA, 0, "Network transfer: Done");
 368
 369     total = 0;
 370     do
 371     {
 372         if (! pInternetReadFile(request, buffer+total, BUFLEN-total, &bytes_read)) {
 373             report (R_WARNING, "Error receiving reply: %u", GetLastError ());
 374             goto done;
 375         }
 376         total += bytes_read;
 377         if (total == BUFLEN) {
 378             report (R_WARNING, "Buffer overflow");
 379             goto done;
 380         }
 381     }
 382     while (bytes_read != 0);
 383
 384     heap_free (str);
 385     str = strmake (&count, "Received %s (%d bytes).\n",
 386                    name, filesize);
 387     if (total < count || memcmp (str, buffer + total - count, count) != 0) {
 388         buffer[total] = 0;
 389         report (R_ERROR, "Can't submit logfile '%s'. "
 390                 "Server response: %s", name, buffer);
 391     }
 392
 393  done:
 394     if (buffers_in.lpcszHeader != NULL)
 395         heap_free((void *) buffers_in.lpcszHeader);
 396     if (str != NULL)
 397         heap_free (str);
 398     if (pInternetCloseHandle != NULL && request != NULL)
 399         pInternetCloseHandle (request);
 400     if (pInternetCloseHandle != NULL && connection != NULL)
 401         pInternetCloseHandle (connection);
 402     if (pInternetCloseHandle != NULL && session != NULL)
 403         pInternetCloseHandle (session);
 404     if (file != INVALID_HANDLE_VALUE)
 405         CloseHandle (file);
 406     if (wininet_mod != NULL)
 407         FreeLibrary (wininet_mod);
 408
 409     return ret;
 410 }
 411
 412 int
 413 send_file (const char *name)
 414 {
 415     return send_file_wininet( name ) || send_file_direct( name );
 416 }