2017-2018-1 20155321 20155330 《信息安全系统设计基础》实验三——实时系统
任务一
- 学习使用Linux命令wc(1)
- 基于Linux Socket程序设计实现wc服务器(端口号是你学号的后6位)和客户端
- 客户端传一个文本文件给服务器
- 服务器返加文本文件中的单词数
实验步骤
-
wc命令：加上参数-w可查看当前文档的单词数，如下图所示
-
wc命令代码如下所示：
#include <stdio.h>
#include <string.h>
int *getCharNum(char *filename, int *totalNum);
int main(){
char filename[30];
int totalNum[1] = {0};
printf("Input file name: ");
scanf("%s", filename);
if(getCharNum(filename, totalNum)){
printf("Total: %d words
", totalNum[0]);
}else{
printf("Error!
");
}
return 0;
}
int *getCharNum(char *filename, int *totalNum){
FILE *fp; // 指向文件的指针
char buffer[1003]; //缓冲区，存储读取到的每行的内容
int bufferLen; // 缓冲区中实际存储的内容的长度
int i; // 当前读到缓冲区的第i个字符
char c; // 读取到的字符
int isLastBlank = 0; // 上个字符是否是空格
int charNum = 0; // 当前行的字符数
int wordNum = 0; // 当前行的单词数
if( (fp=fopen(filename, "rb")) == NULL ){
perror(filename);
return NULL;
}
while(fgets(buffer, 1003, fp) != NULL){
bufferLen = strlen(buffer);
for(i=0; i<bufferLen; i++){
c = buffer[i];
if( c==' ' || c==' '){ // 遇到空格
!isLastBlank && wordNum++; // 如果上个字符不是空格，那么单词数加1
isLastBlank = 1;
}else if(c!='
'&&c!='
'){ // 忽略换行符
charNum++; // 如果既不是换行符也不是空格，字符数加1
isLastBlank = 0;
}
}
!isLastBlank && wordNum++; // 如果最后一个字符不是空格，那么单词数加1
isLastBlank = 1; // 每次换行重置为1
// 一行结束，计算总单词数
totalNum[0] += wordNum; // 总单词数
// 置零，重新统计下一行
charNum = 0;
wordNum = 0;
}
return totalNum;
}
- 实现思路：先由客户端上传文件给服务器，服务器保存好此文件后再进行统计字数工作，统计好后打印出相应结果
- 服务器端代码如下
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <dirent.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <fcntl.h>
#define LEN 4096
#define MAXLINE 40
#define BUFFERSIZE 4096
#define PORT 13321
ssize_t writen(int fd, const void * vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char * ptr;
ptr = vptr;
nleft = n;
while ( nleft > 0)
{
if ((nwritten = write(fd, ptr, nleft)) <= 0)
{
if (nwritten < 0 && errno == EINTR)
{
nwritten = 0;
}
else
{
return -1;
}
}
nleft -= nwritten;
ptr += nwritten;
}
return n;
}
ssize_t readline(int fd, void * vptr, size_t maxlen)
{
ssize_t n, rc;
char c, *ptr;
ptr = vptr;
for (n = 1; n < maxlen; n++)
{
again:
if ((rc = read(fd, &c, 1)) == 1)
{
*ptr++ = c;
if (c == '
')
{
break;
}
}
else if (rc == 0)
{
*ptr = 0;
return (n - 1);
}
else
{
if (errno == EINTR)
{
goto again;
}
return (-1);
}
}
*ptr = 0;
return (n);
}
int Socket(int domain, int type, int protocol)
{
int sockfd;
if ((sockfd = socket(domain, type, protocol)) < 0)
{
fprintf(stderr, "socket error
");
exit(1);
}
return sockfd;
}
int Accept(int sockfd, struct sockaddr * addr, socklen_t * addrlen)
{
int ret;
if ((ret = accept(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "accept error
");
exit(1);
}
return ret;
}
int Bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
{
int ret;
if ((ret = bind(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "bind error
");
exit(1);
}
return ret;
}
int Listen(int sockfd, int backlog)
{
int ret;
if ((ret = listen(sockfd, backlog)) < 0)
{
fprintf(stderr, "listen error
");
exit(1);
}
return ret;
}
int Close(int fd)
{
int ret;
if ((ret = close(fd)) < 0)
{
fprintf(stderr, "close error
");
exit(1);
}
return ret;
}
int main(void)
{
int listenfd, connfd;
char buff[BUFFERSIZE + 1];
char filename[BUFFERSIZE + 1];
char cd[BUFFERSIZE+1];
char choose[10];
struct sockaddr_in servaddr, cliaddr;
int cliaddrlen;
int filefd;
int count;
DIR *dir;
struct dirent *ptr;
listenfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = INADDR_ANY;
servaddr.sin_port = htons(PORT);
Bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
Listen(listenfd, 5);
while(1)
{
printf("开始监听
");
cliaddrlen = sizeof(cliaddr);
connfd = Accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddrlen);
if (readline(connfd, buff, BUFFERSIZE) < 0)
{
fprintf(stderr, "readline error
");
exit(1);
}
buff[strlen(buff) - 1] = 0; /* change '
' to NUL */
memcpy(filename, buff, BUFFERSIZE + 1);
printf("统计的文件名: %s
", buff);
printf("Input the direct you want to store %s:
", buff);
scanf("%s", cd);
if(chdir(cd) < 0)
{
fprintf(stderr, "direct error
");
exit(1);
}
dir = opendir(cd);
while((ptr = readdir(dir)) != NULL)
{
if(strcmp(buff, ptr->d_name) == 0)
{
printf("已存在文件：%s
", buff);
printf("若想重命名请输入yes,否则请输入no
");
scanf("%s", choose);
if(strcmp(choose, "yes") == 0)
{
printf("重命名为: ");
scanf("%s", buff);
}
else
{
printf("已存在该文件, 已退出！
");
exit (1);
}
}
}
filefd = open(buff, O_WRONLY | O_CREAT);
if (filefd < 0)
{
fprintf(stderr, "can't open the file: %s
", buff);
exit(1);
}
while(count = read(connfd, buff, BUFFERSIZE))
{
if (count < 0)
{
fprintf(stderr, "connfd read error
");
exit(1);
}
if (writen(filefd, buff, count) < 0)
{
fprintf(stderr, "writing to filefd error
");
exit(1);
}
}
closedir(dir);
Close(filefd);
Close(connfd);
printf("file %s received!
", filename);
FILE *fp; // 指向文件的指针
char buffer[1003]; //缓冲区，存储读取到的每行的内容
int bufferLen; // 缓冲区中实际存储的内容的长度
int i; // 当前读到缓冲区的第i个字符
char c; // 读取到的字符
int isLastBlank = 0; // 上个字符是否是空格
int charNum = 0; // 当前行的字符数
int wordNum = 0; // 当前行的单词数
int totalNum[1] = {0};
if( (fp = fopen("/home/rafel/shiyan3/1/save/text1.txt","r+")) == NULL ){
perror(filename);
exit(1);
}
while(fgets(buffer, 1003, fp) != NULL){
bufferLen = strlen(buffer);
// 遍历缓冲区的内容
for(i=0; i<bufferLen; i++){
c = buffer[i];
if( c==' ' || c==' '){ // 遇到空格
!isLastBlank && wordNum++; // 如果上个字符不是空格，那么单词数加1
isLastBlank = 1;
}else if(c!='
'&&c!='
'){ // 忽略换行符
charNum++; // 如果既不是换行符也不是空格，字符数加1
isLastBlank = 0;
}
}
!isLastBlank && wordNum++; // 如果最后一个字符不是空格，那么单词数加1
isLastBlank = 1; // 每次换行重置为1
// 一行结束，计算总单词数
totalNum[0] += wordNum; // 总单词数
// 置零，重新统计下一行
charNum = 0;
wordNum = 0;
}
printf("Total: %d words
", totalNum[0]);
}
Close(listenfd);
return 0;
}
- 客户端代码如下
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <fcntl.h>
#define MAXLINE 40
#define BUFFERSIZE 4096
#define PORT 13321
ssize_t writen(int fd, const void * vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char * ptr;
ptr = vptr;
nleft = n;
while (nleft > 0)
{
if ((nwritten = write(fd, ptr, nleft)) <= 0)
{
if (nwritten < 0 && errno == EINTR)
{
nwritten = 0;
}
else
{
return -1;
}
}
nleft -= nwritten;
ptr += nwritten;
}
return n;
}
int Socket(int domain, int type, int protocol)
{
int sockfd;
if ((sockfd = socket(domain, type, protocol)) < 0)
{
fprintf(stderr, "socket error
");
exit(1);
}
return sockfd;
}
int Connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
{
int ret;
if ((ret = connect(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "connect error
");
exit(1);
}
return ret;
}
int Close(int fd)
{
int ret;
if ((ret = close(fd)) < 0)
{
fprintf(stderr, "close error
");
exit(1);
}
return ret;
}
int main(int argc, char *argv[])
{
if (argc != 3)
{
fprintf(stderr, "Usage: ./fileclient <file> <serverIP>
");
exit(1);
}
int sockfd;
char buff[BUFFERSIZE + 1];
char filenameheader[BUFFERSIZE + 1];
struct sockaddr_in servaddr;
int filefd; /* file descriptor */
int count;
sockfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr=INADDR_ANY;
servaddr.sin_port = htons(PORT);
Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
printf("已连接服务器
");
printf("需统计的文件名为: %s........
", argv[1]);
memcpy(filenameheader, argv[1], strlen(argv[1]));
filenameheader[strlen(argv[1])] = '
';
filenameheader[strlen(argv[1]) + 1] = 0;
writen(sockfd, filenameheader, strlen(filenameheader));
printf("正上传文件%s至服务器
", argv[1]);
filefd = open(argv[1], O_RDONLY);
if (filefd < 0)
{
fprintf(stderr, "can't open the file: %s
", argv[1]);
exit(1);
}
while(count = read(filefd, buff, BUFFERSIZE))
{
if (count < 0)
{
fprintf(stderr, "filefd read error
");
exit(1);
}
if (writen(sockfd, buff, count) < 0)
{
fprintf(stderr, "writing to sockfd error
");
exit(1);
}
}
Close(filefd);
Close(sockfd);
printf("文件%s已上传至服务器!
", argv[1]);
return 0;
}
- 运行结果如下图所示
任务二
-
使用多线程实现wc服务器并使用同步互斥机制保证计数正确
-
服务器端代码如下所示：
#include <stdio.h>
#include <stdlib.h>
#include<pthread.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <dirent.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <fcntl.h>
#define LEN 4096
#define MAXLINE 40
#define BUFFERSIZE 4096
#define PORT 13321
#define BUFFER_LENGTH 1024
#define MAX_CONN_LIMIT 512 //MAX connection limit
ssize_t writen(int fd, const void * vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char * ptr;
ptr = vptr;
nleft = n;
while ( nleft > 0)
{
if ((nwritten = write(fd, ptr, nleft)) <= 0)
{
if (nwritten < 0 && errno == EINTR)
{
nwritten = 0;
}
else
{
return -1;
}
}
nleft -= nwritten;
ptr += nwritten;
}
return n;
}
ssize_t readline(int fd, void * vptr, size_t maxlen)
{
ssize_t n, rc;
char c, *ptr;
ptr = vptr;
for (n = 1; n < maxlen; n++)
{
again:
if ((rc = read(fd, &c, 1)) == 1)
{
*ptr++ = c;
if (c == '
')
{
break;
}
}
else if (rc == 0)
{
*ptr = 0;
return (n - 1);
}
else
{
if (errno == EINTR)
{
goto again;
}
return (-1);
}
}
*ptr = 0;
return (n);
}
int Socket(int domain, int type, int protocol)
{
int sockfd;
if ((sockfd = socket(domain, type, protocol)) < 0)
{
fprintf(stderr, "socket error
");
exit(1);
}
return sockfd;
}
int Accept(int sockfd, struct sockaddr * addr, socklen_t * addrlen)
{
int ret;
if ((ret = accept(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "accept error
");
exit(1);
}
return ret;
}
int Bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
{
int ret;
if ((ret = bind(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "bind error
");
exit(1);
}
return ret;
}
int Listen(int sockfd, int backlog)
{
int ret;
if ((ret = listen(sockfd, backlog)) < 0)
{
fprintf(stderr, "listen error
");
exit(1);
}
return ret;
}
int Close(int fd)
{
int ret;
if ((ret = close(fd)) < 0)
{
fprintf(stderr, "close error
");
exit(1);
}
return ret;
}
static void Data_handle(void * sock_fd)
{
int fd = *((int *)sock_fd);
int i_recvBytes;
char data_recv[BUFFER_LENGTH];
const char * data_send = "Server has received your request!
";
pthread_mutex_t counter_mutex = PTHREAD_MUTEX_INITIALIZER;
while(1)
{
pthread_mutex_lock( &counter_mutex );
int totalNum[1] = {0};
FILE *fp; // 指向文件的指针
char buffer[1003]; //缓冲区，存储读取到的每行的内容
int bufferLen; // 缓冲区中实际存储的内容的长度
int i; // 当前读到缓冲区的第i个字符
char c; // 读取到的字符
int isLastBlank = 0; // 上个字符是否是空格
int charNum = 0; // 当前行的字符数
int wordNum = 0; // 当前行的单词数
if( (fp=fopen("/home/rafel/shiyan3/2/save", "rb")) == NULL ){
perror(filename);
exit(1);
}
while(fgets(buffer, 1003, fp) != NULL){
bufferLen = strlen(buffer);
// 遍历缓冲区的内容
for(i=0; i<bufferLen; i++){
c = buffer[i];
if( c==' ' || c==' '){ // 遇到空格
!isLastBlank && wordNum++; // 如果上个字符不是空格，那么单词数加1
isLastBlank = 1;
}else if(c!='
'&&c!='
'){ // 忽略换行符
charNum++; // 如果既不是换行符也不是空格，字符数加1
isLastBlank = 0;
}
}
!isLastBlank && wordNum++; // 如果最后一个字符不是空格，那么单词数加1
isLastBlank = 1; // 每次换行重置为1
// 一行结束，计算总单词数
totalNum[0] += wordNum; // 总单词数
// 置零，重新统计下一行
charNum = 0;
wordNum = 0;
pthread_mutex_unlock( &counter_mutex );
}
printf("Total: %d words
", totalNum[0]);
}
printf("terminating current client_connection...
");
close(fd); //close a file descriptor.
pthread_exit(NULL); //terminate calling thread!
}
int main(void)
{
int listenfd, connfd;
char buff[BUFFERSIZE + 1];
char filename[BUFFERSIZE + 1];
char cd[BUFFERSIZE+1];
char choose[10];
struct sockaddr_in servaddr, cliaddr;
int cliaddrlen;
int filefd;
int count;
DIR *dir;
struct dirent *ptr;
listenfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = INADDR_ANY;
servaddr.sin_port = htons(PORT);
Bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
Listen(listenfd, 5);
while(1)
{
printf("开始监听
");
cliaddrlen = sizeof(cliaddr);
connfd = Accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddrlen);
if (readline(connfd, buff, BUFFERSIZE) < 0)
{
fprintf(stderr, "readline error
");
exit(1);
}
buff[strlen(buff) - 1] = 0; /* change '
' to NUL */
memcpy(filename, buff, BUFFERSIZE + 1);
printf("统计的文件名: %s
", buff);
printf("Input the direct you want to store %s:
", buff);
scanf("%s", cd);
if(chdir(cd) < 0)
{
fprintf(stderr, "direct error
");
exit(1);
}
dir = opendir(cd);
while((ptr = readdir(dir)) != NULL)
{
if(strcmp(buff, ptr->d_name) == 0)
{
printf("已存在文件：%s
", buff);
printf("若想重命名请输入yes,否则请输入no
");
scanf("%s", choose);
if(strcmp(choose, "yes") == 0)
{
printf("重命名为: ");
scanf("%s", buff);
}
}
}
filefd = open(buff, O_WRONLY | O_CREAT);
if (filefd < 0)
{
fprintf(stderr, "can't open the file: %s
", buff);
exit(1);
}
while(count = read(connfd, buff, BUFFERSIZE))
{
if (count < 0)
{
fprintf(stderr, "connfd read error
");
exit(1);
}
if (writen(filefd, buff, count) < 0)
{
fprintf(stderr, "writing to filefd error
");
exit(1);
}
}
int sockfd = Accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddrlen);
//sockfd = accept(sockfd_server,(struct sockaddr*)(&s_addr_client),(socklen_t *)(&client_length));
pthread_t thread_id;
if(pthread_create(&thread_id,NULL,(void *)(&Data_handle),(void *)(&sockfd)) == -1)
{
fprintf(stderr,"pthread_create error!
");
break; //break while loop
}
if(pthread_create(&thread_id,NULL,(void *)(&Data_handle),(void *)(&sockfd)) == -1)
{
fprintf(stderr,"pthread_create error!
");
break; //break while loop
}
closedir(dir);
Close(filefd);
Close(connfd);
printf("file %s received!
", filename);
}
}
- 客户端代码如下所示：
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <fcntl.h>
#define MAXLINE 40
#define BUFFERSIZE 4096
#define PORT 13321
ssize_t writen(int fd, const void * vptr, size_t n)
{
size_t nleft;
ssize_t nwritten;
const char * ptr;
ptr = vptr;
nleft = n;
while (nleft > 0)
{
if ((nwritten = write(fd, ptr, nleft)) <= 0)
{
if (nwritten < 0 && errno == EINTR)
{
nwritten = 0;
}
else
{
return -1;
}
}
nleft -= nwritten;
ptr += nwritten;
}
return n;
}
int Socket(int domain, int type, int protocol)
{
int sockfd;
if ((sockfd = socket(domain, type, protocol)) < 0)
{
fprintf(stderr, "socket error
");
exit(1);
}
return sockfd;
}
int Connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
{
int ret;
if ((ret = connect(sockfd, addr, addrlen)) < 0)
{
fprintf(stderr, "connect error
");
exit(1);
}
return ret;
}
int Close(int fd)
{
int ret;
if ((ret = close(fd)) < 0)
{
fprintf(stderr, "close error
");
exit(1);
}
return ret;
}
int main(int argc, char *argv[])
{
if (argc != 3)
{
fprintf(stderr, "Usage: ./fileclient <file> <serverIP>
");
exit(1);
}
int sockfd;
char buff[BUFFERSIZE + 1];
char filenameheader[BUFFERSIZE + 1];
struct sockaddr_in servaddr;
int filefd; /* file descriptor */
int count;
sockfd = Socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr=INADDR_ANY;
servaddr.sin_port = htons(PORT);
Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
printf("已连接服务器
");
printf("需统计的文件名为: %s........
", argv[1]);
memcpy(filenameheader, argv[1], strlen(argv[1]));
filenameheader[strlen(argv[1])] = '
';
filenameheader[strlen(argv[1]) + 1] = 0;
writen(sockfd, filenameheader, strlen(filenameheader));
printf("正上传文件%s至服务器
", argv[1]);
filefd = open(argv[1], O_RDONLY);
if (filefd < 0)
{
fprintf(stderr, "can't open the file: %s
", argv[1]);
exit(1);
}
while(count = read(filefd, buff, BUFFERSIZE))
{
if (count < 0)
{
fprintf(stderr, "filefd read error
");
exit(1);
}
if (writen(sockfd, buff, count) < 0)
{
fprintf(stderr, "writing to sockfd error
");
exit(1);
}
}
Close(filefd);
Close(sockfd);
printf("文件%s已上传至服务器!
", argv[1]);
return 0;
}
- 运行结果如下图所示
新学到的知识点
- 服务器-客户端模型
- 基于C语言的Socket编程
- 多线程编程
- 使用函数pthread_create创建一个新的线程
- 使用函数pthread_join等待某进程结束
- 线程间的分工合作
- 使用互斥机制保证线程间数据的安全共享