0703 socket服务器多线程处理客户请求

/*************************************************************************
    > File Name: my_socket.h
    > Author:Monica
    > Mail:liling222@126.com 
    > Created Time: Thu 03 Jul 2014 10:40:54 AM CST
 ************************************************************************/

#ifndef __MY_SOCKET_H__
#define __MY_SOCKET_H__

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "msg.h"

int listenfd_init(char* ip, char* port);


#endif

/*************************************************************************
    > File Name: server.c
    > Author: Monica
    > Mail:liling222@126.com 
    > Created Time: Thu 03 Jul 2014 10:02:48 AM CST
 ************************************************************************/

#include  "my_socket.h"

int listenfd_init(char* ip, char* port){

    int fd_server;
    int reuse = 1 ;
    struct sockaddr_in server_addr;
    fd_server = socket(AF_INET,SOCK_STREAM, 0);
    if(fd_server == -1)
    {
        perror("socket");
        exit(-1);
    }
    if(setsockopt(fd_server,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(int)) == -1)
    {
        perror("setsockopt");
        close(fd_server);
        exit(-1);
    }
    memset(&server_addr, 0 , sizeof(server_addr)) ;
    server_addr.sin_family = AF_INET ;
    server_addr.sin_port = htons(atoi(port));
    server_addr.sin_addr.s_addr = inet_addr(ip);
    if( bind(fd_server,(struct sockaddr*)&server_addr,sizeof(server_addr)) == -1)
    {
        perror("bind");
        close(fd_server);
        exit(-1);
    }
    if(listen(fd_server,5) == -1)
    {
        perror("listen");
        close(fd_server);
        exit(-1);
    }
    return fd_server;
}

#include "my_socket.h"

void  * thread_handle(void* fd_client);
int main(int argc , char* argv[])//exe ip port
{
    int fd_server,fd_client;
    struct sockaddr_in server_addr,client_addr ;
    int addrlen = sizeof(client_addr);
    int tid;
    if(argc != 3)
    {
        printf("usge : exe ip port");
        exit(-1);
    }
    fd_server = listenfd_init(argv[1], argv[2]);
    while(fd_client = accept(fd_server,(struct sockaddr*)&client_addr,&addrlen)){
        printf("a client connect IP:%s\tPORT:%d\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
        pthread_create(&tid, NULL, thread_handle, (void *)fd_client);
    }
    close(fd_server);
    return 0;
}
void*  thread_handle(void* fd_client){
    pthread_detach(pthread_self());
    MSG recv_msg, snd_msg;
    int read_n;
    int fd_file;
    int fd_cli = (int)fd_client;
    memset(&recv_msg, 0, sizeof(recv_msg));
    recv(fd_cli, &recv_msg.s_len, MSG_LEN , 0) ;
    recv(fd_cli, recv_msg.s_buf, recv_msg.s_len, 0) ;
    printf("recv msg :%s\n", recv_msg.s_buf);
    fd_file = open(recv_msg.s_buf, O_RDONLY);
    while(memset(&snd_msg, 0, sizeof(snd_msg)), (read_n =  read(fd_file, snd_msg.s_buf, MSG_SIZE)) > 0)
    {
        snd_msg.s_len = read_n ;
        send(fd_cli, &snd_msg, snd_msg.s_len + MSG_LEN ,0);
    }
    snd_msg.s_len = 0 ;
    send(fd_cli, &snd_msg, snd_msg.s_len + MSG_LEN ,0);
    close(fd_file);
    close(fd_cli);
}

/*************************************************************************
    > File Name: client.c
    > Author:Monica
    > Mail:liling222@126.com 
    > Created Time: Thu 03 Jul 2014 10:46:20 AM CST
 ************************************************************************/

#include "my_socket.h"
int main(int argc, char* argv[]){
    int fd_client;
    struct sockaddr_in  server_addr;
    MSG recv_msg, snd_msg;
    int fd_file;
    int recv_len = 0 ;
    char filename[128];
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr  = inet_addr(argv[1]);
    server_addr.sin_port   = htons(atoi(argv[2]));
    if((fd_client = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        perror("socket");
        exit(-1);
    }
    if(connect(fd_client, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1){
        perror("connect");
        close(fd_client);
        exit(-1);
    }
    printf("inputs filename:");
    scanf("%s", snd_msg.s_buf);
    snd_msg.s_len = strlen(snd_msg.s_buf);
    send(fd_client,&snd_msg, snd_msg.s_len+MSG_LEN, 0 );
    sprintf(filename, "download_%s", snd_msg.s_buf);
    fd_file = open(filename, O_WRONLY|O_CREAT, 0666);
    while(1){
        printf("begin recv\n");
        memset(&recv_msg, 0, sizeof(recv_msg));
        recv(fd_client, &recv_msg.s_len, MSG_LEN, 0);
        if(recv_msg.s_len >0){
            recv(fd_client, recv_msg.s_buf, MSG_SIZE, 0);
            write(fd_file, recv_msg.s_buf, strlen(recv_msg.s_buf));
            recv_len += recv_msg.s_len;
            printf("download...%d\n", recv_len);
        }
        else{
            break;
        }
    }
    close(fd_file);
    close(fd_client);
    return 0;
}