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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
|
/*
** 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 <pthread.h>
#include <arpa/inet.h>
#include "protocol.c"
#define MAXDATASIZE 100 // max number of bytes we can get at once
#define MAX_READ_SIZE 1024
#define LINE_MAX 1024
void *reply_thread_routine(void *args);
// 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);
freeaddrinfo(servinfo); // all done with this structure
pthread_t reply_thread;
pthread_create(&reply_thread, NULL, reply_thread_routine, (void*)sockfd);
usleep(1000);
// sleep(1);
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);
}
// CONSIDER: could recieve the welcome message here
char input[LINE_MAX];
printf("Enter a number to change to it's station. Click q to end stream.\n");
fflush(stdout);
while (1) {
char *line = fgets(input, LINE_MAX, stdin);
if (line == NULL) {
continue;
} else if (strncmp("q\n", input, LINE_MAX) == 0) {
// end code if type in q
printf("Exiting.\n");
break;
} else {
// convert input to an int
int inputInt = atoi(input);
// printf("Changing to station %d.\n", inputInt);
// send the command to change the station
struct SetStation setStation;
setStation.commandType = 1;
setStation.stationNumber = htons(inputInt);
int bytes_to_send = sizeof(struct SetStation);
if (send_all(sockfd, &setStation, &bytes_to_send) == -1) {
perror("send_all");
exit(1);
}
}
}
return 0;
}
void *reply_thread_routine(void* args) {
int sockfd = (int)args;
// int recvbytes;
while (1) {
// recv the first byte of the message to get it's type
uint8_t reply_type = -1;
// print size of utin8
if (recv(sockfd, &reply_type, 1, 0) == -1) {
perror("recv");
exit(1);
}
if (reply_type == 2) { // we have a welcome message
// recv the message, check for errors too
int16_t num_stations = -1;
int bytes_to_read = sizeof(uint16_t);
if (recv_all(sockfd, &num_stations, &bytes_to_read) == -1) {
perror("recv_all");
exit(1);
}
num_stations = ntohs(num_stations);
printf("Welcome to Snowcast! The server has %d stations.\n", num_stations);
continue;
}
if (reply_type == 3) { // we have an announce message
// get the string size
u_int8_t string_size = -1;
if (recv(sockfd, &string_size, 1, 0) == -1) {
perror("recv");
exit(1);
}
char *song_name = malloc(string_size);
if(song_name == NULL) { perror("malloc in song name"); }
int bytes_to_read = string_size;
if (recv_all(sockfd, song_name, &bytes_to_read) == -1) {
perror("recv_all");
exit(1);
}
printf("New song announced: %s\n", song_name);
free(song_name);
continue;
} else if (reply_type == 4) { // we have an invalid command message
// get the string size
u_int8_t string_size = -1;
if (recv(sockfd, &string_size, 1, 0) == -1) {
perror("recv");
exit(1);
}
char *message = malloc(string_size);
if(message == NULL) { perror("malloc in message"); }
int bytes_to_read = string_size;
if (recv_all(sockfd, message, &bytes_to_read) == -1) {
perror("recv_all");
exit(1);
}
printf("Invalid protocol: %s. Exiting.\n", message);
free(message);
close(sockfd);
exit(1);
}
printf("Lost connection to server. Exiting.\n");
close(sockfd);
exit(1);
}
}
|
