comparison netinstall/nio-http.cc @ 448:3078fd1074e8 r21-2-39

Import from CVS: tag r21-2-39
author cvs
date Mon, 13 Aug 2007 11:38:25 +0200
parents
children
comparison
equal deleted inserted replaced
447:4fc5f13f3bd3 448:3078fd1074e8
1 /*
2 * Copyright (c) 2000, Red Hat, Inc.
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * A copy of the GNU General Public License can be found at
10 * http://www.gnu.org/
11 *
12 * Written by DJ Delorie <dj@cygnus.com>
13 *
14 */
15
16 /* This file is responsible for implementing all direct HTTP protocol
17 channels. It is intentionally simplistic. */
18
19 #include "win32.h"
20 #include "winsock.h"
21 #include <stdio.h>
22 #include <stdlib.h>
23
24 #include "resource.h"
25 #include "state.h"
26 #include "simpsock.h"
27 #include "msg.h"
28
29 #include "netio.h"
30 #include "nio-http.h"
31
32 static char six2pr[64] = {
33 'A','B','C','D','E','F','G','H','I','J','K','L','M',
34 'N','O','P','Q','R','S','T','U','V','W','X','Y','Z',
35 'a','b','c','d','e','f','g','h','i','j','k','l','m',
36 'n','o','p','q','r','s','t','u','v','w','x','y','z',
37 '0','1','2','3','4','5','6','7','8','9','+','/'
38 };
39
40 static char *
41 base64_encode (char *username, char *password)
42 {
43 unsigned char *ep;
44 char *rp;
45 static char *rv = 0;
46 if (rv)
47 free (rv);
48 rv = (char *) malloc (2 * (strlen (username) + strlen (password)) + 5);
49
50 char *up = (char *) malloc (strlen (username) + strlen (password) + 6);
51 strcpy (up, username);
52 strcat (up, ":");
53 strcat (up, password);
54 ep = (unsigned char *)up + strlen (up);
55 *ep++ = 0;
56 *ep++ = 0;
57 *ep++ = 0;
58
59 char block[4];
60
61 rp = rv;
62
63 for (ep = (unsigned char *)up; *ep; ep += 3)
64 {
65 block[0] = six2pr[ep[0] >> 2];
66 block[1] = six2pr[((ep[0] << 4) & 0x30) | ((ep[1] >> 4) & 0x0f)];
67 block[2] = six2pr[((ep[1] << 2) & 0x3c) | ((ep[2] >> 6) & 0x03)];
68 block[3] = six2pr[ep[2] & 0x3f];
69
70 if (ep[1] == 0)
71 block[2] = block[3] = '=';
72 if (ep[2] == 0)
73 block[3] = '=';
74 memcpy (rp, block, 4);
75 rp += 4;
76 }
77 *rp = 0;
78
79 free (up);
80
81 return rv;
82 }
83
84 NetIO_HTTP::NetIO_HTTP (char *Purl)
85 : NetIO (Purl)
86 {
87 retry_get:
88 if (port == 0)
89 port = 80;
90
91 if (net_method == IDC_NET_PROXY)
92 s = new SimpleSocket (net_proxy_host, net_proxy_port);
93 else
94 s = new SimpleSocket (host, port);
95
96 if (!s->ok())
97 {
98 s = 0;
99 return;
100 }
101
102 if (net_method == IDC_NET_PROXY)
103 s->printf ("GET %s HTTP/1.0\r\n", url);
104 else
105 s->printf ("GET %s HTTP/1.0\r\n", path);
106 s->printf ("Host: %s:%d\r\n", host, port);
107
108 if (net_user && net_passwd)
109 s->printf ("Authorization: Basic %s\r\n",
110 base64_encode (net_user, net_passwd));
111
112 if (net_proxy_user && net_proxy_passwd)
113 s->printf ("Proxy-Authorization: Basic %s\r\n",
114 base64_encode (net_proxy_user, net_proxy_passwd));
115
116 s->printf ("\r\n");
117
118 char *l = s->gets ();
119 int code;
120 sscanf (l, "%*s %d", &code);
121 if (code >= 300 && code < 400)
122 {
123 do {
124 l = s->gets ();
125 if (_strnicmp (l, "Location:", 9) == 0)
126 {
127 char *u = l + 9;
128 while (*u == ' ' || *u == '\t')
129 u++;
130 set_url (u);
131 delete s;
132 goto retry_get;
133 }
134 } while (*l);
135 }
136 if (code == 401) /* authorization required */
137 {
138 get_auth ();
139 delete s;
140 goto retry_get;
141 }
142 if (code == 407) /* proxy authorization required */
143 {
144 get_proxy_auth ();
145 delete s;
146 goto retry_get;
147 }
148 if (code >= 300)
149 {
150 delete s;
151 s = 0;
152 return;
153 }
154 do {
155 l = s->gets ();
156 if (_strnicmp (l, "Content-Length:", 15) == 0)
157 sscanf (l, "%*s %d", &file_size);
158 } while (*l);
159 }
160
161 NetIO_HTTP::~NetIO_HTTP ()
162 {
163 if (s)
164 delete s;
165 }
166
167 int
168 NetIO_HTTP::ok ()
169 {
170 if (s)
171 return 1;
172 return 0;
173 }
174
175 int
176 NetIO_HTTP::read (char *buf, int nbytes)
177 {
178 return s->read (buf, nbytes);
179 }