file server & client

/*
 * file_server.c
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>

#define BUFSIZE 30
#define FILESIZE 256

void error_handling(char *message);

int main(int argc, char **argv)
{
 int serv_sd;
 int clnt_sd;
 int fd;
 char buf[BUFSIZE];
 char filebuf[FILESIZE];
 char check;
 
 struct sockaddr_in serv_addr;
 struct sockaddr_in clnt_addr;
 int clnt_addr_size;
 int len;
 
 if(argc!=2){
  printf("Usage : %s <port>\n", argv[0]);
  exit(1);
 }
 
 serv_sd=socket(PF_INET, SOCK_STREAM, 0);  
 if(serv_sd == -1)
  error_handling("socket() error");
 
 memset(&serv_addr, 0, sizeof(serv_addr));
 serv_addr.sin_family=AF_INET;
 serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
 serv_addr.sin_port=htons(atoi(argv[1]));
 
 if( bind(serv_sd, (struct sockaddr*) &serv_addr, sizeof(serv_addr))==-1 )
  error_handling("bind() error");
 
 if( listen(serv_sd, 5)==-1 )
  error_handling("listen() error");
 
 clnt_addr_size=sizeof(clnt_addr);   
 clnt_sd=accept(serv_sd, (struct sockaddr*)&clnt_addr,&clnt_addr_size);
 if(clnt_sd==-1)
  error_handling("accept() error");
 
 /* 클라이언트가 연결되면 성공 메시지 전송 */
 write(clnt_sd, "Server Connect Success!\n\n", 25);
 
 /* 파일명을 클라이언트로 부터 받아 open 및 에러 출력 */
 if((len=read(clnt_sd, filebuf, FILESIZE)) != 0)
 {
  filebuf[len-1]=0;
  fd = open(filebuf, O_RDONLY);
  if(fd == -1)
   error_handling("File open error");
  else
   write(clnt_sd, "ok", 2); /* 파일 open 성공시 메시지 전송 */
   
 }
 else
  error_handling("File open error");

 /* 파일 전송 클라이언트에서 원하는지 파악 */  
 read(clnt_sd, buf, BUFSIZE);
 check = buf[0];

 /* 전송시 */
 if(check == 'y')
 {
  while((len=read(fd, buf, BUFSIZE)) != 0)
   write(clnt_sd, buf, len);
 }
 else
  error_handling("bye~");
 
 
 /* 데이터 전송후 소켓의 일부(전송영역)를 닫음 */
 if(shutdown(clnt_sd, SHUT_WR) ==-1)
  error_handling("shutdown error");
 
 /* 인사의 메시지를 수신 함 */
 len = read(clnt_sd, buf, BUFSIZE);
 write(1, buf, len);
 
 close(fd);
 close(clnt_sd);
 return 0;
}

void error_handling(char *message)
{
 fputs(message, stderr);
 fputc('\n', stderr);
 exit(1);
}

==========================================================================

/*
 * file_client.c
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>

#define BUFSIZE 30
#define FILESIZE 256

void error_handling(char *message);

int main(int argc, char **argv)
{
 int fd;
 int sd;
 
 char buf[BUFSIZE];
 char filebuf[FILESIZE];
 char check;

 int len;
 struct sockaddr_in serv_addr;

 if(argc!=3){
  printf("Usage : %s <IP> <port>\n", argv[0]);
  exit(1);
 }
 
 /* 서버 접속을 위한 소켓 생성 */
 sd=socket(PF_INET, SOCK_STREAM, 0);  
 if(sd == -1)
  error_handling("socket() error");
 
 memset(&serv_addr, 0, sizeof(serv_addr));
 serv_addr.sin_family=AF_INET;
 serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
 serv_addr.sin_port=htons(atoi(argv[2]));

 if( connect(sd, (struct sockaddr*)&serv_addr, sizeof(serv_addr))==-1 )
  error_handling("connect() error!");

 /* 연결 성공 메시지 수신 */
 if((len=read(sd, buf, BUFSIZE)) != 0 )
  write(1, buf, len);

 /* 파일 이름 입력 받아 전송 */
 printf("Send file name : ");
 fgets(filebuf, FILESIZE, stdin);
 write(sd, filebuf, strlen(filebuf));

 /* 파일 유무 확인 메시지 수신 */
 if((len=read(sd, buf, BUFSIZE)) != 0)
  buf[len]=0;
 
 if(strcmp(buf, "ok") != 0)
  error_handling("File open error");
 else
  printf("File open success!!\n\n");  

 /* 파일 open 성공시 파일 전송 할지를 결정 */
 printf("File transmission? (y/n) : ");
 fgets(buf, BUFSIZE, stdin);
 write(sd, buf, strlen(buf));
 check = buf[0];

 /* 전송일 경우 */
 if(check == 'y')
 {
  fd = open("receive.txt", O_WRONLY|O_CREAT|O_TRUNC);
  if(fd == -1)
   error_handling("File open error");
 
  /* 데이터를 전송 받아서 파일에 저장한다 */
  while((len=read(sd, buf, BUFSIZE)) != 0 )
   write(fd, buf, len);  
 }
 else
  error_handling("bye~");
 
 /* 전송해 준것에 대한 감사의 메시지 전달 */
 write(sd, "Transmission Success! bye~\n", 27);
 printf("Transmission Success! bye~\n");
 close(fd);
 close(sd);
 return 0;
}

void error_handling(char *message)
{
 fputs(message, stderr);
 fputc('\n', stderr);
 exit(1);
}