|
448
|
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 /* Simplified socket access functions */
|
|
|
17
|
|
|
18 #include "win32.h"
|
|
|
19 #include <winsock.h>
|
|
|
20 #include <stdio.h>
|
|
|
21 #include <stdarg.h>
|
|
|
22 #include <stdlib.h>
|
|
|
23
|
|
|
24 #include "simpsock.h"
|
|
|
25 #include "msg.h"
|
|
|
26
|
|
|
27 #define SSBUFSZ 1024
|
|
|
28
|
|
|
29 SimpleSocket::SimpleSocket (char *hostname, int port)
|
|
|
30 {
|
|
|
31 static int initted = 0;
|
|
|
32 if (!initted)
|
|
|
33 {
|
|
|
34 initted = 1;
|
|
|
35 WSADATA d;
|
|
|
36 WSAStartup (MAKEWORD (1,1), &d);
|
|
|
37 }
|
|
|
38
|
|
|
39 s = INVALID_SOCKET;
|
|
|
40 buf = (char *) malloc (SSBUFSZ + 3);
|
|
|
41 putp = getp = 0;
|
|
|
42
|
|
|
43 int i1, i2, i3, i4;
|
|
|
44 unsigned char ip[4];
|
|
|
45
|
|
|
46 if (sscanf (hostname, "%d.%d.%d.%d", &i1, &i2, &i3, &i4) == 4)
|
|
|
47 {
|
|
|
48 ip[0] = i1;
|
|
|
49 ip[1] = i2;
|
|
|
50 ip[2] = i3;
|
|
|
51 ip[3] = i4;
|
|
|
52 }
|
|
|
53 else
|
|
|
54 {
|
|
|
55 struct hostent *he;
|
|
|
56 he = gethostbyname (hostname);
|
|
|
57 if (!he)
|
|
|
58 {
|
|
|
59 msg ("Can't resolve `%s'\n", hostname);
|
|
|
60 return;
|
|
|
61 }
|
|
|
62 memcpy (ip, he->h_addr_list[0], 4);
|
|
|
63 }
|
|
|
64
|
|
|
65 s = socket (AF_INET, SOCK_STREAM, 0);
|
|
|
66 if (s == INVALID_SOCKET)
|
|
|
67 {
|
|
|
68 msg ("Can't create socket, %d", WSAGetLastError ());
|
|
|
69 return;
|
|
|
70 }
|
|
|
71
|
|
|
72 struct sockaddr_in name;
|
|
|
73
|
|
|
74 memset (&name, 0, sizeof (name));
|
|
|
75 name.sin_family = AF_INET;
|
|
|
76 name.sin_port = htons (port);
|
|
|
77 memcpy (&name.sin_addr, ip, 4);
|
|
|
78
|
|
|
79 if (connect (s, (sockaddr *)&name, sizeof(name)))
|
|
|
80 {
|
|
|
81 msg ("Can't connect to %s:%d", hostname, port);
|
|
|
82 closesocket (s);
|
|
|
83 s = INVALID_SOCKET;
|
|
|
84 return;
|
|
|
85 }
|
|
|
86
|
|
|
87 return;
|
|
|
88 }
|
|
|
89
|
|
|
90 SimpleSocket::~SimpleSocket ()
|
|
|
91 {
|
|
|
92 if (s != INVALID_SOCKET)
|
|
|
93 closesocket (s);
|
|
|
94 s = INVALID_SOCKET;
|
|
|
95 if (buf)
|
|
|
96 free (buf);
|
|
|
97 buf = 0;
|
|
|
98 }
|
|
|
99
|
|
|
100 int
|
|
|
101 SimpleSocket::ok ()
|
|
|
102 {
|
|
|
103 if (s == INVALID_SOCKET)
|
|
|
104 return 0;
|
|
|
105 return 1;
|
|
|
106 }
|
|
|
107
|
|
|
108 int
|
|
|
109 SimpleSocket::printf (char *fmt, ...)
|
|
|
110 {
|
|
|
111 char localbuf[SSBUFSZ];
|
|
|
112 va_list args;
|
|
|
113 va_start (args, fmt);
|
|
|
114 vsprintf (localbuf, fmt, args);
|
|
|
115 return send (s, localbuf, strlen (localbuf), 0);
|
|
|
116 }
|
|
|
117
|
|
|
118 int
|
|
|
119 SimpleSocket::write (char *localbuf, int len)
|
|
|
120 {
|
|
|
121 return send (s, localbuf, len, 0);
|
|
|
122 }
|
|
|
123
|
|
|
124 int
|
|
|
125 SimpleSocket::fill ()
|
|
|
126 {
|
|
|
127 if (putp == getp)
|
|
|
128 putp = getp = 0;
|
|
|
129
|
|
|
130 int n = SSBUFSZ - putp;
|
|
|
131 if (n == 0)
|
|
|
132 return 0;
|
|
|
133 int r = recv (s, buf + putp, n, 0);
|
|
|
134 if (r > 0)
|
|
|
135 {
|
|
|
136 putp += r;
|
|
|
137 return r;
|
|
|
138 }
|
|
|
139 return 0;
|
|
|
140 }
|
|
|
141
|
|
|
142 char *
|
|
|
143 SimpleSocket::gets ()
|
|
|
144 {
|
|
|
145 if (getp > 0 && putp > getp)
|
|
|
146 {
|
|
|
147 memmove (buf, buf+getp, putp-getp);
|
|
|
148 putp -= getp;
|
|
|
149 getp = 0;
|
|
|
150 }
|
|
|
151 if (putp == getp)
|
|
|
152 fill();
|
|
|
153
|
|
|
154 // getp is zero, always, here, and putp is the count
|
|
|
155 char *nl;
|
|
|
156 while ((nl = (char *)memchr (buf, '\n', putp)) == NULL && putp < SSBUFSZ)
|
|
|
157 if (fill () <= 0)
|
|
|
158 break;
|
|
|
159
|
|
|
160 if (nl)
|
|
|
161 {
|
|
|
162 getp = nl - buf + 1;
|
|
|
163 while ((*nl == '\n' || *nl == '\r') && nl >= buf)
|
|
|
164 *nl-- = 0;
|
|
|
165 }
|
|
|
166 else
|
|
|
167 {
|
|
|
168 getp = putp;
|
|
|
169 nl = buf + putp;
|
|
|
170 nl[1] = 0;
|
|
|
171 }
|
|
|
172
|
|
|
173 return buf;
|
|
|
174 }
|
|
|
175
|
|
|
176 #define MIN(a,b) ((a) < (b) ? (a) : (b))
|
|
|
177
|
|
|
178 int
|
|
|
179 SimpleSocket::read (char *ubuf, int ulen)
|
|
|
180 {
|
|
|
181 int n, rv=0;
|
|
|
182 if (putp > getp)
|
|
|
183 {
|
|
|
184 n = MIN (ulen, putp-getp);
|
|
|
185 memmove (ubuf, buf+getp, n);
|
|
|
186 getp += n;
|
|
|
187 ubuf += n;
|
|
|
188 ulen -= n;
|
|
|
189 rv += n;
|
|
|
190 }
|
|
|
191 while (ulen > 0)
|
|
|
192 {
|
|
|
193 n = recv (s, ubuf, ulen, 0);
|
|
|
194 if (n <= 0)
|
|
|
195 return rv;
|
|
|
196 ubuf += n;
|
|
|
197 ulen -= n;
|
|
|
198 rv += n;
|
|
|
199 }
|
|
|
200 return rv;
|
|
|
201 }
|
