123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189 |
- /* -*- 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 <unistd.h>
- #include <stdlib.h>
- #include <string.h>
- #include <stdio.h>
- #include <fcntl.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <sys/un.h>
- #include <sys/stat.h>
- #include <sys/wait.h>
- #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_un address;
- if ((create_socket = socket(AF_UNIX,SOCK_STREAM,
- 0)) > 0)
- printf("The socket was created\n");
- address.sun_family = AF_UNIX;
- memcpy(address.sun_path,"u",10);
- 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);
- printf("The client is connected...\n");
- if (do_fork) {
- if (fork() > 0) {
- asm volatile ("int $3");
- close(new_socket);
- wait(NULL);
- return 0;
- }
- }
- if ((fd = open(fname,O_RDONLY,0)) < 0) {
- perror("File Open Failed");
- close(new_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_un address;
- if (mode == PARALLEL) {
- close(pipefds[1]);
- char byte = 0;
- read(pipefds[0], &byte, 1);
- }
- if ((create_socket = socket(AF_UNIX,SOCK_STREAM,0)) >= 0)
- printf("The socket was created\n");
- address.sun_family = AF_UNIX;
- memcpy(address.sun_path,"u",10);
- if (connect(create_socket,(struct sockaddr *)&address,
- sizeof(address)) == 0)
- printf("The connection was accepted with the server\n");
- else {
- printf("The connection was not accepted with the server\n");
- exit(0);
- }
- if (do_fork) {
- if (fork() > 0) {
- close(create_socket);
- wait(NULL);
- return 0;
- }
- }
- printf("The contents of file are...\n\n");
- while((count=recv(create_socket,buffer,bufsize,0))>0)
- write(1,buffer,count);
- printf("\nEOF\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, "unix");
- strcat(fnamebuf, ".c");
- fname = fnamebuf;
- 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();
- }
- return 0;
- }
|