/* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */ /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */ /* copied from http://www.daniweb.com/software-development/c/threads/179814 */ #include #include #include #include #include #include #include #include #include #include #include #define SRV_IP "127.0.0.1" #define PORT 9930 #define BUFLEN 512 #define NPACK 10 const char * fname; enum { SINGLE, PARALLEL } mode = PARALLEL; int do_fork = 0; int pipefds[2]; int server(void) { int conn, create_socket, new_socket, addrlen, fd; int bufsize = 1024; char * buffer = malloc(bufsize); struct sockaddr_in address; if ((create_socket = socket(AF_INET, SOCK_STREAM, 0)) > 0) printf("The socket was created\n"); address.sin_family = AF_INET; inet_pton(AF_INET, SRV_IP, &(address.sin_addr)); address.sin_port = htons(PORT); if (bind(create_socket, (struct sockaddr *) &address, sizeof(address)) < 0) { perror("bind"); close(create_socket); exit(-1); } if (listen(create_socket, 3) < 0) { perror("listen"); close(create_socket); exit(-1); } if (mode == PARALLEL) { close(pipefds[0]); char byte = 0; write(pipefds[1], &byte, 1); } addrlen = sizeof(address); new_socket = accept(create_socket, (struct sockaddr *) &address, &addrlen); if (new_socket < 0) { perror("accept"); close(create_socket); exit(-1); } close(create_socket); if (do_fork) { if (fork() > 0) { close(new_socket); wait(NULL); return 0; } } char buff[18] = {'\0'}; if (inet_ntop(AF_INET, &address.sin_addr, buff, 18) <= 0) perror("address conversion"); printf("The client %s is connected...\n", buff); if ((fd = open(fname, O_RDONLY, 0)) < 0) { perror("file open failed"); close(new_socket); close(create_socket); exit(-1); } while ((conn = read(fd, buffer, bufsize)) > 0) sendto(new_socket, buffer, conn, 0, 0, 0); printf("Request completed\n"); close(new_socket); if (do_fork) exit(0); return 0; } int client(void) { int count, create_socket; int bufsize = 1024; char * buffer = malloc(bufsize); struct sockaddr_in address; if (mode == PARALLEL) { close(pipefds[1]); char byte = 0; read(pipefds[0], &byte, 1); } if ((create_socket = socket(AF_INET, SOCK_STREAM, 0)) >= 0) printf("The Socket was created\n"); address.sin_family = AF_INET; inet_pton(AF_INET, SRV_IP, &address.sin_addr); address.sin_port = htons(PORT); if (connect(create_socket, (struct sockaddr *) &address, sizeof(address)) == 0) { printf("The connection was accepted with the server\n"); } else { perror("accept"); exit(0); } if (do_fork) { if (fork() > 0) { close(create_socket); wait(NULL); return 0; } } printf("Content:\n"); while((count=recv(create_socket, buffer, bufsize, 0))>0) write(1, buffer, count); printf("EOF\n"); buffer[0] = 0; close(create_socket); if (do_fork) exit(0); return 0; } int main(int argc, char ** argv) { char fnamebuf[40]; strcpy(fnamebuf, argv[0]); strcat(fnamebuf, ".c"); fname = fnamebuf; setvbuf(stdout, NULL, _IONBF, 0); if (argc > 1) { if (strcmp(argv[1], "client") == 0) { mode = SINGLE; client(); return 0; } if (strcmp(argv[1], "server") == 0) { mode = SINGLE; server(); return 0; } if (strcmp(argv[1], "fork") == 0) { do_fork = 1; goto old; } } else { old: pipe(pipefds); int pid = fork(); if (pid == 0) client(); else { server(); waitpid(pid, NULL, -1); } } return 0; }