最简单的socket c/s模型

//=================[server]====================
#include "stdafx.h"
#include "stdio.h"
#include "winsock2.h"
#include <vector>
using std::vector;

SOCKET s;
vector<SOCKET> cc;

void onAccept()
{
    sockaddr_in sa;
    int add_len = sizeof(sa);
    SOCKET c = accept(s,(sockaddr*)&sa,&add_len);
    if(c!=INVALID_SOCKET)
    {
        printf("client:%s port:%d connected!/n",inet_ntoa(sa.sin_addr),sa.sin_port);
        send(c,"hello",5,0);
        cc.push_back(c);
    }
    else
    {
        printf("invalid connect/n");
    }
}

void onRecv(const fd_set &fs)
{
    for(int i=0;i<cc.size();++i)
    {
        if(FD_ISSET(cc[i],&fs))
        {
            char buffer[256]={0};
            int rc= recv(cc[i],buffer,255,0);
            if(rc>0)
            {
                printf("recv msg:%s/n",buffer);
                send(cc[i],buffer,strlen(buffer)+1,0);
            }
            else if(rc == SOCKET_ERROR)
            {
                printf("recv msg failed:%d/n",::WSAGetLastError());
                closesocket(cc[i]);
                cc.erase(cc.begin()+i);
                break;
            }
        }
    }
}


int main(int argc, char* argv[])
{
    //init
    WORD wVersionRequested;
    WSADATA wsaData;
    wVersionRequested = MAKEWORD( 2, 2 );
    WSAStartup( wVersionRequested, &wsaData );
    
    s = socket(AF_INET,SOCK_STREAM,0);
    if(s!=INVALID_SOCKET)
    {
        printf("create socket success!/n");
    }
    
    {
        sockaddr_in sa;
        sa.sin_family = AF_INET;
        sa.sin_port = htons( 1500 );
        sa.sin_addr.s_addr = 0 ; //*²»Ö¸¶¨µØÖ·
        int rc = bind(s,(sockaddr *)&sa,sizeof(sa));
        if(rc == SOCKET_ERROR)
        {
            printf("bind failed:%d/n",::WSAGetLastError());
            return -1;
        }
    }
    
    listen(s,SOMAXCONN);
    
    timeval timeout={0,100};
    
    while(true)
    {
        fd_set fs;
        FD_ZERO(&fs);
        FD_SET(s,&fs);
        
        int rc = select(0,&fs,0,0,&timeout);
        if(rc>0)
        {
            onAccept();
        }
        else if(rc == 0)
        {
            //printf("no connect!/n");
        }
        else
        {
            printf("select1 error:%d/n",::WSAGetLastError());
        }
        
        
        if(cc.size()>0)
        {
            FD_ZERO(&fs);
            for(int i=0;i<cc.size();++i)
            {
                FD_SET(cc[i],&fs);
            }
            
            rc = select(0,&fs,0,0,&timeout);
            if(rc>0)
            {
                onRecv(fs);
            }
            else if(rc == 0)
            {
                //printf("no recv!/n");
            }
            else
            {
                printf("select2 error:%d/n",::WSAGetLastError());
            }
        }
    }
    
    closesocket(s);
    
    //clean
    WSACleanup( );
    return 1;
}

//=====================[client]====================
#include "stdafx.h"
#include <stdio.h>
#include <winsock2.h>
#include <windows.h>
#include <process.h>

SOCKET c;

unsigned int __stdcall myrecv(void *)
{
    while(true)
    {
        char recv_buffer[256]={0};
        int rc =recv(c,recv_buffer,255,0)>0;
        if(rc>0)
        {
            printf("recv:%s/n",recv_buffer);
        }
        else if(rc==SOCKET_ERROR)
        {
            printf("recv failed:%d/n",::WSAGetLastError());
            break;
        }
    }
    _endthreadex(0);
    return 0;
}

int main(int argc, char* argv[])
{
    //init
    WORD wVersionRequested;
    WSADATA wsaData;
    wVersionRequested = MAKEWORD( 2, 2 );
    WSAStartup( wVersionRequested, &wsaData );
    
    c = socket(AF_INET,SOCK_STREAM,0);
    if(c!=INVALID_SOCKET)
    {
        printf("create socket success!/n");
    }
    
    //connect
    {
        sockaddr_in sa;
        sa.sin_family = AF_INET;
        sa.sin_port = htons( 1500 );
        sa.sin_addr.s_addr = inet_addr("127.0.0.1") ; //*²»Ö¸¶¨µØÖ·
        int rc = connect(c,(sockaddr *)&sa,sizeof(sa));
        if(rc == SOCKET_ERROR)
        {
            printf("connect failed:%d/n",::WSAGetLastError());
            return -1;
        }
    }

    _beginthreadex(0,0,&myrecv,0,0,0);
    
    int count = 0;
    timeval timeout={0,100};
    while(true)
    {
        char send_buffer[256]={0};
        sprintf(send_buffer,"s%d",count++);
        int rc = send(c,send_buffer,strlen(send_buffer),0);
        if(rc>0)
        {
            printf("send:%s/n",send_buffer);
        }
        else if(rc==SOCKET_ERROR)
        {
            printf("send failed:%d/n",::WSAGetLastError());
            break;
        }
    }

    closesocket(c);
    //clean
    WSACleanup( );
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值