目的用来练习怎么使用socket来实现网络通信
整体流程:
Created with Rapha?l 2.3.0
server
received connection?
get data and save file
yes
no
Created with Rapha?l 2.3.0
client
get file name
send file name
send file done?
end
yes
no
server程序:
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#define FILE_PATH "./file"
#define DIR_PATH "./"
#define handle_error(msg) \
do { perror(msg); } while(0)
#define BUF_SIZE 1024
void showPeerInfo(struct sockaddr_in* addr) {
printf("peer addr: %s\n", inet_ntoa(addr->sin_addr));
}
int main (int argc, char *argv[])
{
int sfs, rfs;
ssize_t recvByte;
char buffer[BUF_SIZE];
short port = 4000;
struct sockaddr_in addr, in_addr;
socklen_t len;
sfs = socket(AF_INET, SOCK_STREAM, 0);
if (sfs < 0)
handle_error("socket");
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
if (bind(sfs, (struct sockaddr*)&addr, sizeof(addr)) < 0)
handle_error("bind");
if (listen(sfs, 5) < 0)
handle_error("listen");
size_t count=0;
char fileName[20];
strcpy(fileName, FILE_PATH);
while (1) {
rfs = accept(sfs, (struct sockaddr*)&in_addr, &len);
recvByte = recv(rfs, buffer, BUF_SIZE, 0);
if (recvByte == 0) {
sprintf(fileName, "%s_%ld.dat", FILE_PATH, count++);
}
else {
strcpy(fileName, DIR_PATH);
strcat(fileName, buffer);
}
FILE *file = fopen(fileName, "w+");
if (file == NULL) {
handle_error("file");
}
if (rfs < 0)
handle_error("accept");
while (1) {
recvByte = recv(rfs, buffer, BUF_SIZE, 0);
if ((recvByte==-1) ||(recvByte==0) )
break;
fwrite(buffer, sizeof(char), recvByte, file);
}
fclose(file);
close(rfs);
}
return 0;
}
首先需要使用网络套接字(socket)需要创建一个socket,其实可以理解为一个文件描述符,然后将这个socket绑定到一个网络地址上(让client进行连接时使用的地址,使用的协议,使用的端口等),监听端口等待连接。
- socke()创建一个socket文件描述符
- bind()将一个地址,端口绑定到指定的socket
- listen()对指定的端口监听,看是否有client请求连接
需要注意的是,在使用bind函数的时候,需要传入的地址的类型是:struct sockaddr类型的指针。 struct sockaddr的类型如下: _SOCKADDR_COMMON是一个宏,进行展开是: 需要关注的是第二个参数,是一个14个字符的数组,其中包含了地址和端口的信息,由于其混合了地址和端口,所以方便进行赋值,所以经常使用struct sockaddr_in类型,其定义如下: 其中in_port_t类型定义如下:
其中struct in_addr类型的定义如下: 总的来说就是,struct sockaddr_in类型所占据的内部的大小和struck sockaddr所占的大小是相同的,也就是说两个可以直接就进行相互转换,但是struct sockaddr_in对其中的14字节大小的空间再次进行了划分,分为了2字节存储端口号,4字节存储32位的地址,剩的8字节清零。
字节序的注意: 多字节类型的数据在计算机中进行存储的时候,会有大端序和小端序,而在网络上进行数据传输的时候可能采取的顺序和计算机中的存储不相同,所以就需要进行转换,这就是htons函数的作用。
client程序
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#define BUF_SIZE 1024
#define handle_error(msg) \
do {perror(msg);} while(0)
void getFileName(char* dest, const char* sour) {
size_t i, length = strlen(sour);
i = length-1;
while (sour[i--]!= '/');
i += 2;
strcpy(dest, sour+i);
}
int main(int argc, char* argv[])
{
int rfs;
size_t len;
ssize_t sendByte;
short port = 4000;
char buffer[BUF_SIZE];
struct sockaddr_in addr;
FILE* fp;
if (argc != 2) {
printf("Usage: %s fileName\n", argv[0]);
exit(-1);
}
if ((fp = fopen(argv[1], "r+")) == NULL)
handle_error("file exist");
rfs = socket(AF_INET, SOCK_STREAM, 0);
if (rfs == -1)
handle_error("socket");
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
if (connect(rfs, (struct sockaddr*)&addr, sizeof(addr)) < 0)
handle_error("connect");
printf("Start your conversition\n");
getFileName(buffer, argv[1]);
printf("fileName: %s\n", buffer);
send(rfs, buffer, strlen(buffer), 0);
while (1) {
len = fread(buffer, sizeof(char), BUF_SIZE, fp);
if (len == 0)
break;
send(rfs, buffer, len, 0);
}
fclose(fp);
close(rfs);
return 0;
}
首先创建一个socket用来标识需要连接的类型(IPV4,IPV6等,SOCK_STRAM, SOCK_DATAGRAM等),然后建立连接,当然在建立连接的时候需要知道server的IP地址,以及连接的端口号,同样使用struck sockaddr类型来表示,当连接服务器成功之后就可以使用send和recv来和server进行数据的传输了。
- 使用socket()创建一个socket描述符
- 使用connect()将指定的socket和server的地址,端口号进行关联
- 使用send()和recv()来和server进行通信
效果展示:
结果就是会在server程序存在的目录下,保存client发送的同名文件
|