// tcp协议传输部分借鉴代码,改动部分为把文件中读取的十六进制数据转化为char数组进行传输
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <direct.h>
#include <iostream>
#include <stdlib.h>
#include <WinSock2.h>
#include <fstream>
#include <sstream>
#include <thread>
#pragma comment(lib, "ws2_32.lib") //加载 ws2_32.dll
#define BUF_SIZE 1128
#define DATA_SIZE 1120
using std::cout;
using std::cin;
using std::endl;
//TODO 解决黏包问题
//通过套接字获取IP、Port等地址信息
bool GetAddressBySocket(SOCKET m_socket, SOCKADDR_IN& m_address)
{
memset(&m_address, 0, sizeof(m_address));
int nAddrLen = sizeof(m_address);
//根据套接字获取地址信息
if (::getpeername(m_socket, (SOCKADDR*)& m_address, &nAddrLen) != 0)
{
printf("获取IP地址失败");
return false;
}
//读取IP和Port
return true;
}
int main() {
//初始化DLL
WSADATA wsaData;
int err = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (err != 0) {
cout << "WSAS tart up error";
return -1;
}
//缓冲区大小 send buffer max_size == 8M ==8*1024 *1024 Byte
int trans_len = 40 * 1024;
int len = sizeof(trans_len);
//向服务器发起请求
sockaddr_in sockAddr;
memset(&sockAddr, 0, sizeof(sockAddr)); //每个字节都用0填充
sockAddr.sin_family = AF_INET;
sockAddr.sin_addr.s_addr = inet_addr("172.20.163.41"); //回送地址
sockAddr.sin_port = htons(11999);
char bufSend[DATA_SIZE] = { 0 };
char bufRecv[DATA_SIZE] = { 0 };
//创建套接字
SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
connect(sock, (SOCKADDR*)& sockAddr, sizeof(SOCKADDR));
//设置发送缓冲区
if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (char*)& trans_len, len) < 0)
{
cout << "setsocketopt error";
return -1;
}
//获取服务器地址端口相关
SOCKADDR_IN address;
GetAddressBySocket(sock, address);
cout << "IP: " << ::inet_ntoa(address.sin_addr) << " PORT: " << ntohs(address.sin_port) << endl;
//获取文件位置
char buffer_m[MAX_PATH];
_getcwd(buffer_m, MAX_PATH);
std::ifstream file;
do
{
cout << "input file name" << std::endl;
std::string filename = "n3_out.txt";
//cin >> filename;
filename = buffer_m + std::string("\\") + filename;
file.open(filename, std::ifstream::binary);
} while (!file.is_open());
cout << "The file has been opened to send data" << std::endl;
//传输文件 发送完毕标志
char c;
char data[280][4] = { '\0' };
char msg1[282][4] = { '\0' };
bool flag = true;
char buf1[5] = { '\0' };
while (1)
{
int judge = 0;
int read_number = 0;
int count = 0;
int file_number1 = 1;
for (int j = 0; j < 280; j++)
{
memset(buf1, 0, 5);
file.get(buf1, 5);
memcpy(&data[j][0], buf1, 4);
if (c = file.eof())
{
judge = 1;
read_number = j;
break;
}
read_number = j;
}
if (read_number != 0)
{
//Gets the read data length
count = (read_number + 1) * 4;
}
char middle1[5];
char middle2[5];
_itoa(file_number1, msg1[0], 10); //十进制数据转化为char数组
_itoa(count, msg1[1], 10);
//sizeof(BUF_SIZE) == 4 Byte the size of BUF_SIZE itself
memcpy(&msg1[2], &data, DATA_SIZE); //Use the defined size as the memcpy size
int send_char_num = send(sock, (char*)msg1, BUF_SIZE, 0);
if (send_char_num == 0)
{
cout << "server disconnected" << endl;
flag = false;
break;
}
if (send_char_num < 0)
{
cout << "Go wrong" << endl;
flag = false;
break;
}
file_number1 += 1;
if (judge == 1) break;
}
if (flag)
{
cout << "file has been sent" << endl;
}
file.close();
closesocket(sock); //关闭套接字
WSACleanup(); //终止使用 DLL
return 0;
}