文章目录
1. 服务器要求和功能分析都类似,不再赘述,可参考下面链接:
2. 代码模块分析
create_database 创建数据库模块
create_database.h
创建数据库的头文件
#ifndef CREATE_DATABASE_H
#define CREATE_DATABASE_H
int create_database(void);
#endif
create_database.c
创建数据的c文件,为了防止多次创建数据库,每次启动服务器时,直接创建数据库,多个客户端连接时,直接打开数据库,然后插入数据即可。
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <time.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/resource.h>
#include <fcntl.h>
#include <sqlite3.h>
#include "create_database.h"
#define database_name "temper.db"
int create_database(void)
{
sqlite3 *db=NULL;
char *zErrMsg = NULL;
int rc;
int len;
//char sql ="create table if not exists temperature(ID char(10), datetime char(50), temperature char(10))";
char *sql ="create table temperature(ID char(10), datetime char(50), temperature char(10))"; //描述创建数据库中表的信息
/* Open database 若没有则创建*/
len = sqlite3_open(database_name, &db);
if(len != SQLITE_OK)
{
sqlite3_close(db);
fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
return -1; //子线程不能调用exit(0)
}
printf("Opened database successfully\n");
/* Execute SQL statement */
rc = sqlite3_exec(db, sql, 0, 0, &zErrMsg);
if( rc != SQLITE_OK )
{
sqlite3_close(db);
fprintf(stderr, "Create table error: %s\n", zErrMsg);
return 0;
}
printf("Table created successfully\n");
sqlite3_close(db);
return 1;
}
socket_server_init服务器端初始化
socket_server_init.h
初始化的头文件
#ifndef SOCKET_SERVER_INIT_H
#define SOCKET_SERVER_INIT_H
int socket_server_init(char *listenip, int listen_port);
#endif
socket_server_init.c
文件主要是实现创建服务器的socket,bind,listen,每次创建服务器端都必须调用这些函数。
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <time.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/resource.h>
#include <fcntl.h>
#include <sqlite3.h>
#include "server_socket_init.h"
int server_socket_init(char *listenip, int listen_port)
{
int listenfd;
int on = 1;
int rv = 0;
struct sockaddr_in serv_addr;
if((listenfd=socket(AF_INET, SOCK_STREAM, 0)) < 0 )
{
printf("create socket failure: %s\n", strerror(errno));
return -1;
}
//Set socket port reuseable, fix 'Address already in use' bug when socket server restart
//setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(listen_port);
if( !listenip)
{
/* 监听所有的客户端的IP地址 */
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
}
else
{
if(inet_pton(AF_INET,listenip, &serv_addr.sin_addr) <= 0)
{
printf("inet_pton() set listen IP failure: %s\n", strerror(errno));
rv = -2;
goto Cleanup;
}
}
//Set socket port reuseable, fix 'Address already in use' bug when socket server restart
if((setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, (void*)&on, sizeof(on))) < 0 )
{
printf("setsockopt failure: %s\n", strerror(errno));
return -2;
}
if(bind(listenfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
{
printf("bind socket[%d] failure: %s\n", listenfd, strerror(errno));
rv = -3;
goto Cleanup;
}
if(listen(listenfd, 64) < 0)
{
printf("listen() socket[%d] failure: %s\n", listenfd, strerror(errno));
rv = -4;
goto Cleanup;
}
Cleanup:
if(rv < 0)
{
close(listenfd);
}
else
rv = listenfd;
return rv;
}
server_select主函数
server_select.c
文件创建服务器端温度采集实时上报,因为每次运行服务器程序都会创建数据库,插入表。如果已存在数据库,那么在插入表时会失败,导致程序退出,所有在主函数中一定要判断数据是否存在,存在则不用插入表,在while循环中直接插入数据即可,下面165行代码
有注释。
/*********************************************************************************
* Copyright: (C) 2019 Tang Zhiqiang<t_zhiqiang@163.com>
* All rights reserved.
*
* Filename: server_select.c
* Description: This file is socket server RPI temperature
*
* Version: 1.0.0(11/04/2019)
* Author: Tang Zhiqiang <t_zhiqiang@163.com>
* ChangeLog: 1, Release initial version on "11/04/2019 08:15:46 PM"
*
********************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <time.h>
#include <pthread.h>
#include <getopt.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/resource.h>
#include <signal.h>
#include <syslog.h>
#include <fcntl.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sqlite3.h>
#include "server_socket_init.h"
#include "create_database.h"
#define MAX_EVENTS 512
#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0])) //计算结构体/数组元素个数
#define database_name "temper.db"
//void set_socket_rlimit(void);
int create_database(void);
int server_socket_init(char *listenip, int listen_port);
int g_sigstop = 0;
void signal_stop(int signum)
{
if(SIGTERM == signum)
{
printf("SIGTERM signal detected\n");
g_sigstop = 1;
}
}
static inline void print_usage(char *progname)
{
printf("Usage: %s [OPTION]...\n", progname);
printf(" %s is a socket server program, which used to verify client and echo back string from it\n",progname);
printf("\nMandatory arguments to long options are mandatory for short options too:\n");
printf(" -b[daemon ] set program running on background\n");
printf(" -p[port ] Socket server port address\n");
printf(" -h[help ] Display this help information\n");
printf("\nExample: %s -b -p 8900\n", progname);
return ;
}
int main(int argc, char *argv[])
{
int listenfd,connfd;
int serv_port = 0;
int daemon_run = 0;
char *progname = NULL;
int opt;
int rv;
int log_fd;
int i;
int found;
char buf[1024];
char sql1[28];
int fds_array[1024];
fd_set rdset;
int maxfd=0;
char *zErrMsg = NULL;
int ret;
char delim[]="/";
char id[20];
char data_time[50];
char temper[10];
char *ptr=NULL;
sqlite3 *db;
int len;
//char *p=NULL;
struct option long_options[]=
{
{"daemon", no_argument, NULL, 'd'},
{"port", required_argument, NULL, 'p'},
{"help", no_argument, NULL, 'h'},
{NULL, 0, NULL, 0}
};
progname = basename(argv[0]);
//Parser the command line parameters
while((opt = getopt_long(argc, argv, "dp:h", long_options, NULL)) != -1)
{
switch (opt)
{
case 'd':
daemon_run = 1;
break;
case 'p':
serv_port = atoi(optarg);
break;
case 'h':
print_usage(progname);
break;
default:
break;
}
}
if(!serv_port)
{
print_usage(progname);
return -1;
}
signal(SIGTERM, signal_stop); //15号信号(SIGTERM)可被捕捉
if(daemon_run)
{
printf("program %s running at the backgroud now\n", argv[0]);
//建立日志系统
log_fd=open("receive_temper.log", O_CREAT|O_RDWR, 0666);
if(log_fd < 0)
{
printf("Open the logfile failure: %s\n", strerror(errno));
return 0;
}
//标准输出、标准出错重定向
dup2(log_fd, STDOUT_FILENO);
dup2(log_fd, STDERR_FILENO);
//程序后台运行
if(daemon(1, 1) <0 )
{
printf("Deamon failure: %s", strerror(errno));
return 0;
}
}
if((listenfd=server_socket_init(NULL, serv_port)) < 0)
{
printf("ERROR: %s server listen on port %d failure\n", argv[0], serv_port);
return -2;
}
printf("server socket[%d] start to listen on port %d\n", listenfd, serv_port);
printf("Start create database\n");
/* 打开一个已创建的数据库,并已插入表,插入重复的表会失败,防止主程序退出 */
if (access(database_name, W_OK) < 0)
{
if(rv=create_database() < 0)
{
printf("Create database failure: %s\n", strerror(errno));
return -1;
}
}
printf("Create database OK\n");
/* 将加入 listenfd 到集合中 */
for(i=0; i<ARRAY_SIZE(fds_array) ; i++)
{
fds_array[i]=-1; //清空数组,全部赋值为-1
}
fds_array[0] = listenfd;
//printf("programe will start running...\n");
while(!g_sigstop)
{
//每次while循环,清空rdset集合,然后把存放在数组当中的有效文件描述符,再加入到集合中
//每次调用完select()函数后需要将文件描述符集合清空并重新设置,也就是设置的文件描述符集合是一次性使用的。
//原因是调用完select()后文件描述符集合可能发生改变。
FD_ZERO(&rdset); //fd_set rdset 集合清空原因
for(i=0; i<ARRAY_SIZE(fds_array) ; i++)
{
if( fds_array[i] < 0 )
continue; //遍历数组寻找文件描述符
/* 遍历所以有效的文件描述符,将数值最大的那个保存maxfd(select从0~maxfd都遍历)*/
maxfd = fds_array[i]>maxfd ? fds_array[i] : maxfd;
FD_SET(fds_array[i], &rdset);
}
//program will blocked here
printf("programe will blocked here...\n");
rv = select(maxfd+1, &rdset, NULL, NULL, NULL);
if(rv < 0)
{
printf("select failure: %s\n", strerror(errno));
break;
}
else if(rv == 0)
{
printf("select get timeout\n");
continue;
}
/* listen socket get event means new client start connect now */
if ( FD_ISSET(listenfd, &rdset) )
{
if( (connfd=accept(listenfd, (struct sockaddr *)NULL, NULL)) < 0) //保存客户端的地址和IP,不感兴趣可以传空
{
printf("accept new client failure: %s\n", strerror(errno));
continue;
}
found = 0;
for(i=0; i<ARRAY_SIZE(fds_array) ; i++)
{
if( fds_array[i] < 0 )
{
printf("accept new client[%d] and add it into array\n", connfd );
fds_array[i] = connfd;
found = 1;
break;
}
}
if( !found )
{
printf("accept new client[%d] but full, so refuse it\n", connfd);
close(connfd);
}
}/* listen socket get event means new client start connect now */
else //data arriver from alrady connected client
{
for(i=0; i<ARRAY_SIZE(fds_array); i++)
{
//排除一些无效的文件描述符
if( fds_array[i]<0 || !FD_ISSET(fds_array[i], &rdset) )
continue;
memset(buf, 0, sizeof(buf));
if((rv=read(fds_array[i], buf, sizeof(buf))) <= 0 ) //同上
{
printf("socket[%d] read data failure or disconnect and will be remove.\n", fds_array[i]);
close(fds_array[i]);
fds_array[i] = -1;
continue;
}
else
{
printf("socket[%d] read data: %s\n", fds_array[i], buf);
/* Open database */
len = sqlite3_open(database_name, &db);
if(len != SQLITE_OK)
{
sqlite3_close(db);
fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
close(fds_array[i]); //结束当前链接客户端
fds_array[i] = -1;
continue;
}
printf("Opened database successfully\n");
ptr = strtok(buf, delim);
while(ptr != NULL)
{
strncpy(id, ptr, sizeof(id));
ptr=strtok(NULL, delim);
strncpy(data_time, ptr, sizeof(data_time));
ptr=strtok(NULL, delim);
strncpy(temper, ptr, sizeof(temper));
ptr=strtok(NULL, delim);
}
memset(sql1, 0, sizeof(sql1));
snprintf(sql1, 128, "insert into temperature values('%s', '%s', '%s');", id, data_time, temper);
//保证了数组sql1的内容为字符串
sql1[127] = '\0';
printf("%s\n", sql1);
//调用sqlite3_exec();将数据存储至temperature表中
ret = sqlite3_exec(db, sql1, 0 , 0, &zErrMsg);
if (ret != SQLITE_OK) //判断返回值,如果不等于SQLITE_OK,即插入记录失败
{
sqlite3_close(db);
printf("insert data failure: %s!\n", zErrMsg);
close(fds_array[i]); //结束当前链接客户端
fds_array[i] = -1;
continue;
}
printf("insert data successfully!\n");
}//read data from cilent fd
} //for(i=0; i<ARRAY_SIZE(fds_array); i++)
}//data arriver from alrady connected client
}//while(!g_sigstop)
close(listenfd);
sqlite3_close(db); //关闭数据库
return 0;
}
3. 服务器运行结果
4. 客户端代码相同,方便查看直接贴出运行结果
详细的客户端代码如下连接: