1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#include <arpa/inet.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <lwipv6.h>
#define die(msg) perror(msg), exit(EXIT_FAILURE)
static void setnonblocking(int fd)
{
int opts = fcntl(fd, F_GETFL);
if (opts < 0)
die("fcntl(F_GETFL) error");
if (fcntl(fd, F_SETFL, opts|O_NONBLOCK) < 0)
die("fcntl(F_SETFL) error");
}
static void lwip_setnonblocking(int fd)
{
int opts = lwip_fcntl(fd, F_GETFL, 0);
if (opts < 0)
die("lwip_fcntl(F_GETFL) error");
if (lwip_fcntl(fd, F_SETFL, opts|O_NONBLOCK) < 0)
die("lwip_fcntl(F_SETFL) error");
}
int main(int argc, char **argv)
{
if (argc < 4)
die("insufficient arguments");
char *path = argv[1];
struct stack *stack = lwip_stack_new();
if (!stack)
die("lwip_stack_new() error");
struct netif *interface = lwip_vdeif_add(stack, path);
if (!interface) {
fprintf(stderr, "Couldn't connect to vde switch at %s\n", path);
exit(EXIT_FAILURE);
}
if (lwip_ifup(interface))
die("lwip_ifup() error");
struct ip_addr addr4, mask4;
IP64_ADDR(&addr4, 10,0,2,100);
IP64_MASKADDR(&mask4, 255,255,255,0);
if (lwip_add_addr(interface, &addr4, &mask4))
die("lwip_add_addr() error");
int fd = lwip_msocket(stack, AF_INET, SOCK_STREAM, 0);
if (fd < 0)
die("lwip_msocket() error");
struct sockaddr_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = inet_addr(argv[2]);
addr.sin_port = htons(atoi(argv[3]));
if (lwip_connect(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0)
die("lwip_connect() error");
setnonblocking(STDIN_FILENO);
lwip_setnonblocking(fd);
while (1) {
char buf[4096];
fd_set fds;
int r, w, wrote;
FD_ZERO(&fds);
FD_SET(STDIN_FILENO, &fds);
FD_SET(fd, &fds);
lwip_select(fd + 1, &fds, NULL, &fds, NULL);
while ((r = lwip_read(fd, buf, sizeof(buf))) > 0) {
wrote = 0;
while (wrote < r) {
w = write(STDOUT_FILENO, buf + wrote, r - wrote);
if (w < 0)
die("error writing to stdout");
wrote += w;
}
}
if (!r)
exit(EXIT_SUCCESS);
if (r < 0 && errno != EAGAIN)
die("error reading from socket");
while ((r = read(STDIN_FILENO, buf, sizeof(buf))) > 0) {
wrote = 0;
while (wrote < r) {
w = lwip_write(fd, buf + wrote, r - wrote);
if (w < 0)
die("error writing to socket");
wrote += w;
}
}
if (!r)
exit(EXIT_SUCCESS);
if (r < 0 && errno != EAGAIN)
die("error reading from stdin");
}
}
|