diff options
Diffstat (limited to 'client.c')
| -rw-r--r-- | client.c | 240 |
1 files changed, 240 insertions, 0 deletions
diff --git a/client.c b/client.c new file mode 100644 index 0000000..f39f839 --- /dev/null +++ b/client.c @@ -0,0 +1,240 @@ + +/* +** 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, "usage: <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 + + pthread_t reply_thread; + pthread_create(&reply_thread, NULL, reply_thread_routine, (void*)sockfd); + + usleep(100); + + struct timeval tv; + tv.tv_sec = 0; + tv.tv_usec = TIMEOUT; + if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) { + perror("setsockopt"); + } + + 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]; + 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 + tv.tv_sec = 0; + tv.tv_usec = TIMEOUT; + if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) { + perror("setsockopt"); + } + 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 + int reply_type; + // print size of utin8 + if (recv(sockfd, &reply_type, 1, 0) == -1) { + perror("recv"); + exit(1); + } + + fprintf(stderr, "reply_type: %d\n", reply_type); + + if (reply_type == 2) { // we have a welcome message + // recv the message, check for errors too + uint16_t buf_num_stations; + int bytes_to_read = sizeof(uint16_t); + if (recv_all(sockfd, &buf_num_stations, &bytes_to_read) == -1) { + perror("recv_all"); + exit(1); + } + uint16_t num_stations = ntohs(buf_num_stations); + fprintf(stdout, "Welcome to Snowcast! The server has %u stations.\n", num_stations); + fprintf(stderr, "Welcome to Snowcast! The server has %u stations.\n", num_stations); + // printf("Click q to end stream.\n"); + + struct timeval no_tv; + no_tv.tv_sec = 0; + no_tv.tv_usec = 0; + if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &no_tv, sizeof(no_tv)) < 0) { + perror("setsockopt"); + } + continue; + } + + if (reply_type == 3) { // we have an announce message + // get the string size + int string_size; + 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"); } + + if (recv_all(sockfd, song_name, &string_size) == -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 + int string_size; + if (recv(sockfd, &string_size, 1, 0) == -1) { + perror("recv"); + exit(1); + } + + printf("string size: %d\n", string_size); + + char *message = malloc(string_size); + if(message == NULL) { perror("malloc in message"); } + if (recv_all(sockfd, message, &string_size) == -1) { + perror("recv_all"); + exit(1); + } + struct timeval no_tv; + no_tv.tv_sec = 0; + no_tv.tv_usec = 0; + if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &no_tv, sizeof(no_tv)) < 0) { + perror("setsockopt"); + } + printf("Invalid protocol: %s. Exiting.\n", message); + fprintf(stderr, "Invalid protocol: %s. Exiting.\n", message); + free(message); + close(sockfd); + exit(1); + } + + printf("Lost connection to server. Exiting."); + close(sockfd); + exit(1); + } +}
\ No newline at end of file |