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
|
/*
** 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.h"
#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);
// 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);
}
// print the amount of bytes sent
printf("client: sent %d bytes on a hello call \n", numbytessent);
pthread_t reply_thread;
pthread_create(&reply_thread, NULL, reply_thread_routine, (void*)sockfd);
char input[LINE_MAX];
printf("Enter a number to change to it's station. Click q to end stream.\n");
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);
if ((numbytessent = send(sockfd, &setStation, sizeof(struct SetStation), 0)) == -1) {
perror("send");
exit(1);
}
}
}
return 0;
}
void *reply_thread_routine(void* args) {
int sockfd = (int)args;
int recvbytes;
char buf[MAX_READ_SIZE];
while (1) {
// recv the message, check for errors too
if ((recvbytes = recv(sockfd, &buf, MAX_READ_SIZE, 0)) == -1) {
perror("recv");
exit(1);
}
buf[recvbytes] = '\0';
printf("client: received %d bytes on a reply call \n", recvbytes);
// print the two first field of the call
printf("client: replyType: %d, stringSize: %d\n", buf[0], buf[1]);
// print the while buffer by char
for (int i = 0; i < recvbytes; i++) {
printf("%c ", buf[i]);
}
struct Reply reply;
memcpy(&reply, buf, 2);
// print out the fields of reply on one line
printf("\nclient: replyType: %d, stringSize: %d\n", reply.replyType, reply.stringSize);
// print the size of reply
if (reply.replyType == 3) {
printf("client: received an announce message\n");
char *song_name = malloc(reply.stringSize);
// printf(sizeof(struct Reply));
memcpy(song_name, buf + 2, reply.stringSize);
printf("client: song name: %s\n", song_name);
free(song_name);
continue;
} else if (reply.replyType == 4) {
printf("client: received an invalid command message\n");
continue;
}
printf("client: received an unknown message\n");
memset(buf, 0, MAX_READ_SIZE);
}
}
|
