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
|
#include <sys/types.h>
#include <sys/socket.h>
#include <inttypes.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include <netdb.h>
#include <unistd.h>
#include "client-tcp.h"
struct stream* stream_client_tcp(const char *args) {
struct stream *s = NULL;
int fd = -1;
struct sockaddr_in sa;
struct hostent *h;
size_t l;
uint16_t port = 23;
char hn[256];
l = strcspn(args, ":");
if (l > sizeof(hn)-1)
l = sizeof(hn)-1;
strncpy(hn, args, l);
hn[l] = 0;
if (args[l] == ':')
port = atoi(args+l+1);
s = malloc(sizeof(struct stream));
assert(s);
memset(s, 0, sizeof(struct stream));
if ((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
fprintf(stderr, "socket(): %s\n", strerror(errno));
goto fail;
}
memset(&sa, 0, sizeof(sa));
sa.sin_family = AF_INET;
sa.sin_port = htons(port);
if (!(h = gethostbyname(hn))) {
fprintf(stderr, "Unknown host '%s'.\n", hn);
goto fail;
}
sa.sin_addr = *(struct in_addr *) h->h_addr;
if (connect(fd, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
fprintf(stderr, "connect(): %s\n", strerror(errno));
goto fail;
}
s->input_fd = s->output_fd = fd;
return s;
fail:
free(s);
if (fd >= 0)
close(fd);
return NULL;
}
|
