ubuntu16.04 socket传输文件,开源软件SocketFileTransfer_ubuntu 转发socket-程序员宅基地

技术标签: c++  探索发现  

arupcsedu / SocketFileTransfer
参考网址https://github.com/arupcsedu/SocketFileTransfer#build-and-run-client

简单修改NWClient一个地方

	char *dirName = argv[1]; //basename(argv[1]);

可以用了,其实Qt封装后的socket更简单,但是没有qt就试试这个吧!
编译系统:ubuntu16.04

作者步骤写的很详细
这是服务端server

/*  File Name : NWServer.cpp
    Author : Arup Sarker
    Creation date : 12/28/2020
    Description: Main responsibilities of this source are
    1. Create socket and bind with client to a specific port
    2. Receive file from client concurrently
    3. Utility function to check file integrity by calculating checkSum

    Notes: BSD socket apis are used and compiled with g++ in Ubuntu 18.04. For Other OS, files needs be cross compiled and create executable
*/

#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <pthread.h>
#include <sys/file.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <assert.h>
#include <time.h>

#define MAX_LENGTH_OF_SINGLE_LINE 4096

#define LISTEN_PORT 8050

#define SERVER_PORT 5080

#define BUFFER_SIZE 100000
#define MAX_NAME 1000
#define MAX_THREAD 100
// ToDo: This will be Dynamic; 
// Status: Done

pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
void* socketThreadFT(void *arg);


//CheckSum Calculation
unsigned calculateChecksum(void *buffer, size_t len)
{
    
	unsigned char *buf = (unsigned char *)buffer;
	size_t i;
	unsigned int checkSum = 0;

	for (i = 0; i < len; ++i)
	    checkSum += (unsigned int)(*buf++);
	return checkSum;
}

// This is a utility function for writing data into file

unsigned int fileWriter(char *fileName, int sockfd, FILE *fp) {
    
	// 11.1 Declare data length
	int n;
	// 11.2 Declare and Initialize buffer size

	time_t t = time(NULL);
	struct tm *tm = localtime(&t);
	char s[64];
	assert(strftime(s, sizeof(s), "%c", tm));
	//printf("%s\n", s);

	printf("%s file Receiving is started in Connection ID: %d at :  %s\n", fileName, sockfd, s);

	// 11.3 Receive data from socket and save into buffer
	unsigned int checkSum = 0;
	char buff[BUFFER_SIZE] = {
    0};
	while(1) {
    		
		n = recv(sockfd, buff, BUFFER_SIZE,0);
		//printf("File is receiving from Socket : %d\n", sockfd);
		if (n == -1) {
    
			perror("Error in Receiving File");
			exit(1);
		}
		if(n <= 0) {
    
			break;
		}
        
        //用于校验数据是否接收完整
		checkSum += calculateChecksum(buff, n);

		// 11.4 Read File Data from buffer and Write into file
		// 11.4 从buffer读取数据,然后保存到file
		if (fwrite(buff, sizeof(char), n, fp) != n) {
    
			perror("Error in Writing File");
			exit(1);
		}

		// 11.5 Allocate buffer
		memset(buff, '\0', BUFFER_SIZE);
	}
	fclose(fp);
	return checkSum;
}



void* socketThreadFT(void *arg)
{
    
    //arg转换为connfd,用于接收数据,connfd就是该服务器中的客户端的id,用于区分不同的客户端
	int connfd = *((int *)arg);
	unsigned int checkSum = 0;

	pthread_mutex_lock(&lock);
	 // 8. Receive incoming file from client
	char filename[MAX_NAME] = {
    0};

    //接收传递过来数据的文件名,用于保存
	if (recv(connfd, filename, MAX_NAME,0) == -1) {
    
		perror("Unable to receive file due to connection or file error");
		exit(1);
	}

	int fileLength = strlen(filename);
	pthread_mutex_unlock(&lock);
	sleep(1);

	if(fileLength > 0) {
    
		// 9. After receiving from client, create file to save data
        // 9.收到客户端传来的信息,开始创建文件来保存数据。从这就可以看出传递文件的本质,就是发送接收char数据,然后再保存在另外的地址。
		//printf("File Name : %s\n", filename);
		//保存接收的数据
		FILE *fp = fopen(filename, "wb");
		if (fp == NULL) {
    
			perror("Unable to create File pointer");
			exit(1);
		}

		// 10. Start receiving file Data and print in console
		char addr[INET_ADDRSTRLEN];

		// 11. Write the data into file.
        // 11.开始保存数据到file
		checkSum = fileWriter(filename, connfd, fp);

		// 12. Print success message into console
		// 12. 打印保存信息
		time_t t = time(NULL);
		struct tm *tm = localtime(&t);
		char s[64];
		assert(strftime(s, sizeof(s), "%c", tm));

		printf("%s is received with checkSum %d at :  %s\n",filename, checkSum,s);

	}

    //断开客户端connfd
	close(connfd);	       
	pthread_exit(NULL);

}



int main(int argc, char *argv[]) {
    
	// 1. Create TCP Socket
    // 1.创建 TCP socket(Transfer Control Protocol 传输控制协议、 socket (电器上的)插口,插孔)
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd == -1) {
    
		perror("Unable to create Socket");
		exit(1);
	}
	//线程mutex初始化,mutex是指一个线程进来后,其他线程就会被阻塞到这里,进不了下一步,原理是啥?感觉应该是内部加了一个循环的if判断(mutex 互斥体; 互斥; 互斥锁)
	if (pthread_mutex_init(&lock, NULL) != 0) {
     
        	printf("mutex init has failed\n"); 
        	exit(1); 
    	} 

	// 2. Setup information about client and server
	// 2. 设置client和server的信息
        printf("2. Setup information about client and server\n");
    // 前面加struct是因为告诉你sockaddr_in是个结构体类,不加也可以,可能不同编译器有不同的行为,可以加上
	struct sockaddr_in nwClientAddr, nwServerAddr;
	//为nwServerAddr分配内存
	memset(&nwServerAddr, 0, sizeof(nwServerAddr));

	// 3. IP address should be IPv4
    // 设置IP address时应该为Ipv4格式,Ipv6不支持
	nwServerAddr.sin_family = AF_INET;
	nwServerAddr.sin_addr.s_addr = htonl(INADDR_ANY);
	nwServerAddr.sin_port = htons(SERVER_PORT);

	// 4. Bind to Server Address
    // 4. 绑定(设置)sockfd的地址
	if (bind(sockfd, (const struct sockaddr *) &nwServerAddr, sizeof(nwServerAddr)) == -1) {
    
		perror("Unable to Bind");
		exit(1);
	}

	// 5. Listen Socket for incoming File
	// 5. sockfd对传递过来的消息,进行监听,这里LISTEN_PORT有些定义有些问题,实际应该为backlog,它是建立好连接处于ESTABLISHED状态的队列的长度。参考文章:https://blog.csdn.net/u022812849/article/details/109737020
	
        printf("5. Listen Socket for incoming File\n");
	if (listen(sockfd, LISTEN_PORT) == -1) {
    
		perror("Error in Socket Listening");
		exit(1);
	}
	
	// 6. Accept connection for concurrency number
	// 接受来自多个客户端(client)的同时(concurrency)连接
        printf("6. Accept connection for concurrency number\n");
	socklen_t adlen = sizeof(nwClientAddr);
	// 这里会阻塞,直到有client进行连接,详细理解见https://blog.csdn.net/tjcwt2011/article/details/122142072,同一地址的多个client可以连接sockfd
	int cfd = accept(sockfd, (struct sockaddr *) &nwClientAddr, &adlen);
	if (cfd == -1) {
    
		perror("Error in Accepting Connection for ");
		exit(1);
	}
	int val = 0;
	//这里加锁了,只能有一个线程继续进行下面的操作,其余的被挡在上面的函数地址,即(int val = 0;)处
	pthread_mutex_lock(&lock);

	char conString[MAX_NAME] = {
    0};
	//这里会返回要传多少个文件到服务器,recv是接收client返回的信息
	if (recv(cfd, conString, MAX_NAME,0) == -1) {
    
		perror("Unable to receive concurrency due to connection or file error");
		exit(1);
	}
	char *numstr = basename(conString);
	val = atoi(numstr);
	printf("Number of files to receive: %d\n", val);
	//断开客户端cfd的连接
	close(cfd);
	pthread_mutex_unlock(&lock);
	
	int *new_sock;
	pthread_t tid[60];
	int idx = 0;
	while(1) {
    

		// 7. Accept connection for creating multi-thread
                printf("7. Accept connection for creating multi-thread\n");
        //这里要结合客户端代码来理解,客户端根据要传递的文件数量,会建立和文件数量相同的client,每个client有独立的线程发送数据
		socklen_t addrlen = sizeof(nwClientAddr);
		int connfd = accept(sockfd, (struct sockaddr *) &nwClientAddr, &addrlen);
		if (connfd == -1) {
    
			perror("Error in Accepting Connection");
			exit(1);
		}

		new_sock = (int*)malloc(1);
		*new_sock = connfd;
		//printf("Connection ID: %d\n", *new_sock);
		// 8. Multi-Thread creation
		// 创建新的线程来接收client发送的数据,所有线程都会调用socketThreadFT函数,所以socketThreadFT函数里有线程锁mutex,来阻塞多线程的行为。
                printf("8. Multi-Thread creation\n");
		if(pthread_create(&tid[idx], NULL, socketThreadFT, (void*)new_sock) < 0 ) {
    
			printf("Failed to create thread\n");
			exit(1);
		}

		idx++;
    
	}

	close(sockfd);
	pthread_mutex_destroy(&lock); 
	return 0;
}


使用方式:

按照github上的进行编译,然后这样使用测试
运行服务端,

./NWServer

运行客户端向服务端发送数据

./NWClient /home/peak/send_file 127.0.0.1 4

// argv[0] = ./NWClient (Executable)
// argv[1] = res (Input directory)
// argv[2] = 127.0.0.1 (loopback ip address to run in same pc. But it can be replaced by server ip address)
// argv[3] = 4 (Number of files can be sent concurrently)

测试发现,这样发也可以

./NWClient "/home/peak/send_file" 127.0.0.1 4
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/windxf/article/details/126867841

智能推荐

攻防世界_难度8_happy_puzzle_攻防世界困难模式攻略图文-程序员宅基地

文章浏览阅读645次。这个肯定是末尾的IDAT了,因为IDAT必须要满了才会开始一下个IDAT,这个明显就是末尾的IDAT了。,对应下面的create_head()代码。,对应下面的create_tail()代码。不要考虑爆破,我已经试了一下,太多情况了。题目来源:UNCTF。_攻防世界困难模式攻略图文

达梦数据库的导出(备份)、导入_达梦数据库导入导出-程序员宅基地

文章浏览阅读2.9k次,点赞3次,收藏10次。偶尔会用到,记录、分享。1. 数据库导出1.1 切换到dmdba用户su - dmdba1.2 进入达梦数据库安装路径的bin目录,执行导库操作  导出语句:./dexp cwy_init/[email protected]:5236 file=cwy_init.dmp log=cwy_init_exp.log 注释:   cwy_init/init_123..._达梦数据库导入导出

js引入kindeditor富文本编辑器的使用_kindeditor.js-程序员宅基地

文章浏览阅读1.9k次。1. 在官网上下载KindEditor文件,可以删掉不需要要到的jsp,asp,asp.net和php文件夹。接着把文件夹放到项目文件目录下。2. 修改html文件,在页面引入js文件:<script type="text/javascript" src="./kindeditor/kindeditor-all.js"></script><script type="text/javascript" src="./kindeditor/lang/zh-CN.js"_kindeditor.js

STM32学习过程记录11——基于STM32G431CBU6硬件SPI+DMA的高效WS2812B控制方法-程序员宅基地

文章浏览阅读2.3k次,点赞6次,收藏14次。SPI的详情简介不必赘述。假设我们通过SPI发送0xAA,我们的数据线就会变为10101010,通过修改不同的内容,即可修改SPI中0和1的持续时间。比如0xF0即为前半周期为高电平,后半周期为低电平的状态。在SPI的通信模式中,CPHA配置会影响该实验,下图展示了不同采样位置的SPI时序图[1]。CPOL = 0,CPHA = 1:CLK空闲状态 = 低电平,数据在下降沿采样,并在上升沿移出CPOL = 0,CPHA = 0:CLK空闲状态 = 低电平,数据在上升沿采样,并在下降沿移出。_stm32g431cbu6

计算机网络-数据链路层_接收方收到链路层数据后,使用crc检验后,余数为0,说明链路层的传输时可靠传输-程序员宅基地

文章浏览阅读1.2k次,点赞2次,收藏8次。数据链路层习题自测问题1.数据链路(即逻辑链路)与链路(即物理链路)有何区别?“电路接通了”与”数据链路接通了”的区别何在?2.数据链路层中的链路控制包括哪些功能?试讨论数据链路层做成可靠的链路层有哪些优点和缺点。3.网络适配器的作用是什么?网络适配器工作在哪一层?4.数据链路层的三个基本问题(帧定界、透明传输和差错检测)为什么都必须加以解决?5.如果在数据链路层不进行帧定界,会发生什么问题?6.PPP协议的主要特点是什么?为什么PPP不使用帧的编号?PPP适用于什么情况?为什么PPP协议不_接收方收到链路层数据后,使用crc检验后,余数为0,说明链路层的传输时可靠传输

软件测试工程师移民加拿大_无证移民,未受过软件工程师的教育(第1部分)-程序员宅基地

文章浏览阅读587次。软件测试工程师移民加拿大 无证移民,未受过软件工程师的教育(第1部分) (Undocumented Immigrant With No Education to Software Engineer(Part 1))Before I start, I want you to please bear with me on the way I write, I have very little gen...

随便推点

Thinkpad X250 secure boot failed 启动失败问题解决_安装完系统提示secureboot failure-程序员宅基地

文章浏览阅读304次。Thinkpad X250笔记本电脑,装的是FreeBSD,进入BIOS修改虚拟化配置(其后可能是误设置了安全开机),保存退出后系统无法启动,显示:secure boot failed ,把自己惊出一身冷汗,因为这台笔记本刚好还没开始做备份.....根据错误提示,到bios里面去找相关配置,在Security里面找到了Secure Boot选项,发现果然被设置为Enabled,将其修改为Disabled ,再开机,终于正常启动了。_安装完系统提示secureboot failure

C++如何做字符串分割(5种方法)_c++ 字符串分割-程序员宅基地

文章浏览阅读10w+次,点赞93次,收藏352次。1、用strtok函数进行字符串分割原型: char *strtok(char *str, const char *delim);功能:分解字符串为一组字符串。参数说明:str为要分解的字符串,delim为分隔符字符串。返回值:从str开头开始的一个个被分割的串。当没有被分割的串时则返回NULL。其它:strtok函数线程不安全,可以使用strtok_r替代。示例://借助strtok实现split#include <string.h>#include <stdio.h&_c++ 字符串分割

2013第四届蓝桥杯 C/C++本科A组 真题答案解析_2013年第四届c a组蓝桥杯省赛真题解答-程序员宅基地

文章浏览阅读2.3k次。1 .高斯日记 大数学家高斯有个好习惯:无论如何都要记日记。他的日记有个与众不同的地方,他从不注明年月日,而是用一个整数代替,比如:4210后来人们知道,那个整数就是日期,它表示那一天是高斯出生后的第几天。这或许也是个好习惯,它时时刻刻提醒着主人:日子又过去一天,还有多少时光可以用于浪费呢?高斯出生于:1777年4月30日。在高斯发现的一个重要定理的日记_2013年第四届c a组蓝桥杯省赛真题解答

基于供需算法优化的核极限学习机(KELM)分类算法-程序员宅基地

文章浏览阅读851次,点赞17次,收藏22次。摘要:本文利用供需算法对核极限学习机(KELM)进行优化,并用于分类。

metasploitable2渗透测试_metasploitable2怎么进入-程序员宅基地

文章浏览阅读1.1k次。一、系统弱密码登录1、在kali上执行命令行telnet 192.168.26.1292、Login和password都输入msfadmin3、登录成功,进入系统4、测试如下:二、MySQL弱密码登录:1、在kali上执行mysql –h 192.168.26.129 –u root2、登录成功,进入MySQL系统3、测试效果:三、PostgreSQL弱密码登录1、在Kali上执行psql -h 192.168.26.129 –U post..._metasploitable2怎么进入

Python学习之路:从入门到精通的指南_python人工智能开发从入门到精通pdf-程序员宅基地

文章浏览阅读257次。本文将为初学者提供Python学习的详细指南,从Python的历史、基础语法和数据类型到面向对象编程、模块和库的使用。通过本文,您将能够掌握Python编程的核心概念,为今后的编程学习和实践打下坚实基础。_python人工智能开发从入门到精通pdf

推荐文章

热门文章

相关标签