SOCKET_PRGM--Lin/Win条件编译版本

//pub.h
#ifndef _H_PUB_H_
#define _H_PUB_H_

#ifdef WIN
#include <WinSock2.h>
#else
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <errno.h>
#include <string.h>
#define SOCKET int
#endif
void getfilename(const char * filename,char *name);
int send_work(const char *hostname,int port,const char *filename);
SOCKET socket_accept (SOCKET listen_st);
SOCKET socket_create(int port);
SOCKET socket_connect(const char * hostname,int port);
SOCKET init_socket();
int recv_work(int port);

#endif

---

//pub.c
#ifdef WIN
#include <WinSock2.h>
#else
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <errno.h>
#include <string.h>
#define SOCKET int
#endif

#include <stdio.h>
#include "tcp_pub.h"

#define BUFSIZE 262144//256k

void getfilename(const char * filename,char *name)
{
    int len = strlen(filename);
    int i;
    for (i = (len -1);i >=0;i--)
    {
        if ((filename[i] == '\\') || (filename[i]) == '/')
        {
            break;
        }
    }
    strcpy(name,&filename[i + 1]);
    return ;
}

SOCKET init_socket()//初始化socket
{
//如果是windows,执行如下代码
#ifdef WIN
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    wVersionRequested = MAKEWORD(1,1);//获取版本
    err = WSAStartup(wVersionRequested,&wsaData);
    if (err != 0)
    {
        return -1;
    }

    if (LOBYTE(wsaData.wVersion) != 1 || HIBYTE(wsaData.wVersion) != 1)
    {
        WSACleanup();
        return -1;
    }
#endif
//如果是linux,直接返回
    return 0;

}

SOCKET socket_connect(const char * hostname,int port)//连接到指定的主机和端口号
{
    if (init_socket() == -1)
        return 0;

    SOCKET st = socket(AF_INET,SOCK_STREAM,0);
    if (st == 0)
        return 0;
    struct sockaddr_in addr;
    memset(&addr,0,sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = inet_addr(hostname);
    if (connect(st,(struct sockaddr *)&addr,sizeof(addr)) == -1)
    {
        printf("connect to %s:%d failed %s\n",hostname,port,strerror(errno));
        return 0;//连接失败
    } else
    {
        return st;//连接成功,返回socket描述符
    }
}


SOCKET socket_create(int port)//在port指定端口上建立server端的socket
{
    if (init_socket() == -1)
        return 0;

    SOCKET st = socket(AF_INET,SOCK_STREAM,0);
    if (st == 0)
        return 0;
#ifdef WIN
    const char on =0;
#else
    int on = 0;
#endif

    if (setsockopt(st, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1)
    {
        printf("setsockopt failed %s\n", strerror(errno));
        return EXIT_FAILURE;
    }

    struct sockaddr_in addr;//定义一个IP地址结构
    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;//将addr结构的属性定位为TCP/IP地址
    addr.sin_port = htons(port);//将本地字节顺序转化为网络字节顺序。
    addr.sin_addr.s_addr = htonl(INADDR_ANY);

    //将IP与server程序绑定
    if (bind(st, (struct sockaddr *) &addr, sizeof(addr)) == -1)
    {
        printf("bind failed %s\n", strerror(errno));
        return EXIT_FAILURE;
    }

    //server端开始listen，
    if (listen(st, 20) == -1)
    {
        printf("listen failed %s\n", strerror(errno));
        return EXIT_FAILURE;
    }

    printf("listen %d success\n",port);
    return st;


}


SOCKET socket_accept (SOCKET listen_st)
{

    struct sockaddr_in client_addr;
#ifdef WIN
    int len = 0;
#else
    unsigned int len = 1;
#endif

    len = sizeof(client_addr);
    memset(&client_addr, 0,sizeof(client_addr));
    SOCKET client_st = accept(listen_st,(struct sockaddr *)&client_addr,&len);
    if (client_st == -1)
    {

        printf("accept failed %s\n",strerror(errno));
        return 0;
    }
    else
    {
        printf("accept by %s\n",inet_ntoa(client_addr.sin_addr));
        return client_st;
    }


}


int send_work(const char *hostname,int port,const char *filename)
{
    SOCKET st = socket_connect(hostname,port);//链接到指定的IP和port
    if (st == 0)
        return 0;

    FILE *fd = fopen(filename,"rb");//只读方式打开
    if (fd == NULL)
    {
        printf("open %s failed %s\n",filename,strerror(errno));
        return 0;
    }

    char *buf = malloc(BUFSIZE);
    memset(buf,0,BUFSIZE);
    getfilename(filename,buf);//从完整路径中解析文件名
    size_t rc = send(st,buf,strlen(buf),0);//客户端第一次发送的是文件名
    if (rc <= 0)
    {
        if (rc < 0 )
            printf("send failed %s\n",strerror(errno));
        else
            printf("socket disconnect\n");
    } else
    {
        memset(buf,0,BUFSIZE);
        if ((rc = recv(st, buf, BUFSIZE, 0))<= 0)//接受来自server的回复
        {
            if (rc < 0)
                printf("recv failed %s\n",strerror(errno));
            else
                printf("socket disconnect\n");

        } else
        {
            if (strncmp(buf,"OK",2) ==0)//如果收到服务端的回复,代表服务端认可,可以发送文件了
            {
                while (1)
                {
                    memset(buf,0,BUFSIZE);
                    rc = fread(buf,1,BUFSIZE,fd);//循环读取,直到文件尾部
                    if (rc <= 0)
                    {
                        if (rc < 0)
                            printf("send failed %s\n",strerror(errno));
                        break;

                    } else
                    {
                        rc = send(st,buf,rc,0);
                        if (rc <= 0)//发送失败,代表TCP连接出错,循环break
                        {
                            if (rc < 0)
                                printf("send failed %s\n",strerror(errno));
                            else
                                printf("socket disconnect\n");
                            break;
                        }
                    }
                }
            }
        }
    }

    fclose(fd);
    free(buf);

#ifdef WIN
    closesocket(st);
    WSACleanup();
#else
    close(st);
#endif
    return 1;

}





int recv_work(int port)
{

    SOCKET listen_st = socket_create(port);
    if (listen_st == 0)
        return 0;
    SOCKET st = socket_accept(listen_st);
    if (st == 0)
        return 0;

    char *buf = malloc(BUFSIZE);
    FILE *fd = NULL;

    memset(buf, 0,BUFSIZE);
    size_t rc = recv(st,buf,BUFSIZE,0);
    if (rc <= 0)
    {
        if (rc < 0)
            printf("recv failed %s\n",strerror(errno));
        else
            printf("socket disconnen\n");

    } else
    {
        printf("receiving %s\n",buf);

        fd = fopen(buf,"wb");
        if (fd == NULL)
        {
            printf("open %s failed %s\n",buf,strerror(errno));
        } else
        {
            memset(buf,0,BUFSIZE);
            strcpy(buf,"OK");
            rc = send(st,buf,strlen(buf),0);
            if (rc <= 0)
            {
                if (rc < 0)
                    printf("send failed %s",strerror(errno));
                else
                    printf("socket disconnect\n");
            }

            while (1)
            {
                memset(buf,0,BUFSIZE);
                rc = recv(st,buf,BUFSIZE,0);
                if (rc <=0 )
                {
                    if (rc < 0)
                        printf("recv failed %s\n",strerror(errno));
                    else
                        printf("socket disconnect\n");
                    break;

                } else
                {
                    fwrite(buf,1,rc,fd);
                }
            }
        }
    }


    if (fd)
        fclose(fd);
    free(buf);
#ifdef WIN
    closesocket(st);
    closesocket(listen_st);
    WSACleanup();
#else
    close(st);
    close(listen_st);
#endif

    return 1;


}

---

//tcp_client.c
#include <stdio.h>
#include <stdlib.h>
#include "tcp_pub.h"
int main(int arg, char *args[])
{
    if (arg < 4)
    {
        printf("Usage: %s  <host-ip> <host-port> <trans-filename>\n",args[0]);
        return EXIT_FAILURE;
    }
    int iport = atoi(args[2]);
    if (iport == 0)
    {
        printf("port %d is invalid\n",iport);
        return EXIT_FAILURE;
    }
    printf("%s send begin\n",args[3]);
    if (send_work(args[1],iport,args[3]) == 1)
        printf("%s send success\n",args[3]);
    else
        printf("%s send fail\n",args[3]);

    return EXIT_SUCCESS;

}

---

//tcp_server.c
#include <stdio.h>
#include <stdlib.h>
#include "tcp_pub.h"
int main(int arg, char *args[])
{
    if (arg < 2)
    {
        printf("Usage: %s <port> \n",args[0]);
        return EXIT_FAILURE;
    }
    int iport = atoi(args[1]);
    if (iport == 0)
    {
        printf("port %d is invalid\n",iport);
        return EXIT_FAILURE;
    }
    printf("%s send begin\n",args[3]);
    if (recv_work(iport) == 1)
        printf("%s recv success\n",args[3]);
    else
        printf("%s recv fail\n",args[3]);

    return EXIT_SUCCESS;

}

---

//Linxu_Makefile
.SUFFIXES:.c.o

CC=gcc
SERVERSRCS=tcp_server.c\
            tcp_pub.c
CLIENTSRCS=tcp_client.c\
            tcp_pub.c

SERVEROBJS=$(SERVERSRCS:.c=.o)
CLIENTOBJS=$(CLIENTSRCS:.c=.o)
SERVEREXEC=Tcp_server_lin
CLIENTEXEC=Tcp_client_lin

all:$(SERVEROBJS) $(CLIENTOBJS)
    $(CC)  -o $(CLIENTEXEC) $(CLIENTOBJS)
    $(CC)  -o $(SERVEREXEC) $(SERVEROBJS)
    @echo '----------OK-----------'

.c.o:
    $(CC) -Wall  -o $@ -c $<

clean:
    rm -f $(SERVEROBJS)
    rm -f $(CLIENTOBJS)
    rm -f core*

---

//Windows_Makefile
.SUFFIXES:.c.o

CC=gcc
SERVERSRCS=tcp_server.c\
            tcp_pub.c
CLIENTSRCS=tcp_client.c\
            tcp_pub.c

SERVEROBJS=$(SERVERSRCS:.c=.o)
CLIENTOBJS=$(CLIENTSRCS:.c=.o)
SERVEREXEC=Tcp_server_win.exe
CLIENTEXEC=Tcp_client_win.exe

all:$(SERVEROBJS) $(CLIENTOBJS)
    $(CC) -static -o $(CLIENTEXEC) $(CLIENTOBJS) -lWs2_32
    $(CC) -static -o $(SERVEREXEC) $(SERVEROBJS) -lWs2_32
    @echo '----------OK-----------'

.c.o:
    $(CC) -Wall -DWIN -o $@ -c $<

clean:
    rm -f $(SERVEROBJS)
    rm -f $(CLIENTOBJS)
    rm -f core*