1.随便聊聊
本周的任务要求:基于SOCKET(TCP/UDP)实现文件收发,需要实现虚拟机读文件,然后一个包一个包发送给SE5,然后在SE5上一个包一个包接收后存储到文件里。所以本篇笔记会讲题主自己对socket传输的理解和应用
2.环境配置
在控制台输入指令
sudo apt-get build-dep gcc
#安装gcc
由于我们使用c++,需要gcc来进行编译。
3.知识梳理
1.文件读写操作
fopen()函数:fq = fopen(argv[2],"rb"),这个函数需要两个参数,第一个参数是文件路径,第二个参数是读写模式,rb表示读方式打开一个二进制文件,不允许写数据,文件必须存在
fread()函数:len = fread(buffer, sizeof(char), sizeof(buffer), fq),搞清楚这个函数,我们需要知道一个关键参数buffer的含义:buffer是缓冲区的意思,这里这个参数是将文件读取到内存的位置,在运用这个参数前,我们需要执行bzero()或是memset()进行地址清零操作,如bzero(buffer,sizeof(buffer))
然后是第二和第三个参数:
第二个参数是读取的 基本单元 字节大小 , 单位是字节 , 一般是 buffer 缓冲的单位大小 ;
- 如果 buffer 缓冲区是 char 数组 , 则该参数的值是 sizeof(char) ;
- 如果 buffer 缓冲区是 int 数组 , 则该参数的值是 sizeof(int) ;
第三个参数是读取的基本单元个数,一般写sizeof(buffer),一次读满整个缓冲区即可
然后就是完整的一个读取块,需要用到feof(p)判断文件是否读取完毕,基本框架如下
while(!feof(fp)){
memset(buffer, 0, sizeof(buffer));
// buffer : 将文件读取到内存的位置
// sizeof(char) : 读取的基本单元字节长度
// sizeof(buffer) : 读取的基本单元个数,
// 读取字节个数是 sizeof(buffer) * sizeof(char)
// p : 文件指针
fread(buffer, sizeof(char), sizeof(buffer) , fp);
}
其中fread有返回值,为读取的次数,可以直接作为下面write函数的第三个参数
write()函数:write(sockfd, buffer, len)write()会把参数buf所指的内存写入len个字节到参数fd所指的文件内,这里写到sockfd这个文件里,再传输到服务端,关于sockfd我们后文解释
read()函数:n = read(connfd, buff, MAXLINE),read()会把参数fd所指的文件传送count (第三个参数)个字节到buf 指针所指的内存中,返回值为实际读到的字节数,会用到下一步的fwrite中
fwrite()函数:fwrite(buff, 1, n, fp),buffer是一个指针,对fwrite来说,是要获取数据的地址,1是单字节数,n是要写的项数,fp是要写入的文件的指针
好了,这里就是所有的文件读写操作,大家肯定注意到有两组读写,有f和没f的,我只知道涉及到socket数据的读写不用f,本地文件读写则要f
fp这个本地文件不难理解,之后在socket传输理解时题主试着理解sockfd和connfd这两个文件
2.socket tcp传输操作及函数
这个流程图里非常清晰,但是看着一个个函数名字太过空洞难以理解,那么听我说你先别急,咱把函数理解后,再返回来看图,应该会明确一些思路。
1.socket()函数:
//创建socket,使用IP协议(PF_INET)+TCP协议(SOCK_STREAM)
int fd_listen = socket(PF_INET, SOCK_STREAM, 0);
第一个参数为使用ipv4协议的意思,比如你想用IPv6协议,就改成PF_INET6,第二个参数决定采用tcp还是udp,第三个参数是选择协议,0为第二个参数对应的默认协议
2.bind()函数:
//绑定固定ip:port地址
in_addr_t ip_num = inet_addr(ip);
sockaddr_in addr_server = {AF_INET, port, ip_num};
bind(fd_listen, (sockaddr *)&addr_server, sizeof(addr_server));
inet_addr:如果正确执行将返回一个无符号长整数型数。如果传入的字符串不是一个合法的IP地址,将返回INADDR_NONE,这个函数既是把我们输入的ip地址转成数字,又是对ip的检验
sockaddr_in addr_server:sockaddr_in是一个结构体,这里实例化一个addr_server,包含三个项
bind():第一个参数为socket套接字,就是上面用socket()函数创建的,第二个是我们上面实例化的结构体,第三个是对应地址的长度
只有服务端调用bind的原因:通常服务器在启动时会绑定一个总所周知的地址(ip地址+端口号),客户端不用指定系统自动分配,所以通常服务端在listen之前要调用bind(),而客户端不会调用,在connect()时由系统随机生成一个
3.listen():
listen(fd_listen, 10);
//监听socket
第一个参数同样是用socket()函数创建的socket套接字,第二个参数是等待队列的最大长度
一般情况下,一个进程只有一个主线程(也就是单线程),那么socket允许的最大连接数为: n + 1
如果服务器是多线程,比如开了2个线程,那么socket允许的最大连接数就是: n + 2
4.connect():
//连接服务器
in_addr_t ip_num = inet_addr(ip);
sockaddr_in addr_server = {AF_INET, port, ip_num}; //服务器地址
connect(fd_conn, (sockaddr *)&addr_server, sizeof(addr_server));
上面两个函数和bind一致,不解释了,connect函数用在客户端,fd_conn为客户端的socket套接字,connect的函数和bind函数的参数甚至也大相径庭,用来建立与指定socket的连接
剩下那些,连接上后实际上就是进行文件读写的操作了,只不过读写的主体除了本地文件,还有socket套接字
注:(5):argc和argv:
这个实际上不涉及socket,但也在这里提一下
argc 是 argument count的缩写,表示传入main函数的参数个数;
argv 是 argument vector的缩写,表示传入main函数的参数序列或指针,并且第一个参数argv[0]一定是程序的名称,并且包含了程序所在的完整路径,所以确切的说需要我们输入的main函数的参数个数应该是argc-1个;
实际上就是在执行代码时传给main函数的参数啦
4.代码及结果
server.cpp:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<errno.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<unistd.h>
#define MAXLINE 4096
int main(int argc, char** argv){
int listenfd, connfd;
struct sockaddr_in servaddr;
char buff[4096];
FILE *fp;
int n;
if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 ){
printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);
return 0;
}
printf("----init socket----\n");
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(6666);
//设置端口可重用
int contain;
setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &contain, sizeof(int));
if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){
printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);
return 0;
}
printf("----bind sucess----\n");
if( listen(listenfd, 10) == -1){
printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);
return 0;
}
if((fp = fopen(argv[1],"ab") ) == NULL )
{
printf("File.\n");
close(listenfd);
exit(1);
}
printf("======waiting for client's request======\n");
while(1){
struct sockaddr_in client_addr;
socklen_t size=sizeof(client_addr);
if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &size)) == -1){
printf("accept socket error: %s(errno: %d)",strerror(errno),errno);
continue;
}
while(1){
n = read(connfd, buff, MAXLINE);
if(n == 0)
break;
fwrite(buff, 1, n, fp);
}
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
close(connfd);
fclose(fp);
}
close(listenfd);
return 0;
}
cilent.cpp:
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netdb.h>
#define MAXLINE 4096
int main(int argc, char** argv){
int sockfd, len;
char buffer[MAXLINE];
struct sockaddr_in servaddr;
FILE *fq;
if( argc != 3){
printf("error!\n");
return 0;
}
if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0){
printf("create socket error: %s(errno: %d)\n", strerror(errno),errno);
return 0;
}
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(6666);
if( inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0){
printf("inet_pton error for %s\n",argv[1]);
return 0;
}
if( connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0){
printf("connect error: %s(errno: %d)\n",strerror(errno),errno);
return 0;
}
if( ( fq = fopen(argv[2],"rb") ) == NULL ){
printf("File open.\n");
close(sockfd);
exit(1);
}
bzero(buffer,sizeof(buffer));
while(!feof(fq)){
len = fread(buffer, 1, sizeof(buffer), fq);
if(len != write(sockfd, buffer, len)){
printf("write.\n");
break;
}
}
close(sockfd);
fclose(fq);
return 0;
}
用gcc转换成可执行文件后,直接在终端执行,要注意给相应的入口参数,这里不说给的入口参数是什么了,好好理解上文所讲,自然知道了捏XD
结果:
这里,不带new的是原文件,带new的是经过socket传输后的文件。
好,结束!诚挚感谢@周召生的共同研究