summaryrefslogtreecommitdiff
path: root/Year_3/TSDWL/ex_20161004/socket.c
blob: 7d018c8ab1b3c6852383fdd989f578f1bccbb3b2 (plain)
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
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <pthread.h>
#include <string.h>
#include <arpa/inet.h>

char*
check_string(char buffer[1024], int n)
{
    char* string = "yes";
    int i;

    for(i = 0; i < n; ++i) {
        if (buffer[i] != 'V' && buffer[i] != 'F') {
            strcpy(string, "no");
            break;
        }
    }

    return string;
}

int
main(int argc, char* argv[])
{
    int sockfd, clientfd;
    struct sockaddr_in addr;
    socklen_t addrlen;
    char buffer[1024];
    char* result;
    int n;

    if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket");
        exit(1);
    }

    addr.sin_addr.s_addr = INADDR_ANY;
    addr.sin_port = htons(3333);
    addr.sin_family = AF_INET;

    addrlen = sizeof(addr);

    if (bind(sockfd, (struct sockaddr*) &addr, addrlen) == -1) {
        perror("bind");
        exit(1);
    }

    if (listen(sockfd, 1) == -1) {
        perror("listen");
        exit(1);
    }

    while (1) {
        if ((clientfd = accept(sockfd, (struct sockaddr*) &addr, &addrlen)) == -1) {
            perror("accept");
            exit(1);
        }

        if ((n = read(clientfd, buffer, sizeof(buffer))) == -1) {
            perror("read");
            goto clientclose;
        }

        result = check_string(buffer, n);
        printf("%s\n", result);

clientclose:
        close(clientfd);
    }

    close(sockfd);
    return 0;
}