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 }