65 lines
1.6 KiB
C++
65 lines
1.6 KiB
C++
|
// 示例客户端程序
|
||
|
|
// 发送文件到服务端
|
||
|
|
#include <iostream>
|
||
|
|
#include <fstream>
|
||
|
|
#include "tn.h"
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
struct FileStat { //存放文件信息
|
||
|
|
char name[128]; // 文件名
|
||
|
|
size_t fsize = 0, read = 0, ssize = 0;
|
||
|
|
//文件大小 文件已读大小 本次发送大小
|
||
|
|
};
|
||
|
|
|
||
|
|
int pton(char *, const char *);
|
||
|
|
int file_out(TN &, char *); // 传入一个连接和文件路径
|
||
|
|
|
||
|
|
int main(int argc, char *argv[]) {
|
||
|
|
if(argc != 2) {
|
||
|
|
cout << "Using: ./cli.exe 文件名" << endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
TN test1;
|
||
|
|
if(test1.client("127.0.0.1","12345")!=0)
|
||
|
|
return 0;
|
||
|
|
cout << "连接已成功" << endl;
|
||
|
|
cout << file_out(test1, argv[1]) << endl;
|
||
|
|
cout << "连接断开" << endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int pton(char *name, const char *path) {
|
||
|
|
int ptr = strlen(path);
|
||
|
|
while(ptr!=0 && path[ptr-1]!='/')
|
||
|
|
ptr--;
|
||
|
|
for(int j=0; ptr<=strlen(path); j++,ptr++)
|
||
|
|
name[j]=path[ptr];
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int file_out(TN &tn, char *path) { // 传入一个连接和文件路径
|
||
|
|
const unsigned int BSIZE = 1024;
|
||
|
|
char buffer[BSIZE];
|
||
|
|
FileStat fs;
|
||
|
|
fstream fin(path, ios::in|ios::binary);
|
||
|
|
pton(fs.name, (const char *)path);
|
||
|
|
fin.seekg(0, ios::end);
|
||
|
|
fs.fsize = fin.tellg();
|
||
|
|
fin.seekg(0, ios::beg);
|
||
|
|
|
||
|
|
// 发送信息
|
||
|
|
if(tn.out((FileStat *)&fs, sizeof(fs))!=0) return 0;
|
||
|
|
if(tn.in((char *)buffer)!=0) return 0; // 接收确认报文
|
||
|
|
|
||
|
|
// 正式发送文件
|
||
|
|
while(fin.good() && fs.fsize>fs.read) {
|
||
|
|
memset(buffer, 0, sizeof(buffer));
|
||
|
|
fs.ssize = (fs.fsize-fs.read)>BSIZE? BSIZE:(fs.fsize-fs.read);
|
||
|
|
fin.read(buffer, fs.ssize);
|
||
|
|
if(tn.out(buffer, fs.ssize)!=0) return -1;
|
||
|
|
fs.read += fs.ssize;
|
||
|
|
}
|
||
|
|
fin.close();
|
||
|
|
return 0;
|
||
|
|
}
|