网络编程基础第四讲阻塞模型

#pragma once
#include "stdafx.h"
#include <WinSock2.h>
#include <iostream>
#pragma comment(lib,"ws2_32.lib")
using namespace std;


#include "initSocket.h"

void DebugLog(TCHAR *str);
//初始化数据
int initSocket();

//1.创建套接字.绑定地址,开始监听
SOCKET BindAnListen(int nBacklog);

//接受连接分装

SOCKET AccepeConnect(SOCKET hSocket);

//接受跟发送数据
BOOL ClientReadAnWriteData(SOCKET hSocket);

//关闭数据连接
BOOL ColseConnect(SOCKET hSocket);

#include "initSocket.h"

void DebugLog(TCHAR *str)
{
    cout << str << WSAGetLastError() << endl;
}
//初始化数据
int initSocket()
{
    WSADATA data;
    if (WSAStartup(MAKEWORD(2, 2), &data))
    {
        DebugLog(TEXT("initsocket faile"));
        return 0;
    }
}

//1.创建套接字.绑定地址,开始监听
SOCKET BindAnListen(int nBacklog)
{
    //创建套接字
    BOOL bRet = FALSE;
    SOCKET hSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (INVALID_SOCKET == hSocket)
    {
        DebugLog(TEXT("BindAnListen Fail"));
        return INVALID_SOCKET;
    }
                
    //绑定套接字
    sockaddr_in addr;
    addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    //htonl addr_any
    addr.sin_family = AF_INET;
    addr.sin_port = htons(8524);
    bRet =  bind(hSocket, (sockaddr *)&addr, sizeof(addr));
    if (SOCKET_ERROR == bRet)
    {

        DebugLog(TEXT("bind fail"));
        closesocket(hSocket);
        WSACleanup();
        return INVALID_SOCKET;
    }
    //监听套接字
    bRet = FALSE;
    bRet =  listen(hSocket, nBacklog);
    if (SOCKET_ERROR ==bRet)
    {
        DebugLog(TEXT("Listen fail"));
        closesocket(hSocket);
        WSACleanup();
        return INVALID_SOCKET;
    }
    return hSocket;
}

//接受连接分装

SOCKET AccepeConnect(SOCKET hSocket)
{
    sockaddr_in addr;
    int nSize = sizeof(addr);
    SOCKET hNewSocket = accept(hSocket, (LPSOCKADDR)&addr, &nSize);
    if (hNewSocket == INVALID_SOCKET)
    {

        DebugLog(TEXT("Accept An Connect Fail"));
        return INVALID_SOCKET;
    }
    return hNewSocket;
}

//接受跟发送数据
BOOL ClientReadAnWriteData(SOCKET hSocket)
{
    char szBuffer[1024] = { NULL };
    int nBufferSzie = sizeof(szBuffer);
    //循环处理数据
    int nRecvBytes = 0;
    do
    {
        nRecvBytes = recv(hSocket,szBuffer, nBufferSzie, 0);
        if (SOCKET_ERROR == nRecvBytes)
        {

            DebugLog(TEXT("Recv Data Fail"));
            return FALSE;
        }
        else if (0 != nRecvBytes)
        {
　　　　　　　szBuffer[nRecvBytes] = 0; 　　　　
            cout << "接受到的数据为: " << szBuffer << endl;
            //接着循环发送回去.
            int nSendDataBytes = 0;
            while (nSendDataBytes < nRecvBytes)
            {
                int nRetValue = send(hSocket, szBuffer, nBufferSzie, 0);
                if (nRetValue > 0)
                {
                    nSendDataBytes = nSendDataBytes + nRetValue;
                    //每次发送的数据都增加.这样就会发送过去了

                }
                else if (nRetValue == SOCKET_ERROR)
                {
                    DebugLog(TEXT("发送数据失败"));
                    return FALSE;
                }
                else
                {
                    //send 返回0 也就是send失败了.客户端关闭了
                    DebugLog(TEXT("发送数据失败,客户端已经关闭了"));
                    return FALSE;
                }


            }
        }

    } while (0 != nRecvBytes);
    return FALSE;
}

BOOL ColseConnect(SOCKET hSocket)
{
    //shutdown 跟 closesocket一样.不过 TCP 会发送一个FIN分段.给对方表名已经完成数据发送
    if (shutdown(hSocket,SD_SEND) == SOCKET_ERROR)
    {
        DebugLog(TEXT("关闭连接失败"));
        return FALSE;
    }
    //注意.客户端会发送一个数据.不写也可以.
    return TRUE;
}

// Server.cpp : 定义控制台应用程序的入口点。
//

#include "initSocket.h"


int main()
{
    //初始化
    initSocket();
    //1.绑定并且监听
    SOCKET hSocket = BindAnListen(1);
    if (INVALID_SOCKET == hSocket)
    {
        DebugLog(TEXT("main Bind Fail"));
        goto Opt;
    }
    // 2.循环接受套接字连接
    while (true)                            //主要代码是这里.
    {
        //接受客户端连接
        SOCKET hRetSocket = AccepeConnect(hSocket);

        if (INVALID_SOCKET == hRetSocket)
        {
            DebugLog(TEXT("main accept Fail"));
            break;
        }
        
        //读取数据.
        if (FALSE == ClientReadAnWriteData(hRetSocket))
        {
            //只服务一个socket.对其进行读取写入操作.然后下方进行关闭.

            break;
        }
        if (ColseConnect(hRetSocket))
        {
            break;
        }

    }

Opt:
    getchar(); //等待一下.观看错误内容
    ColseConnect(hSocket);

    return 0;
}