C语言Socket编程（计算机网络作业）

最近我计算机网络课程要做作业了，没办法跟着老师一步一步的写C语言的代码，使用的计算就是Socket通信发送消息；代码实现的功能很简单，客户端向服务器端发送消息，服务器端接收客户端发来的消息，并且输出显示到屏幕上；代码中的每个socket方法的返回值都应该做判断出错情况，例如：创建SOCKET的创建的时候s句柄（或对象），需要判断返回值是否为INVALID_SOCKET，以及socket的所有操作的返回值是否为SOCKET_ERROR，以保证程序的稳定性，这里只是测试代码。

老师是录制的视频讲解的，其他其中使用的到的方法要求我们自己去查相关的MSDN资料库，如果有不懂的，就自己上网查。

服务器端接收代码：

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>

void main()
{
WSAData wsd;
WSAStartup(MAKEWORD(2,0),&wsd);

SOCKET s =NULL;
s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
struct sockaddr_in ch;
memset(&ch,0,sizeof(ch));
ch.sin_family=AF_INET;
ch.sin_addr.s_addr=INADDR_ANY;
ch.sin_port=htons(1041);
int b=bind(s,(struct sockaddr *) &ch,sizeof(ch));
#define QUEUE_SIZE 5
int l=listen(s,QUEUE_SIZE);
printf("正在监听本机的1041端口！\n");
SOCKET sc=accept(s,0,0);
printf("客户端已经连接到本机的1041端口！\n");
#define BUF_SIZE 4096
int receByt=0;
while(1)
{
char buf[BUF_SIZE];
receByt=recv(sc,buf,BUF_SIZE,0);
buf[receByt]='\0';
if(receByt>0)
{
printf("接收的消息是：%s\n",buf);
}
else
{
printf("接收消息结束！");
break;
}

}
int ic=closesocket(sc);
int is=closesocket(s);

}

======================================================
客户端发送的代码：

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>
#include <string.h>

void main()
{
WSAData wsd;
WSAStartup(MAKEWORD(2,0),&wsd);

SOCKET s =NULL;
s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
struct sockaddr_in ch;
memset(&ch,0,sizeof(ch));
ch.sin_family=AF_INET;
ch.sin_addr.s_addr=inet_addr("127.0.0.1");
ch.sin_port=htons(1041);

int c=connect(s,(struct sockaddr *) &ch,sizeof(ch));
printf("已经连接到服务器的1041端口！现在可以向服务器发送消息了！\n");
#define BUF_SIZE 4096
char info[1024],buf[BUF_SIZE];

while(1)
{
gets(info);
if(info[0]=='\0')
break;
strcpy(buf,info);
int nsend=send(s,buf,strlen(buf),0);

}
int ic=closesocket(s);
}

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

程序代码经过了优化，并且整合多线程，把接收和发送放到同一个文件中，使用参数模式调用发送和接收模块。增加了创建SOCKET的创建的时候s句柄（或对象）判断返回值是否为INVALID_SOCKET，以及socket的bind操作的返回值是否为SOCKET_ERROR，其他socket的操作应该也判断SOCKET_ERROR，以保证程序的稳定性，这里只是测试代码就不去写这么多了，剩下的就由你个人发挥。

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>
#include <string.h>
#include <pthread.h>


void Receive();
void Send();
void creatThread();

SOCKET s =NULL;
pthread_t t[1000];
int threadCount=0;

void main(int argc,char* argv[])
{
    printf("本程序制作人学号：713901040041\n");
    printf("程序说明：服务器端和客户端为同一个程序，请使用不同的参数运行.\n");
    printf("接收程序请使用 r参数；发送程序请使用 s参数。\n");
    //printf("len : %d\n", argc);
    //printf("count %d\n",argc);
    //printf("value: %s\n",argv[1]);
    //printf("%d",argv[1][0]=='r');

    if(argc<=1)
    {
        printf("please input program arguments ...\n");
        exit(0);
    }
    if(argc>1 && argv[1][0]=='r')
    {
        printf("run receive ...\n");
        Receive();
    }
    if(argc>1 && argv[1][0]=='s')
    {
        printf("run send ...\n");
        Send();
    }
}


void* receiveWork(void * args)
{
    SOCKET sc=accept(s,0,0);
    if(sc==INVALID_SOCKET)
    {
        printf("sc Error");
    }
    creatThread();

    printf("----------客户端已经连接到本机的%d线程连接！\n",threadCount-2);
#define BUF_SIZE 4096
    int receByt=0;
    while(1)
    {
        char buf[BUF_SIZE];
        receByt=recv(sc,buf,BUF_SIZE,0);
        buf[receByt]='\0';
        if(receByt>0)
        {
            printf("线程接收的消息是：%s\n",buf);
        }
        else
        {
            printf("客户端已退出，");
            break;
        }
            
    }
    int ic=closesocket(sc);
    printf("服务器结束连接！\n");
    return NULL;
}

void creatThread()
{
    pthread_create(&t[threadCount++],NULL,receiveWork,NULL);
}


void Receive()
{
    WSAData wsd;
    WSAStartup(MAKEWORD(2,0),&wsd);    
    s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    if(s==INVALID_SOCKET)
    {
        printf("socket created Error");
    }
    struct sockaddr_in ch;
    memset(&ch,0,sizeof(ch));
    ch.sin_family=AF_INET;
    ch.sin_addr.s_addr=INADDR_ANY;
    ch.sin_port=htons(1041);
    int b=bind(s,(struct sockaddr *) &ch,sizeof(ch));
    if(b==SOCKET_ERROR)
    {
        printf("bind 失败，出错代码是：%d\n",WSAGetLastError());
        exit(0);
    }
#define QUEUE_SIZE 5
    int l=listen(s,QUEUE_SIZE);
    printf("正在监听本机的1041端口！\n");
    
    creatThread();

    for(int i=0;i<1000;i++)
    {
        pthread_join(t[i],NULL);
    }

int is=closesocket(s);
}



void Send()
{
    WSAData wsd;
    WSAStartup(MAKEWORD(2,0),&wsd);

    SOCKET s =NULL;
    s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    if(s==INVALID_SOCKET)
    {
        printf("socket created Error");
    }
    struct sockaddr_in ch;
    memset(&ch,0,sizeof(ch));
    ch.sin_family=AF_INET;
    ch.sin_addr.s_addr=inet_addr("127.0.0.1");
    ch.sin_port=htons(1041);

    int c=connect(s,(struct sockaddr *) &ch,sizeof(ch));
    printf("已经连接到服务器的1041端口！现在可以向服务器发送消息了！\n");
#define BUF_SIZE 4096
    char info[1024],buf[BUF_SIZE];

    while(1)
    {
        gets(info);
        if(info[0]=='\0')
            break;
        strcpy(buf,info);
        int nsend=send(s,buf,strlen(buf),0);
    }
    int ic=closesocket(s);
}