//注意數(shù)據(jù)包中多字節(jié)在傳輸過(guò)程中的字節(jié)序的問(wèn)題 htons/htonl
//通過(guò)socket發(fā)送結(jié)構(gòu)體空間時(shí)的字節(jié)對(duì)齊導(dǎo)致的大小不一致問(wèn)題
//使用某種技術(shù)實(shí)現(xiàn)發(fā)送數(shù)據(jù)的長(zhǎng)度可變化的結(jié)構(gòu)體
//方法一
struct msg_node
{
unsigned short len ; //發(fā)送的數(shù)據(jù)長(zhǎng)度
char *data ; //數(shù)據(jù)部分苔悦，以'\0'作為結(jié)束符
};
//方法二
struct msg_node
{
unsigned short len ; //發(fā)送的數(shù)據(jù)長(zhǎng)度
char data[0] ; //數(shù)據(jù)部分轩褐，以'\0'作為結(jié)束符 占位，
};
malloc
strcpy
//方法三
struct msg_node
{
unsigned short len; //發(fā)送的數(shù)據(jù)長(zhǎng)度
char data[1] ; //數(shù)據(jù)部分玖详，以'\0'作為結(jié)束符
};
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <dirent.h>
#include "my_header.h"
#include "my_ftp_server.h"
int main(int argc, char *argv[])
{
if(argc != 3)
{
printf("usage : %s <server_ip> <server_port>\n", argv[0]);
return 1;
}
int sock_fd = 0;
int conn_fd = 0;
int n = 0;
struct sockaddr_in serveraddr;
struct sockaddr_in clientaddr;
socklen_t len = 0;
// 存儲(chǔ)客戶端和服務(wù)器的交互數(shù)據(jù)
struct data_package pkg;
memset(&pkg, 0, sizeof(pkg));
// 1.socket
sock_fd = socket(AF_INET, SOCK_STREAM, 0);
// 2.bind
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
bind(sock_fd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
// 3.listen
listen(sock_fd, 15);
// 4.accept
while(1)
{
printf("waiting for connetion...\n");
len = sizeof(clientaddr);
conn_fd = accept(sock_fd, (struct sockaddr *)&clientaddr, &len);
printf("connection with %s : %d\n",
inet_ntoa(clientaddr.sin_addr),
ntohs(clientaddr.sin_port));
while(1)
{
// 循環(huán)讀取客戶端發(fā)送來(lái)的數(shù)據(jù)把介，分析并執(zhí)行相應(yīng)操作
// 直到客戶端退出為止
n = recv(conn_fd, &pkg, sizeof(pkg), 0);
switch(pkg.cmd)
{
case CMD_LIST:
// 執(zhí)行l(wèi)ist操作
put_list_data_to_client(conn_fd);
break;
case CMD_GET:
// 執(zhí)行g(shù)et操作
put_file_data_to_client(conn_fd, pkg.filename);
break;
case CMD_PUT:
// 執(zhí)行put操作
get_file_data_from_client(conn_fd, pkg.filename);
break;
case CMD_QUIT:
printf("server quit...\n");
// 執(zhí)行quit操作
break;
default:
break;
}
}
// 5.關(guān)閉連接套接字
close(conn_fd);
}
// 6.close監(jiān)聽(tīng)套接字
close(sock_fd);
return 0;
}
// 響應(yīng)客戶端的list命令
// conn_fd 通過(guò)指定的連接套接字傳遞數(shù)據(jù)
void put_list_data_to_client(int conn_fd)
{
struct data_package pkg;
// 1.打開(kāi)文件服務(wù)器根目錄
DIR *p_dir = opendir(SERVER_ROOT_DIR);
// 2.循環(huán)讀取目錄項(xiàng)信息，發(fā)送給客戶端
struct dirent *p = NULL;
while((p = readdir(p_dir)) != NULL)
{
if(strncmp(p->d_name, ".", 1) == 0)
continue;
memset(&pkg, 0, sizeof(pkg));
pkg.cmd = CMD_LIST;
strcpy(pkg.data, p->d_name);
pkg.flag = 0;
send(conn_fd, &pkg, sizeof(pkg), 0);
}
// 2.1發(fā)送數(shù)據(jù)結(jié)束狀態(tài)數(shù)據(jù)包
memset(&pkg, 0, sizeof(pkg));
pkg.cmd = CMD_LIST;
pkg.flag = DATA_END_FLAG; // 結(jié)束標(biāo)志
send(conn_fd, &pkg, sizeof(pkg), 0);
// 3.關(guān)閉目錄
closedir(p_dir);
}
// 響應(yīng)客戶端的get命令
// conn_fd 連接套接字
// file 客戶端待下載的文件名
void put_file_data_to_client(int conn_fd, char *file)
{
struct data_package pkg;
char path[MAX_FILE_NAME_LEN] = {'\0'};
int fd = 0;
// 構(gòu)造待下載文件在服務(wù)器文件系統(tǒng)中的絕對(duì)路徑
sprintf(path, "%s/%s", SERVER_ROOT_DIR, file);
// 1.判斷待下載文件是否存在蟋座，如果不存在拗踢，直接發(fā)送數(shù)據(jù)結(jié)束狀態(tài)包
if(access(path, F_OK) == -1)
{
memset(&pkg, 0, sizeof(pkg));
pkg.cmd = CMD_GET;
pkg.flag = DATA_END_FLAG;
send(conn_fd, &pkg, sizeof(pkg), 0);
return;
}
// 2.如果文件存在，打開(kāi)文件
fd = open(path, O_RDONLY);
// 3.循環(huán)的讀取文件內(nèi)容向臀，發(fā)送給客戶端
memset(&pkg, 0, sizeof(pkg));
pkg.cmd = CMD_GET;
pkg.flag = 0;
while(read(fd, pkg.data, MAX_DATA_LEN-1) > 0)
{
send(conn_fd, &pkg, sizeof(pkg), 0);
memset(pkg.data, 0, MAX_DATA_LEN);
}
// 4.構(gòu)造數(shù)據(jù)結(jié)束狀態(tài)包
memset(&pkg, 0, sizeof(pkg));
pkg.cmd = CMD_GET;
pkg.flag = DATA_END_FLAG;
send(conn_fd, &pkg, sizeof(pkg), 0);
// 5.關(guān)閉文件
close(fd);
}
// 響應(yīng)客戶端的put命令
// conn_fd 連接套接字
// file 客戶端待上傳的文件名
void get_file_data_from_client(int conn_fd, char *file)
{
struct data_package pkg;
int fd = 0;
int n = 0;
char path[MAX_FILE_NAME_LEN] = {'\0'};
// 構(gòu)建待上傳文件在服務(wù)器文件系統(tǒng)中的絕對(duì)路徑
sprintf(path, "%s/%s", SERVER_ROOT_DIR, file);
// 1.打開(kāi)文件巢墅，如果文件不存在，則創(chuàng)建飒硅；如果文件存在砂缩，則將其長(zhǎng)度截短為0
fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0644);
// 2.循環(huán)讀取文件內(nèi)容，寫(xiě)入文件
while(1)
{
recv(conn_fd, &pkg, sizeof(pkg), 0);
if(pkg.cmd != CMD_PUT)
continue;
if(pkg.flag == DATA_END_FLAG)
break;
n = strlen(pkg.data);
write(fd, pkg.data, n);
}
// 3.關(guān)閉文件
close(fd);
}
// udp echo 客戶端
// 從鍵盤(pán)讀入數(shù)據(jù)三娩，發(fā)送給服務(wù)器庵芭，并接收客戶端回彈的數(shù)據(jù)，打印
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define N 1024
int main(int argc, char *argv[])
{
if(argc != 3)
{
printf("usage : %s <server_ip> <server_port>\n", argv[0]);
return 1;
}
int sock_fd;
char buf[N] = {'\0'};
struct sockaddr_in serveraddr;
memset(&serveraddr, 0, sizeof(serveraddr));
// 1.socket SOCK_DGRAM
if((sock_fd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket failed");
return 1;
}
// 構(gòu)造服務(wù)器地址
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
while(1)
{
printf(">>>>>");
fgets(buf, N, stdin);
buf[strlen(buf)-1] = '\0';
// 2.sendto
// ssize_t sendto(int sockfd, const void *buf, size_t len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen);
// 參數(shù)：
// 前四個(gè)參數(shù)與send函數(shù)一致
// dest_addr 提供發(fā)送目的地址結(jié)構(gòu)體的起始地址
// addrlen 發(fā)送目的地址結(jié)構(gòu)體的長(zhǎng)度
sendto(sock_fd, buf, N, 0, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
if(strncmp(buf, ".exit", 5) == 0)
break;
memset(buf, 0, N);
// 3.recvfrom
// ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags, struct sockaddr *src_addr, socklen_t *addrlen);
// 參數(shù)：
// 前四個(gè)參數(shù)與recv函數(shù)一致
// src_addr 用于提供獲取發(fā)送該數(shù)據(jù)的地址信息結(jié)構(gòu)體的起始地址
// addrlen 用于提供地址信息結(jié)構(gòu)體的長(zhǎng)度雀监，并接收實(shí)際地址信息結(jié)構(gòu)體的長(zhǎng)度
recvfrom(sock_fd, buf, N, 0, NULL, NULL);
printf("echo from server : %s\n", buf);
}
// 4.close
return 0;
}
// udp echo 客戶端
// 從鍵盤(pán)讀入數(shù)據(jù)双吆，發(fā)送給服務(wù)器眨唬，并接收客戶端回彈的數(shù)據(jù)，打印
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define N 1024
int main(int argc, char *argv[])
{
if(argc != 3)
{
printf("usage : %s <server_ip> <server_port>\n", argv[0]);
return 1;
}
int sock_fd;
char buf[N] = {'\0'};
struct sockaddr_in serveraddr;
memset(&serveraddr, 0, sizeof(serveraddr));
// 1.socket SOCK_DGRAM
if((sock_fd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket failed");
return 1;
}
// 構(gòu)造服務(wù)器地址
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
while(1)
{
printf(">>>>>");
fgets(buf, N, stdin);
buf[strlen(buf)-1] = '\0';
// 2.sendto
// ssize_t sendto(int sockfd, const void *buf, size_t len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen);
// 參數(shù)：
// 前四個(gè)參數(shù)與send函數(shù)一致
// dest_addr 提供發(fā)送目的地址結(jié)構(gòu)體的起始地址
// addrlen 發(fā)送目的地址結(jié)構(gòu)體的長(zhǎng)度
sendto(sock_fd, buf, N, 0, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
if(strncmp(buf, ".exit", 5) == 0)
break;
memset(buf, 0, N);
// 3.recvfrom
// ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags, struct sockaddr *src_addr, socklen_t *addrlen);
// 參數(shù)：
// 前四個(gè)參數(shù)與recv函數(shù)一致
// src_addr 用于提供獲取發(fā)送該數(shù)據(jù)的地址信息結(jié)構(gòu)體的起始地址
// addrlen 用于提供地址信息結(jié)構(gòu)體的長(zhǎng)度好乐，并接收實(shí)際地址信息結(jié)構(gòu)體的長(zhǎng)度
recvfrom(sock_fd, buf, N, 0, NULL, NULL);
printf("echo from server : %s\n", buf);
}
// 4.close
return 0;
}
