root / slirp / slirp.c @ 1b2b0af5
History | View | Annotate | Download (16.2 kB)
1 | f0cbd3ec | bellard | #include "slirp.h" |
---|---|---|---|
2 | f0cbd3ec | bellard | |
3 | f0cbd3ec | bellard | /* host address */
|
4 | f0cbd3ec | bellard | struct in_addr our_addr;
|
5 | f0cbd3ec | bellard | /* host dns address */
|
6 | f0cbd3ec | bellard | struct in_addr dns_addr;
|
7 | f0cbd3ec | bellard | /* host loopback address */
|
8 | f0cbd3ec | bellard | struct in_addr loopback_addr;
|
9 | f0cbd3ec | bellard | |
10 | f0cbd3ec | bellard | /* address for slirp virtual addresses */
|
11 | f0cbd3ec | bellard | struct in_addr special_addr;
|
12 | f0cbd3ec | bellard | |
13 | f0cbd3ec | bellard | const uint8_t special_ethaddr[6] = { |
14 | f0cbd3ec | bellard | 0x52, 0x54, 0x00, 0x12, 0x35, 0x00 |
15 | f0cbd3ec | bellard | }; |
16 | f0cbd3ec | bellard | |
17 | f0cbd3ec | bellard | uint8_t client_ethaddr[6];
|
18 | f0cbd3ec | bellard | |
19 | f0cbd3ec | bellard | int do_slowtimo;
|
20 | f0cbd3ec | bellard | int link_up;
|
21 | f0cbd3ec | bellard | struct timeval tt;
|
22 | f0cbd3ec | bellard | FILE *lfd; |
23 | a3d4af03 | bellard | struct ex_list *exec_list;
|
24 | f0cbd3ec | bellard | |
25 | f0cbd3ec | bellard | /* XXX: suppress those select globals */
|
26 | f0cbd3ec | bellard | fd_set *global_readfds, *global_writefds, *global_xfds; |
27 | f0cbd3ec | bellard | |
28 | 115defd1 | pbrook | const char slirp_hostname[33]; |
29 | 115defd1 | pbrook | |
30 | f0cbd3ec | bellard | #ifdef _WIN32
|
31 | f0cbd3ec | bellard | |
32 | f0cbd3ec | bellard | static int get_dns_addr(struct in_addr *pdns_addr) |
33 | f0cbd3ec | bellard | { |
34 | 379ff53d | bellard | FIXED_INFO *FixedInfo=NULL;
|
35 | 379ff53d | bellard | ULONG BufLen; |
36 | 379ff53d | bellard | DWORD ret; |
37 | 379ff53d | bellard | IP_ADDR_STRING *pIPAddr; |
38 | 379ff53d | bellard | struct in_addr tmp_addr;
|
39 | 379ff53d | bellard | |
40 | 379ff53d | bellard | FixedInfo = (FIXED_INFO *)GlobalAlloc(GPTR, sizeof(FIXED_INFO));
|
41 | 379ff53d | bellard | BufLen = sizeof(FIXED_INFO);
|
42 | 379ff53d | bellard | |
43 | 379ff53d | bellard | if (ERROR_BUFFER_OVERFLOW == GetNetworkParams(FixedInfo, &BufLen)) {
|
44 | 379ff53d | bellard | if (FixedInfo) {
|
45 | 379ff53d | bellard | GlobalFree(FixedInfo); |
46 | 379ff53d | bellard | FixedInfo = NULL;
|
47 | 379ff53d | bellard | } |
48 | 379ff53d | bellard | FixedInfo = GlobalAlloc(GPTR, BufLen); |
49 | 379ff53d | bellard | } |
50 | 379ff53d | bellard | |
51 | 379ff53d | bellard | if ((ret = GetNetworkParams(FixedInfo, &BufLen)) != ERROR_SUCCESS) {
|
52 | 379ff53d | bellard | printf("GetNetworkParams failed. ret = %08x\n", (u_int)ret );
|
53 | 379ff53d | bellard | if (FixedInfo) {
|
54 | 379ff53d | bellard | GlobalFree(FixedInfo); |
55 | 379ff53d | bellard | FixedInfo = NULL;
|
56 | 379ff53d | bellard | } |
57 | 379ff53d | bellard | return -1; |
58 | 379ff53d | bellard | } |
59 | 379ff53d | bellard | |
60 | 379ff53d | bellard | pIPAddr = &(FixedInfo->DnsServerList); |
61 | 379ff53d | bellard | inet_aton(pIPAddr->IpAddress.String, &tmp_addr); |
62 | 379ff53d | bellard | *pdns_addr = tmp_addr; |
63 | 379ff53d | bellard | #if 0
|
64 | 379ff53d | bellard | printf( "DNS Servers:\n" );
|
65 | 379ff53d | bellard | printf( "DNS Addr:%s\n", pIPAddr->IpAddress.String );
|
66 | 379ff53d | bellard |
|
67 | 379ff53d | bellard | pIPAddr = FixedInfo -> DnsServerList.Next;
|
68 | 379ff53d | bellard | while ( pIPAddr ) {
|
69 | 379ff53d | bellard | printf( "DNS Addr:%s\n", pIPAddr ->IpAddress.String );
|
70 | 379ff53d | bellard | pIPAddr = pIPAddr ->Next;
|
71 | 379ff53d | bellard | }
|
72 | 379ff53d | bellard | #endif
|
73 | 379ff53d | bellard | if (FixedInfo) {
|
74 | 379ff53d | bellard | GlobalFree(FixedInfo); |
75 | 379ff53d | bellard | FixedInfo = NULL;
|
76 | 379ff53d | bellard | } |
77 | 379ff53d | bellard | return 0; |
78 | f0cbd3ec | bellard | } |
79 | f0cbd3ec | bellard | |
80 | f0cbd3ec | bellard | #else
|
81 | f0cbd3ec | bellard | |
82 | f0cbd3ec | bellard | static int get_dns_addr(struct in_addr *pdns_addr) |
83 | f0cbd3ec | bellard | { |
84 | f0cbd3ec | bellard | char buff[512]; |
85 | f0cbd3ec | bellard | char buff2[256]; |
86 | f0cbd3ec | bellard | FILE *f; |
87 | f0cbd3ec | bellard | int found = 0; |
88 | f0cbd3ec | bellard | struct in_addr tmp_addr;
|
89 | f0cbd3ec | bellard | |
90 | f0cbd3ec | bellard | f = fopen("/etc/resolv.conf", "r"); |
91 | f0cbd3ec | bellard | if (!f)
|
92 | f0cbd3ec | bellard | return -1; |
93 | f0cbd3ec | bellard | |
94 | f0cbd3ec | bellard | lprint("IP address of your DNS(s): ");
|
95 | f0cbd3ec | bellard | while (fgets(buff, 512, f) != NULL) { |
96 | f0cbd3ec | bellard | if (sscanf(buff, "nameserver%*[ \t]%256s", buff2) == 1) { |
97 | f0cbd3ec | bellard | if (!inet_aton(buff2, &tmp_addr))
|
98 | f0cbd3ec | bellard | continue;
|
99 | f0cbd3ec | bellard | if (tmp_addr.s_addr == loopback_addr.s_addr)
|
100 | f0cbd3ec | bellard | tmp_addr = our_addr; |
101 | f0cbd3ec | bellard | /* If it's the first one, set it to dns_addr */
|
102 | f0cbd3ec | bellard | if (!found)
|
103 | f0cbd3ec | bellard | *pdns_addr = tmp_addr; |
104 | f0cbd3ec | bellard | else
|
105 | f0cbd3ec | bellard | lprint(", ");
|
106 | f0cbd3ec | bellard | if (++found > 3) { |
107 | f0cbd3ec | bellard | lprint("(more)");
|
108 | f0cbd3ec | bellard | break;
|
109 | f0cbd3ec | bellard | } else
|
110 | f0cbd3ec | bellard | lprint("%s", inet_ntoa(tmp_addr));
|
111 | f0cbd3ec | bellard | } |
112 | f0cbd3ec | bellard | } |
113 | 1d43a717 | bellard | fclose(f); |
114 | f0cbd3ec | bellard | if (!found)
|
115 | f0cbd3ec | bellard | return -1; |
116 | f0cbd3ec | bellard | return 0; |
117 | f0cbd3ec | bellard | } |
118 | f0cbd3ec | bellard | |
119 | f0cbd3ec | bellard | #endif
|
120 | f0cbd3ec | bellard | |
121 | 379ff53d | bellard | #ifdef _WIN32
|
122 | 379ff53d | bellard | void slirp_cleanup(void) |
123 | 379ff53d | bellard | { |
124 | 379ff53d | bellard | WSACleanup(); |
125 | 379ff53d | bellard | } |
126 | 379ff53d | bellard | #endif
|
127 | 379ff53d | bellard | |
128 | f0cbd3ec | bellard | void slirp_init(void) |
129 | f0cbd3ec | bellard | { |
130 | 512176db | bellard | // debug_init("/tmp/slirp.log", DEBUG_DEFAULT);
|
131 | 1d43a717 | bellard | |
132 | 379ff53d | bellard | #ifdef _WIN32
|
133 | 379ff53d | bellard | { |
134 | 379ff53d | bellard | WSADATA Data; |
135 | 379ff53d | bellard | WSAStartup(MAKEWORD(2,0), &Data); |
136 | 379ff53d | bellard | atexit(slirp_cleanup); |
137 | 379ff53d | bellard | } |
138 | 379ff53d | bellard | #endif
|
139 | 379ff53d | bellard | |
140 | f0cbd3ec | bellard | link_up = 1;
|
141 | f0cbd3ec | bellard | |
142 | f0cbd3ec | bellard | if_init(); |
143 | f0cbd3ec | bellard | ip_init(); |
144 | f0cbd3ec | bellard | |
145 | f0cbd3ec | bellard | /* Initialise mbufs *after* setting the MTU */
|
146 | f0cbd3ec | bellard | m_init(); |
147 | f0cbd3ec | bellard | |
148 | f0cbd3ec | bellard | /* set default addresses */
|
149 | f0cbd3ec | bellard | inet_aton("127.0.0.1", &loopback_addr);
|
150 | f0cbd3ec | bellard | |
151 | f0cbd3ec | bellard | if (get_dns_addr(&dns_addr) < 0) { |
152 | 0f8134bf | pbrook | dns_addr = loopback_addr; |
153 | 0f8134bf | pbrook | fprintf (stderr, "Warning: No DNS servers found\n");
|
154 | f0cbd3ec | bellard | } |
155 | f0cbd3ec | bellard | |
156 | f0cbd3ec | bellard | inet_aton(CTL_SPECIAL, &special_addr); |
157 | f4e15b4b | pbrook | getouraddr(); |
158 | f0cbd3ec | bellard | } |
159 | f0cbd3ec | bellard | |
160 | f0cbd3ec | bellard | #define CONN_CANFSEND(so) (((so)->so_state & (SS_FCANTSENDMORE|SS_ISFCONNECTED)) == SS_ISFCONNECTED)
|
161 | f0cbd3ec | bellard | #define CONN_CANFRCV(so) (((so)->so_state & (SS_FCANTRCVMORE|SS_ISFCONNECTED)) == SS_ISFCONNECTED)
|
162 | f0cbd3ec | bellard | #define UPD_NFDS(x) if (nfds < (x)) nfds = (x) |
163 | f0cbd3ec | bellard | |
164 | f0cbd3ec | bellard | /*
|
165 | f0cbd3ec | bellard | * curtime kept to an accuracy of 1ms
|
166 | f0cbd3ec | bellard | */
|
167 | 379ff53d | bellard | #ifdef _WIN32
|
168 | 379ff53d | bellard | static void updtime(void) |
169 | 379ff53d | bellard | { |
170 | 379ff53d | bellard | struct _timeb tb;
|
171 | 379ff53d | bellard | |
172 | 379ff53d | bellard | _ftime(&tb); |
173 | 379ff53d | bellard | curtime = (u_int)tb.time * (u_int)1000;
|
174 | 379ff53d | bellard | curtime += (u_int)tb.millitm; |
175 | 379ff53d | bellard | } |
176 | 379ff53d | bellard | #else
|
177 | f0cbd3ec | bellard | static void updtime(void) |
178 | f0cbd3ec | bellard | { |
179 | f0cbd3ec | bellard | gettimeofday(&tt, 0);
|
180 | f0cbd3ec | bellard | |
181 | f0cbd3ec | bellard | curtime = (u_int)tt.tv_sec * (u_int)1000;
|
182 | f0cbd3ec | bellard | curtime += (u_int)tt.tv_usec / (u_int)1000;
|
183 | f0cbd3ec | bellard | |
184 | f0cbd3ec | bellard | if ((tt.tv_usec % 1000) >= 500) |
185 | f0cbd3ec | bellard | curtime++; |
186 | f0cbd3ec | bellard | } |
187 | 379ff53d | bellard | #endif
|
188 | f0cbd3ec | bellard | |
189 | f0cbd3ec | bellard | void slirp_select_fill(int *pnfds, |
190 | f0cbd3ec | bellard | fd_set *readfds, fd_set *writefds, fd_set *xfds) |
191 | f0cbd3ec | bellard | { |
192 | f0cbd3ec | bellard | struct socket *so, *so_next;
|
193 | f0cbd3ec | bellard | struct timeval timeout;
|
194 | f0cbd3ec | bellard | int nfds;
|
195 | f0cbd3ec | bellard | int tmp_time;
|
196 | f0cbd3ec | bellard | |
197 | f0cbd3ec | bellard | /* fail safe */
|
198 | f0cbd3ec | bellard | global_readfds = NULL;
|
199 | f0cbd3ec | bellard | global_writefds = NULL;
|
200 | f0cbd3ec | bellard | global_xfds = NULL;
|
201 | f0cbd3ec | bellard | |
202 | f0cbd3ec | bellard | nfds = *pnfds; |
203 | f0cbd3ec | bellard | /*
|
204 | f0cbd3ec | bellard | * First, TCP sockets
|
205 | f0cbd3ec | bellard | */
|
206 | f0cbd3ec | bellard | do_slowtimo = 0;
|
207 | f0cbd3ec | bellard | if (link_up) {
|
208 | f0cbd3ec | bellard | /*
|
209 | f0cbd3ec | bellard | * *_slowtimo needs calling if there are IP fragments
|
210 | f0cbd3ec | bellard | * in the fragment queue, or there are TCP connections active
|
211 | f0cbd3ec | bellard | */
|
212 | f0cbd3ec | bellard | do_slowtimo = ((tcb.so_next != &tcb) || |
213 | f0cbd3ec | bellard | ((struct ipasfrag *)&ipq != (struct ipasfrag *)ipq.next)); |
214 | f0cbd3ec | bellard | |
215 | f0cbd3ec | bellard | for (so = tcb.so_next; so != &tcb; so = so_next) {
|
216 | f0cbd3ec | bellard | so_next = so->so_next; |
217 | f0cbd3ec | bellard | |
218 | f0cbd3ec | bellard | /*
|
219 | f0cbd3ec | bellard | * See if we need a tcp_fasttimo
|
220 | f0cbd3ec | bellard | */
|
221 | f0cbd3ec | bellard | if (time_fasttimo == 0 && so->so_tcpcb->t_flags & TF_DELACK) |
222 | f0cbd3ec | bellard | time_fasttimo = curtime; /* Flag when we want a fasttimo */
|
223 | f0cbd3ec | bellard | |
224 | f0cbd3ec | bellard | /*
|
225 | f0cbd3ec | bellard | * NOFDREF can include still connecting to local-host,
|
226 | f0cbd3ec | bellard | * newly socreated() sockets etc. Don't want to select these.
|
227 | f0cbd3ec | bellard | */
|
228 | f0cbd3ec | bellard | if (so->so_state & SS_NOFDREF || so->s == -1) |
229 | f0cbd3ec | bellard | continue;
|
230 | f0cbd3ec | bellard | |
231 | f0cbd3ec | bellard | /*
|
232 | f0cbd3ec | bellard | * Set for reading sockets which are accepting
|
233 | f0cbd3ec | bellard | */
|
234 | f0cbd3ec | bellard | if (so->so_state & SS_FACCEPTCONN) {
|
235 | f0cbd3ec | bellard | FD_SET(so->s, readfds); |
236 | f0cbd3ec | bellard | UPD_NFDS(so->s); |
237 | f0cbd3ec | bellard | continue;
|
238 | f0cbd3ec | bellard | } |
239 | f0cbd3ec | bellard | |
240 | f0cbd3ec | bellard | /*
|
241 | f0cbd3ec | bellard | * Set for writing sockets which are connecting
|
242 | f0cbd3ec | bellard | */
|
243 | f0cbd3ec | bellard | if (so->so_state & SS_ISFCONNECTING) {
|
244 | f0cbd3ec | bellard | FD_SET(so->s, writefds); |
245 | f0cbd3ec | bellard | UPD_NFDS(so->s); |
246 | f0cbd3ec | bellard | continue;
|
247 | f0cbd3ec | bellard | } |
248 | f0cbd3ec | bellard | |
249 | f0cbd3ec | bellard | /*
|
250 | f0cbd3ec | bellard | * Set for writing if we are connected, can send more, and
|
251 | f0cbd3ec | bellard | * we have something to send
|
252 | f0cbd3ec | bellard | */
|
253 | f0cbd3ec | bellard | if (CONN_CANFSEND(so) && so->so_rcv.sb_cc) {
|
254 | f0cbd3ec | bellard | FD_SET(so->s, writefds); |
255 | f0cbd3ec | bellard | UPD_NFDS(so->s); |
256 | f0cbd3ec | bellard | } |
257 | f0cbd3ec | bellard | |
258 | f0cbd3ec | bellard | /*
|
259 | f0cbd3ec | bellard | * Set for reading (and urgent data) if we are connected, can
|
260 | f0cbd3ec | bellard | * receive more, and we have room for it XXX /2 ?
|
261 | f0cbd3ec | bellard | */
|
262 | f0cbd3ec | bellard | if (CONN_CANFRCV(so) && (so->so_snd.sb_cc < (so->so_snd.sb_datalen/2))) { |
263 | f0cbd3ec | bellard | FD_SET(so->s, readfds); |
264 | f0cbd3ec | bellard | FD_SET(so->s, xfds); |
265 | f0cbd3ec | bellard | UPD_NFDS(so->s); |
266 | f0cbd3ec | bellard | } |
267 | f0cbd3ec | bellard | } |
268 | f0cbd3ec | bellard | |
269 | f0cbd3ec | bellard | /*
|
270 | f0cbd3ec | bellard | * UDP sockets
|
271 | f0cbd3ec | bellard | */
|
272 | f0cbd3ec | bellard | for (so = udb.so_next; so != &udb; so = so_next) {
|
273 | f0cbd3ec | bellard | so_next = so->so_next; |
274 | f0cbd3ec | bellard | |
275 | f0cbd3ec | bellard | /*
|
276 | f0cbd3ec | bellard | * See if it's timed out
|
277 | f0cbd3ec | bellard | */
|
278 | f0cbd3ec | bellard | if (so->so_expire) {
|
279 | f0cbd3ec | bellard | if (so->so_expire <= curtime) {
|
280 | f0cbd3ec | bellard | udp_detach(so); |
281 | f0cbd3ec | bellard | continue;
|
282 | f0cbd3ec | bellard | } else
|
283 | f0cbd3ec | bellard | do_slowtimo = 1; /* Let socket expire */ |
284 | f0cbd3ec | bellard | } |
285 | f0cbd3ec | bellard | |
286 | f0cbd3ec | bellard | /*
|
287 | f0cbd3ec | bellard | * When UDP packets are received from over the
|
288 | f0cbd3ec | bellard | * link, they're sendto()'d straight away, so
|
289 | f0cbd3ec | bellard | * no need for setting for writing
|
290 | f0cbd3ec | bellard | * Limit the number of packets queued by this session
|
291 | f0cbd3ec | bellard | * to 4. Note that even though we try and limit this
|
292 | f0cbd3ec | bellard | * to 4 packets, the session could have more queued
|
293 | f0cbd3ec | bellard | * if the packets needed to be fragmented
|
294 | f0cbd3ec | bellard | * (XXX <= 4 ?)
|
295 | f0cbd3ec | bellard | */
|
296 | f0cbd3ec | bellard | if ((so->so_state & SS_ISFCONNECTED) && so->so_queued <= 4) { |
297 | f0cbd3ec | bellard | FD_SET(so->s, readfds); |
298 | f0cbd3ec | bellard | UPD_NFDS(so->s); |
299 | f0cbd3ec | bellard | } |
300 | f0cbd3ec | bellard | } |
301 | f0cbd3ec | bellard | } |
302 | f0cbd3ec | bellard | |
303 | f0cbd3ec | bellard | /*
|
304 | f0cbd3ec | bellard | * Setup timeout to use minimum CPU usage, especially when idle
|
305 | f0cbd3ec | bellard | */
|
306 | f0cbd3ec | bellard | |
307 | f0cbd3ec | bellard | /*
|
308 | f0cbd3ec | bellard | * First, see the timeout needed by *timo
|
309 | f0cbd3ec | bellard | */
|
310 | f0cbd3ec | bellard | timeout.tv_sec = 0;
|
311 | f0cbd3ec | bellard | timeout.tv_usec = -1;
|
312 | f0cbd3ec | bellard | /*
|
313 | f0cbd3ec | bellard | * If a slowtimo is needed, set timeout to 500ms from the last
|
314 | f0cbd3ec | bellard | * slow timeout. If a fast timeout is needed, set timeout within
|
315 | f0cbd3ec | bellard | * 200ms of when it was requested.
|
316 | f0cbd3ec | bellard | */
|
317 | f0cbd3ec | bellard | if (do_slowtimo) {
|
318 | f0cbd3ec | bellard | /* XXX + 10000 because some select()'s aren't that accurate */
|
319 | f0cbd3ec | bellard | timeout.tv_usec = ((500 - (curtime - last_slowtimo)) * 1000) + 10000; |
320 | f0cbd3ec | bellard | if (timeout.tv_usec < 0) |
321 | f0cbd3ec | bellard | timeout.tv_usec = 0;
|
322 | f0cbd3ec | bellard | else if (timeout.tv_usec > 510000) |
323 | f0cbd3ec | bellard | timeout.tv_usec = 510000;
|
324 | f0cbd3ec | bellard | |
325 | f0cbd3ec | bellard | /* Can only fasttimo if we also slowtimo */
|
326 | f0cbd3ec | bellard | if (time_fasttimo) {
|
327 | f0cbd3ec | bellard | tmp_time = (200 - (curtime - time_fasttimo)) * 1000; |
328 | f0cbd3ec | bellard | if (tmp_time < 0) |
329 | f0cbd3ec | bellard | tmp_time = 0;
|
330 | f0cbd3ec | bellard | |
331 | f0cbd3ec | bellard | /* Choose the smallest of the 2 */
|
332 | f0cbd3ec | bellard | if (tmp_time < timeout.tv_usec)
|
333 | f0cbd3ec | bellard | timeout.tv_usec = (u_int)tmp_time; |
334 | f0cbd3ec | bellard | } |
335 | f0cbd3ec | bellard | } |
336 | f0cbd3ec | bellard | *pnfds = nfds; |
337 | f0cbd3ec | bellard | } |
338 | f0cbd3ec | bellard | |
339 | f0cbd3ec | bellard | void slirp_select_poll(fd_set *readfds, fd_set *writefds, fd_set *xfds)
|
340 | f0cbd3ec | bellard | { |
341 | f0cbd3ec | bellard | struct socket *so, *so_next;
|
342 | f0cbd3ec | bellard | int ret;
|
343 | f0cbd3ec | bellard | |
344 | f0cbd3ec | bellard | global_readfds = readfds; |
345 | f0cbd3ec | bellard | global_writefds = writefds; |
346 | f0cbd3ec | bellard | global_xfds = xfds; |
347 | f0cbd3ec | bellard | |
348 | f0cbd3ec | bellard | /* Update time */
|
349 | f0cbd3ec | bellard | updtime(); |
350 | f0cbd3ec | bellard | |
351 | f0cbd3ec | bellard | /*
|
352 | f0cbd3ec | bellard | * See if anything has timed out
|
353 | f0cbd3ec | bellard | */
|
354 | f0cbd3ec | bellard | if (link_up) {
|
355 | df5f8956 | bellard | if (time_fasttimo && ((curtime - time_fasttimo) >= 2)) { |
356 | f0cbd3ec | bellard | tcp_fasttimo(); |
357 | f0cbd3ec | bellard | time_fasttimo = 0;
|
358 | f0cbd3ec | bellard | } |
359 | f0cbd3ec | bellard | if (do_slowtimo && ((curtime - last_slowtimo) >= 499)) { |
360 | f0cbd3ec | bellard | ip_slowtimo(); |
361 | f0cbd3ec | bellard | tcp_slowtimo(); |
362 | f0cbd3ec | bellard | last_slowtimo = curtime; |
363 | f0cbd3ec | bellard | } |
364 | f0cbd3ec | bellard | } |
365 | f0cbd3ec | bellard | |
366 | f0cbd3ec | bellard | /*
|
367 | f0cbd3ec | bellard | * Check sockets
|
368 | f0cbd3ec | bellard | */
|
369 | f0cbd3ec | bellard | if (link_up) {
|
370 | f0cbd3ec | bellard | /*
|
371 | f0cbd3ec | bellard | * Check TCP sockets
|
372 | f0cbd3ec | bellard | */
|
373 | f0cbd3ec | bellard | for (so = tcb.so_next; so != &tcb; so = so_next) {
|
374 | f0cbd3ec | bellard | so_next = so->so_next; |
375 | f0cbd3ec | bellard | |
376 | f0cbd3ec | bellard | /*
|
377 | f0cbd3ec | bellard | * FD_ISSET is meaningless on these sockets
|
378 | f0cbd3ec | bellard | * (and they can crash the program)
|
379 | f0cbd3ec | bellard | */
|
380 | f0cbd3ec | bellard | if (so->so_state & SS_NOFDREF || so->s == -1) |
381 | f0cbd3ec | bellard | continue;
|
382 | f0cbd3ec | bellard | |
383 | f0cbd3ec | bellard | /*
|
384 | f0cbd3ec | bellard | * Check for URG data
|
385 | f0cbd3ec | bellard | * This will soread as well, so no need to
|
386 | f0cbd3ec | bellard | * test for readfds below if this succeeds
|
387 | f0cbd3ec | bellard | */
|
388 | f0cbd3ec | bellard | if (FD_ISSET(so->s, xfds))
|
389 | f0cbd3ec | bellard | sorecvoob(so); |
390 | f0cbd3ec | bellard | /*
|
391 | f0cbd3ec | bellard | * Check sockets for reading
|
392 | f0cbd3ec | bellard | */
|
393 | f0cbd3ec | bellard | else if (FD_ISSET(so->s, readfds)) { |
394 | f0cbd3ec | bellard | /*
|
395 | f0cbd3ec | bellard | * Check for incoming connections
|
396 | f0cbd3ec | bellard | */
|
397 | f0cbd3ec | bellard | if (so->so_state & SS_FACCEPTCONN) {
|
398 | f0cbd3ec | bellard | tcp_connect(so); |
399 | f0cbd3ec | bellard | continue;
|
400 | f0cbd3ec | bellard | } /* else */
|
401 | f0cbd3ec | bellard | ret = soread(so); |
402 | f0cbd3ec | bellard | |
403 | f0cbd3ec | bellard | /* Output it if we read something */
|
404 | f0cbd3ec | bellard | if (ret > 0) |
405 | f0cbd3ec | bellard | tcp_output(sototcpcb(so)); |
406 | f0cbd3ec | bellard | } |
407 | f0cbd3ec | bellard | |
408 | f0cbd3ec | bellard | /*
|
409 | f0cbd3ec | bellard | * Check sockets for writing
|
410 | f0cbd3ec | bellard | */
|
411 | f0cbd3ec | bellard | if (FD_ISSET(so->s, writefds)) {
|
412 | f0cbd3ec | bellard | /*
|
413 | f0cbd3ec | bellard | * Check for non-blocking, still-connecting sockets
|
414 | f0cbd3ec | bellard | */
|
415 | f0cbd3ec | bellard | if (so->so_state & SS_ISFCONNECTING) {
|
416 | f0cbd3ec | bellard | /* Connected */
|
417 | f0cbd3ec | bellard | so->so_state &= ~SS_ISFCONNECTING; |
418 | f0cbd3ec | bellard | |
419 | 02d2c54c | bellard | ret = send(so->s, &ret, 0, 0); |
420 | f0cbd3ec | bellard | if (ret < 0) { |
421 | f0cbd3ec | bellard | /* XXXXX Must fix, zero bytes is a NOP */
|
422 | f0cbd3ec | bellard | if (errno == EAGAIN || errno == EWOULDBLOCK ||
|
423 | f0cbd3ec | bellard | errno == EINPROGRESS || errno == ENOTCONN) |
424 | f0cbd3ec | bellard | continue;
|
425 | f0cbd3ec | bellard | |
426 | f0cbd3ec | bellard | /* else failed */
|
427 | f0cbd3ec | bellard | so->so_state = SS_NOFDREF; |
428 | f0cbd3ec | bellard | } |
429 | f0cbd3ec | bellard | /* else so->so_state &= ~SS_ISFCONNECTING; */
|
430 | f0cbd3ec | bellard | |
431 | f0cbd3ec | bellard | /*
|
432 | f0cbd3ec | bellard | * Continue tcp_input
|
433 | f0cbd3ec | bellard | */
|
434 | f0cbd3ec | bellard | tcp_input((struct mbuf *)NULL, sizeof(struct ip), so); |
435 | f0cbd3ec | bellard | /* continue; */
|
436 | f0cbd3ec | bellard | } else
|
437 | f0cbd3ec | bellard | ret = sowrite(so); |
438 | f0cbd3ec | bellard | /*
|
439 | f0cbd3ec | bellard | * XXXXX If we wrote something (a lot), there
|
440 | f0cbd3ec | bellard | * could be a need for a window update.
|
441 | f0cbd3ec | bellard | * In the worst case, the remote will send
|
442 | f0cbd3ec | bellard | * a window probe to get things going again
|
443 | f0cbd3ec | bellard | */
|
444 | f0cbd3ec | bellard | } |
445 | f0cbd3ec | bellard | |
446 | f0cbd3ec | bellard | /*
|
447 | f0cbd3ec | bellard | * Probe a still-connecting, non-blocking socket
|
448 | f0cbd3ec | bellard | * to check if it's still alive
|
449 | f0cbd3ec | bellard | */
|
450 | f0cbd3ec | bellard | #ifdef PROBE_CONN
|
451 | f0cbd3ec | bellard | if (so->so_state & SS_ISFCONNECTING) {
|
452 | 02d2c54c | bellard | ret = recv(so->s, (char *)&ret, 0,0); |
453 | f0cbd3ec | bellard | |
454 | f0cbd3ec | bellard | if (ret < 0) { |
455 | f0cbd3ec | bellard | /* XXX */
|
456 | f0cbd3ec | bellard | if (errno == EAGAIN || errno == EWOULDBLOCK ||
|
457 | f0cbd3ec | bellard | errno == EINPROGRESS || errno == ENOTCONN) |
458 | f0cbd3ec | bellard | continue; /* Still connecting, continue */ |
459 | f0cbd3ec | bellard | |
460 | f0cbd3ec | bellard | /* else failed */
|
461 | f0cbd3ec | bellard | so->so_state = SS_NOFDREF; |
462 | f0cbd3ec | bellard | |
463 | f0cbd3ec | bellard | /* tcp_input will take care of it */
|
464 | f0cbd3ec | bellard | } else {
|
465 | 02d2c54c | bellard | ret = send(so->s, &ret, 0,0); |
466 | f0cbd3ec | bellard | if (ret < 0) { |
467 | f0cbd3ec | bellard | /* XXX */
|
468 | f0cbd3ec | bellard | if (errno == EAGAIN || errno == EWOULDBLOCK ||
|
469 | f0cbd3ec | bellard | errno == EINPROGRESS || errno == ENOTCONN) |
470 | f0cbd3ec | bellard | continue;
|
471 | f0cbd3ec | bellard | /* else failed */
|
472 | f0cbd3ec | bellard | so->so_state = SS_NOFDREF; |
473 | f0cbd3ec | bellard | } else
|
474 | f0cbd3ec | bellard | so->so_state &= ~SS_ISFCONNECTING; |
475 | f0cbd3ec | bellard | |
476 | f0cbd3ec | bellard | } |
477 | f0cbd3ec | bellard | tcp_input((struct mbuf *)NULL, sizeof(struct ip),so); |
478 | f0cbd3ec | bellard | } /* SS_ISFCONNECTING */
|
479 | f0cbd3ec | bellard | #endif
|
480 | f0cbd3ec | bellard | } |
481 | f0cbd3ec | bellard | |
482 | f0cbd3ec | bellard | /*
|
483 | f0cbd3ec | bellard | * Now UDP sockets.
|
484 | f0cbd3ec | bellard | * Incoming packets are sent straight away, they're not buffered.
|
485 | f0cbd3ec | bellard | * Incoming UDP data isn't buffered either.
|
486 | f0cbd3ec | bellard | */
|
487 | f0cbd3ec | bellard | for (so = udb.so_next; so != &udb; so = so_next) {
|
488 | f0cbd3ec | bellard | so_next = so->so_next; |
489 | f0cbd3ec | bellard | |
490 | f0cbd3ec | bellard | if (so->s != -1 && FD_ISSET(so->s, readfds)) { |
491 | f0cbd3ec | bellard | sorecvfrom(so); |
492 | f0cbd3ec | bellard | } |
493 | f0cbd3ec | bellard | } |
494 | f0cbd3ec | bellard | } |
495 | f0cbd3ec | bellard | |
496 | f0cbd3ec | bellard | /*
|
497 | f0cbd3ec | bellard | * See if we can start outputting
|
498 | f0cbd3ec | bellard | */
|
499 | f0cbd3ec | bellard | if (if_queued && link_up)
|
500 | f0cbd3ec | bellard | if_start(); |
501 | 02d2c54c | bellard | |
502 | 02d2c54c | bellard | /* clear global file descriptor sets.
|
503 | 02d2c54c | bellard | * these reside on the stack in vl.c
|
504 | 02d2c54c | bellard | * so they're unusable if we're not in
|
505 | 02d2c54c | bellard | * slirp_select_fill or slirp_select_poll.
|
506 | 02d2c54c | bellard | */
|
507 | 02d2c54c | bellard | global_readfds = NULL;
|
508 | 02d2c54c | bellard | global_writefds = NULL;
|
509 | 02d2c54c | bellard | global_xfds = NULL;
|
510 | f0cbd3ec | bellard | } |
511 | f0cbd3ec | bellard | |
512 | f0cbd3ec | bellard | #define ETH_ALEN 6 |
513 | f0cbd3ec | bellard | #define ETH_HLEN 14 |
514 | f0cbd3ec | bellard | |
515 | f0cbd3ec | bellard | #define ETH_P_IP 0x0800 /* Internet Protocol packet */ |
516 | f0cbd3ec | bellard | #define ETH_P_ARP 0x0806 /* Address Resolution packet */ |
517 | f0cbd3ec | bellard | |
518 | f0cbd3ec | bellard | #define ARPOP_REQUEST 1 /* ARP request */ |
519 | f0cbd3ec | bellard | #define ARPOP_REPLY 2 /* ARP reply */ |
520 | f0cbd3ec | bellard | |
521 | f0cbd3ec | bellard | struct ethhdr
|
522 | f0cbd3ec | bellard | { |
523 | f0cbd3ec | bellard | unsigned char h_dest[ETH_ALEN]; /* destination eth addr */ |
524 | f0cbd3ec | bellard | unsigned char h_source[ETH_ALEN]; /* source ether addr */ |
525 | f0cbd3ec | bellard | unsigned short h_proto; /* packet type ID field */ |
526 | f0cbd3ec | bellard | }; |
527 | f0cbd3ec | bellard | |
528 | f0cbd3ec | bellard | struct arphdr
|
529 | f0cbd3ec | bellard | { |
530 | f0cbd3ec | bellard | unsigned short ar_hrd; /* format of hardware address */ |
531 | f0cbd3ec | bellard | unsigned short ar_pro; /* format of protocol address */ |
532 | f0cbd3ec | bellard | unsigned char ar_hln; /* length of hardware address */ |
533 | f0cbd3ec | bellard | unsigned char ar_pln; /* length of protocol address */ |
534 | f0cbd3ec | bellard | unsigned short ar_op; /* ARP opcode (command) */ |
535 | f0cbd3ec | bellard | |
536 | f0cbd3ec | bellard | /*
|
537 | f0cbd3ec | bellard | * Ethernet looks like this : This bit is variable sized however...
|
538 | f0cbd3ec | bellard | */
|
539 | f0cbd3ec | bellard | unsigned char ar_sha[ETH_ALEN]; /* sender hardware address */ |
540 | f0cbd3ec | bellard | unsigned char ar_sip[4]; /* sender IP address */ |
541 | f0cbd3ec | bellard | unsigned char ar_tha[ETH_ALEN]; /* target hardware address */ |
542 | f0cbd3ec | bellard | unsigned char ar_tip[4]; /* target IP address */ |
543 | f0cbd3ec | bellard | }; |
544 | f0cbd3ec | bellard | |
545 | f0cbd3ec | bellard | void arp_input(const uint8_t *pkt, int pkt_len) |
546 | f0cbd3ec | bellard | { |
547 | f0cbd3ec | bellard | struct ethhdr *eh = (struct ethhdr *)pkt; |
548 | f0cbd3ec | bellard | struct arphdr *ah = (struct arphdr *)(pkt + ETH_HLEN); |
549 | f0cbd3ec | bellard | uint8_t arp_reply[ETH_HLEN + sizeof(struct arphdr)]; |
550 | f0cbd3ec | bellard | struct ethhdr *reh = (struct ethhdr *)arp_reply; |
551 | f0cbd3ec | bellard | struct arphdr *rah = (struct arphdr *)(arp_reply + ETH_HLEN); |
552 | f0cbd3ec | bellard | int ar_op;
|
553 | a3d4af03 | bellard | struct ex_list *ex_ptr;
|
554 | f0cbd3ec | bellard | |
555 | f0cbd3ec | bellard | ar_op = ntohs(ah->ar_op); |
556 | f0cbd3ec | bellard | switch(ar_op) {
|
557 | f0cbd3ec | bellard | case ARPOP_REQUEST:
|
558 | a3d4af03 | bellard | if (!memcmp(ah->ar_tip, &special_addr, 3)) { |
559 | a3d4af03 | bellard | if (ah->ar_tip[3] == CTL_DNS || ah->ar_tip[3] == CTL_ALIAS) |
560 | a3d4af03 | bellard | goto arp_ok;
|
561 | a3d4af03 | bellard | for (ex_ptr = exec_list; ex_ptr; ex_ptr = ex_ptr->ex_next) {
|
562 | a3d4af03 | bellard | if (ex_ptr->ex_addr == ah->ar_tip[3]) |
563 | a3d4af03 | bellard | goto arp_ok;
|
564 | a3d4af03 | bellard | } |
565 | a3d4af03 | bellard | return;
|
566 | a3d4af03 | bellard | arp_ok:
|
567 | f0cbd3ec | bellard | /* XXX: make an ARP request to have the client address */
|
568 | f0cbd3ec | bellard | memcpy(client_ethaddr, eh->h_source, ETH_ALEN); |
569 | f0cbd3ec | bellard | |
570 | f0cbd3ec | bellard | /* ARP request for alias/dns mac address */
|
571 | f0cbd3ec | bellard | memcpy(reh->h_dest, pkt + ETH_ALEN, ETH_ALEN); |
572 | f0cbd3ec | bellard | memcpy(reh->h_source, special_ethaddr, ETH_ALEN - 1);
|
573 | f0cbd3ec | bellard | reh->h_source[5] = ah->ar_tip[3]; |
574 | f0cbd3ec | bellard | reh->h_proto = htons(ETH_P_ARP); |
575 | f0cbd3ec | bellard | |
576 | f0cbd3ec | bellard | rah->ar_hrd = htons(1);
|
577 | f0cbd3ec | bellard | rah->ar_pro = htons(ETH_P_IP); |
578 | f0cbd3ec | bellard | rah->ar_hln = ETH_ALEN; |
579 | f0cbd3ec | bellard | rah->ar_pln = 4;
|
580 | f0cbd3ec | bellard | rah->ar_op = htons(ARPOP_REPLY); |
581 | f0cbd3ec | bellard | memcpy(rah->ar_sha, reh->h_source, ETH_ALEN); |
582 | f0cbd3ec | bellard | memcpy(rah->ar_sip, ah->ar_tip, 4);
|
583 | f0cbd3ec | bellard | memcpy(rah->ar_tha, ah->ar_sha, ETH_ALEN); |
584 | f0cbd3ec | bellard | memcpy(rah->ar_tip, ah->ar_sip, 4);
|
585 | f0cbd3ec | bellard | slirp_output(arp_reply, sizeof(arp_reply));
|
586 | f0cbd3ec | bellard | } |
587 | f0cbd3ec | bellard | break;
|
588 | f0cbd3ec | bellard | default:
|
589 | f0cbd3ec | bellard | break;
|
590 | f0cbd3ec | bellard | } |
591 | f0cbd3ec | bellard | } |
592 | f0cbd3ec | bellard | |
593 | f0cbd3ec | bellard | void slirp_input(const uint8_t *pkt, int pkt_len) |
594 | f0cbd3ec | bellard | { |
595 | f0cbd3ec | bellard | struct mbuf *m;
|
596 | f0cbd3ec | bellard | int proto;
|
597 | f0cbd3ec | bellard | |
598 | f0cbd3ec | bellard | if (pkt_len < ETH_HLEN)
|
599 | f0cbd3ec | bellard | return;
|
600 | f0cbd3ec | bellard | |
601 | f0cbd3ec | bellard | proto = ntohs(*(uint16_t *)(pkt + 12));
|
602 | f0cbd3ec | bellard | switch(proto) {
|
603 | f0cbd3ec | bellard | case ETH_P_ARP:
|
604 | f0cbd3ec | bellard | arp_input(pkt, pkt_len); |
605 | f0cbd3ec | bellard | break;
|
606 | f0cbd3ec | bellard | case ETH_P_IP:
|
607 | f0cbd3ec | bellard | m = m_get(); |
608 | f0cbd3ec | bellard | if (!m)
|
609 | f0cbd3ec | bellard | return;
|
610 | f0cbd3ec | bellard | m->m_len = pkt_len; |
611 | f0cbd3ec | bellard | memcpy(m->m_data, pkt, pkt_len); |
612 | f0cbd3ec | bellard | |
613 | f0cbd3ec | bellard | m->m_data += ETH_HLEN; |
614 | f0cbd3ec | bellard | m->m_len -= ETH_HLEN; |
615 | f0cbd3ec | bellard | |
616 | f0cbd3ec | bellard | ip_input(m); |
617 | f0cbd3ec | bellard | break;
|
618 | f0cbd3ec | bellard | default:
|
619 | f0cbd3ec | bellard | break;
|
620 | f0cbd3ec | bellard | } |
621 | f0cbd3ec | bellard | } |
622 | f0cbd3ec | bellard | |
623 | f0cbd3ec | bellard | /* output the IP packet to the ethernet device */
|
624 | f0cbd3ec | bellard | void if_encap(const uint8_t *ip_data, int ip_data_len) |
625 | f0cbd3ec | bellard | { |
626 | f0cbd3ec | bellard | uint8_t buf[1600];
|
627 | f0cbd3ec | bellard | struct ethhdr *eh = (struct ethhdr *)buf; |
628 | f0cbd3ec | bellard | |
629 | f0cbd3ec | bellard | if (ip_data_len + ETH_HLEN > sizeof(buf)) |
630 | f0cbd3ec | bellard | return;
|
631 | f0cbd3ec | bellard | |
632 | f0cbd3ec | bellard | memcpy(eh->h_dest, client_ethaddr, ETH_ALEN); |
633 | f0cbd3ec | bellard | memcpy(eh->h_source, special_ethaddr, ETH_ALEN - 1);
|
634 | a3d4af03 | bellard | /* XXX: not correct */
|
635 | f0cbd3ec | bellard | eh->h_source[5] = CTL_ALIAS;
|
636 | f0cbd3ec | bellard | eh->h_proto = htons(ETH_P_IP); |
637 | f0cbd3ec | bellard | memcpy(buf + sizeof(struct ethhdr), ip_data, ip_data_len); |
638 | f0cbd3ec | bellard | slirp_output(buf, ip_data_len + ETH_HLEN); |
639 | f0cbd3ec | bellard | } |
640 | 9bf05444 | bellard | |
641 | 9bf05444 | bellard | int slirp_redir(int is_udp, int host_port, |
642 | 9bf05444 | bellard | struct in_addr guest_addr, int guest_port) |
643 | 9bf05444 | bellard | { |
644 | 9bf05444 | bellard | if (is_udp) {
|
645 | 9bf05444 | bellard | if (!udp_listen(htons(host_port), guest_addr.s_addr,
|
646 | 9bf05444 | bellard | htons(guest_port), 0))
|
647 | 9bf05444 | bellard | return -1; |
648 | 9bf05444 | bellard | } else {
|
649 | 9bf05444 | bellard | if (!solisten(htons(host_port), guest_addr.s_addr,
|
650 | 9bf05444 | bellard | htons(guest_port), 0))
|
651 | 9bf05444 | bellard | return -1; |
652 | 9bf05444 | bellard | } |
653 | 9bf05444 | bellard | return 0; |
654 | 9bf05444 | bellard | } |
655 | a3d4af03 | bellard | |
656 | a3d4af03 | bellard | int slirp_add_exec(int do_pty, const char *args, int addr_low_byte, |
657 | a3d4af03 | bellard | int guest_port)
|
658 | a3d4af03 | bellard | { |
659 | a3d4af03 | bellard | return add_exec(&exec_list, do_pty, (char *)args, |
660 | a3d4af03 | bellard | addr_low_byte, htons(guest_port)); |
661 | a3d4af03 | bellard | } |