windows环境下使用C++&Socket实现文件传输

server

#include <stdio.h>

#include <iostream>

#include <cstring>

#include <fstream>

#include <winsock2.h>

#pragma comment(lib, "ws2_32.lib")

using namespace std;

SOCKET m_Client; 

void SendFile();

int main(int argc, char* argv[])

{

    //初始化WSA

    WORD sockVersion = MAKEWORD(2,2);

    WSADATA wsaData;

    if(WSAStartup(sockVersion, &wsaData)!=0)

    {

        return 0;

    }  

    //创建套接字

    SOCKET slisten = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if(slisten == INVALID_SOCKET)

    {

        printf("socket error !");

        return 0;

    }  

    //绑定IP和端口

    sockaddr_in sin;

    sin.sin_family = AF_INET;

    sin.sin_port = htons(8888);

    sin.sin_addr.S_un.S_addr = INADDR_ANY;

    if(bind(slisten, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)

    {

        printf("bind error !");

    }  

    //开始监听

    if(listen(slisten, 5) == SOCKET_ERROR)

    {

        printf("listen error !");

        return 0;

    }  

    //循环接收数据

    sockaddr_in remoteAddr;

    int nAddrlen = sizeof(remoteAddr);

    char revData[255];

    while (true)

    {

        printf("等待连接...\n");

        m_Client = accept(slisten, (SOCKADDR *)&remoteAddr, &nAddrlen);

        if(m_Client == INVALID_SOCKET)

        {

            printf("accept error !");

            continue;

        }

        printf("接受到一个连接：%s \r\n", inet_ntoa(remoteAddr.sin_addr));

		SendFile();

        closesocket(m_Client);

    }  

    closesocket(slisten);

    WSACleanup();

    return 0;

}

void SendFile(){

	int haveSend = 0;

	const int bufferSize = 1024;

	char buffer[bufferSize] = {0};

	int readLen = 0;

	string srcFileName = "in";

	ifstream srcFile;

	srcFile.open(srcFileName.c_str(),ios::binary);

	if(!srcFile){

		return;

	}

	while(!srcFile.eof()){

		srcFile.read(buffer,bufferSize);

		readLen = srcFile.gcount();

		send(m_Client,buffer,readLen,0);

		haveSend += readLen;

	}

	srcFile.close();

	cout<<"send: "<<haveSend<<"B"<<endl;

}

client

#include <stdio.h>

#include <iostream>

#include <cstring>

#include <fstream>

#include <winsock2.h>

#pragma comment(lib, "ws2_32.lib")

using namespace std;

SOCKET m_Client; 

void RecvFile();

int main()

{

	WORD sockVersion = MAKEWORD(2, 2);

	WSADATA data;

	if(WSAStartup(sockVersion, &data)!=0)

	{

		return 0;

	}

	//while(true){

		m_Client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

		if(m_Client == INVALID_SOCKET)

		{

			printf("invalid socket!");

			return 0;

		}

		sockaddr_in serAddr;

		serAddr.sin_family = AF_INET;

		serAddr.sin_port = htons(8888);

		serAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");

		if(connect(m_Client, (sockaddr *)&serAddr, sizeof(serAddr)) == SOCKET_ERROR)

		{  //连接失败

			printf("connect error !");

			closesocket(m_Client);

			return 0;

		}

		RecvFile();

		closesocket(m_Client);

	//}

	WSACleanup();

	return 0;

}

void RecvFile(){

	cout<<"start recv!"<<endl;

	const int bufferSize = 1024;

	char buffer[bufferSize] = {0};

	int readLen = 0;

	string desFileName = "new_in";

	ofstream desFile;

	desFile.open(desFileName.c_str(), ios::binary);

	if (!desFile)

	{

	    return ;

	}

	do

	{

	    readLen = recv(m_Client,buffer,bufferSize, 0);

	    if (readLen == 0)

	    {

	        break;

	    }

	    else

	    {

	        desFile.write(buffer, readLen);

	    }

	} while(true);

	desFile.close();

}

巴特西

windows环境下使用C++&Socket实现文件传输

server

client

最新文章

热门文章