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
134
135
136
|
/*
** 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 <arpa/inet.h>
#include "protocol.h"
#define MAXDATASIZE 100 // max number of bytes we can get at once
// 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 snowcast_message), 0)) == -1) {
perror("recv");
exit(1);
}
msg.numStations = ntohs(msg.numStations);
// print the num bytes received
// printf("client: received '%s'\n", msg);
// printf("size of '%d'\n", recvbytes);
printf("Welcome to Snowcast! The server has %d stations.\n", msg.numStations);
// printf("type %d", msg.replyType);
// close(sockfd);
// if ((numbytesrecv = recv(sockfd, buf, MAXDATASIZE-1, 0)) == -1) {
// perror("recv");
// exit(1);
// }
// buf[numbytesrecv] = '\0';
// printf("client: received '%s'\n",buf);
// make a struct for the message, number is thennubmer of stations
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);
}
// make new struct for the incoming message
// buf[numbytesrecv] = '\0';
// char bufff[MAXDATASIZE];
// if ((recvbytes = recv(sockfd, bufff, MAXDATASIZE, 0)) == -1)
// {
// perror("recv");
// exit(1);
// }
// bufff[recvbytes] = '\0';
// // print the num bytes received
// printf("client: received '%d'\n", recvbytes);
// //printf("Welcome to Snowcast! The server has %d stations.\n", ntohl(msg.number));
// printf("%s", bufff);
close(sockfd);
return 0;
}
|
