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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/*
** client.c -- a stream socket client demo
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <ctype.h>
#include <arpa/inet.h>
#include "protocol.h"
#define MAXDATASIZE 100 // max number of bytes we can get at once
#define MAX_READ_SIZE 1024
#define LINE_MAX 1024
// get sockaddr, IPv4 or IPv6:
void *get_in_addr(struct sockaddr *sa)
{
if (sa->sa_family == AF_INET) {
return &(((struct sockaddr_in*)sa)->sin_addr);
}
return &(((struct sockaddr_in6*)sa)->sin6_addr);
}
int main(int argc, char *argv[])
{
int sockfd, numbytesrecv, numbytessent, recvbytes;
// char buf[MAXDATASIZE];
struct addrinfo hints, *servinfo, *p;
int rv;
char s[INET6_ADDRSTRLEN];
if (argc != 4) {
fprintf(stderr,"<server IP> <server port> <listener port>\n");
exit(1);
}
memset(&hints, 0, sizeof hints);
hints.ai_family = AF_INET; // only IPv4
hints.ai_socktype = SOCK_STREAM;
char* tcpPort = argv[2]; // port we use to connect to server's tcp stream
char* udpPort = argv[3]; // port we use to connect to server's udp info and command
if ((rv = getaddrinfo(argv[1], tcpPort, &hints, &servinfo)) != 0) {
fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(rv));
return 1;
}
// loop through all the results and connect to the first we can
for(p = servinfo; p != NULL; p = p->ai_next) {
if ((sockfd = socket(p->ai_family, p->ai_socktype,
p->ai_protocol)) == -1) {
perror("client: socket");
continue;
}
if (connect(sockfd, p->ai_addr, p->ai_addrlen) == -1) {
close(sockfd);
perror("client: connect");
continue;
}
break;
}
if (p == NULL) {
fprintf(stderr, "client: failed to connect\n");
return 2;
}
inet_ntop(p->ai_family, get_in_addr((struct sockaddr *)p->ai_addr),
s, sizeof s);
// printf("client: connecting to %s\n", s);
freeaddrinfo(servinfo); // all done with this structure
struct Welcome msg;
// recv the message, check for errors too
if ((recvbytes = recv(sockfd, (char*)&msg, sizeof(struct Welcome), 0)) == -1) {
perror("recv");
exit(1);
}
msg.numStations = ntohs(msg.numStations);
printf("Welcome to Snowcast! The server has %d stations.\n", msg.numStations);
struct Hello hello;
hello.commandType = 0;
// convert updPort to an int
int udpPortInt = atoi(udpPort);
hello.udpPort = htons(udpPortInt);
if ((numbytessent = send(sockfd, &hello, sizeof(struct Hello), 0)) == -1) {
perror("send");
exit(1);
}
char input[LINE_MAX];
printf("Enter a number to change to it's station. Click q to end stream.\n");
while (1==1) {
char *line = fgets(input, LINE_MAX, stdin);
if (line == NULL) {
continue;
} else if (strncmp("q\n", input, LINE_MAX) == 0) {
printf("Exiting.\n");
break;
} else {
// convert input to an int
int inputInt = (uint16_t)atoi(input);
struct SetStation setStation;
setStation.commandType = 1;
setStation.stationNumber = htons(inputInt);
if ((numbytessent = send(sockfd, &setStation, sizeof(struct SetStation), 0)) == -1) {
perror("send");
exit(1);
}
}
}
return 0;
}
|
