如何在嵌入式开发板上内置网页,通过多个客服端来同时访问呢?因为需要大量线程来处理网页的任务,为了避免频繁的申请释放线程所带来的开销,使用了线程池来操作!
源代码:
//服务端监听客服端访问网页的请求
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/select.h>
#include <fcntl.h>
#include "thread_pool.h"
void wb_fun(int sockfd,char *buff,int buff_len)
{
char head[]="HTTP/1.1 200 OK\r\n" \
"Content-Type: text/html\r\n" \
"\r\n";
char err[]= "HTTP/1.1 404 Not Found\r\n" \
"Content-Type: text/html\r\n" \
"\r\n" \
"<HTML><BODY>File not found</BODY></HTML>";
char filename[1024] = ""; //文件名
char file[1024] = ""; //存储文件内容
uint32_t len = 0;
if(sscanf(buff, "GET /%s", filename) != 0){
int fd = open(filename,O_RDONLY);//打开文件
if(fd == -1){ //打开失败
send(sockfd, err, strlen(err), 0);
perror("open");
}
else{
send(sockfd, head, strlen(head), 0);
bzero(file, sizeof(file));
printf("open file:%s\n",filename);
while ( (len = read(fd, file, sizeof(file))) != 0){//读取文件
send(sockfd, file, len, 0);//发送文件
bzero(file, sizeof(file));
}
close(fd);
printf("close file:%s\n",filename);
}
}
}
void *task_test(void *arg)
{
int fd = (int) arg;
char recv_buff[1024];
bzero(recv_buff,sizeof(recv_buff));
int len = recv(fd, recv_buff, sizeof(recv_buff), 0);//接收数据
if(len >0){
printf("recv:%s\n",recv_buff);
wb_fun(fd,recv_buff,sizeof(recv_buff));
}
close(fd); //关闭连接
return NULL;
}
int main(int argc, char *argv[])
{
struct sockaddr_in addr;
bzero(&addr,sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(13000);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
int temp = 1;
setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR, &temp, sizeof(int));
int res = bind(sockfd, (struct sockaddr *)&addr, sizeof(addr));
if(res != 0){
perror("bind");
}
//修改套接字为被动属性
listen(sockfd, 5);
//创建文件监听
fd_set readfds,tmpfds;
FD_ZERO(&readfds);
FD_ZERO(&tmpfds);
FD_SET(sockfd, &readfds);//将sockfd 加入读属性变化文件描述符
//线程池 描述结构体
pool_t pool;
pool_init(&pool, 5);//初始化一个线程池,其中创建2个线程
int fd = 0;
while(1){
tmpfds = readfds;
res = select(1024, &tmpfds, NULL, NULL, NULL);
if(res > 0){
for(fd = 0; fd < 1024; fd ++){
if(FD_ISSET(fd, &tmpfds)){//检查1~1024 谁在内
if(fd == sockfd){//新连接建立
int connect = accept(fd, NULL, NULL);
FD_SET(connect, &readfds);//将新连接加入 监听
}
else{
pool_add_task(&pool, task_test, (void *)fd);//添加一个任务
FD_CLR(fd, &readfds); //清除描述符
}
}
}
}
}
pool_uninit(&pool);//删除线程池
close(sockfd);
return 0;
}
//功能:线程池
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <assert.h>
#include "thread_pool.h"
static void *pool_thread_server(void *arg);
/*********************************************************************
*功能: 初始化线程池结构体并创建线程
*参数:
pool:线程池句柄
threads_limit:线程池中线程的数量
*返回值: 无
*********************************************************************/
void pool_init(pool_t *pool, int threads_limit)
{
pool->threads_limit = threads_limit;
pool->queue_head = NULL;
pool->task_in_queue = 0;
pool->destroy_flag = 0;
/*创建存放线程ID的空间*/
pool->threadid = (pthread_t *)calloc(threads_limit, sizeof(pthread_t));
int i = 0;
/*初始化互斥锁和条件变量*/
pthread_mutex_init(&(pool->queue_lock), NULL);
pthread_cond_init(&(pool->queue_ready), NULL);
/*循环创建threads_limit个线程*/
for (i = 0; i < threads_limit; i++){
pthread_create(&(pool->threadid[i]), NULL, pool_thread_server, pool);
}
return;
}
/*********************************************************************
*功能: 销毁线程池,等待队列中的任务不会再被执行,
但是正在运行的线程会一直,把任务运行完后再退出
*参数: 线程池句柄
*返回值: 成功:0,失败非0
*********************************************************************/
int pool_uninit(pool_t *pool)
{
pool_task *head = NULL;
int i;
pthread_mutex_lock(&(pool->queue_lock));
if(pool->destroy_flag)/* 防止两次调用 */
return -1;
pool->destroy_flag = 1;
pthread_mutex_unlock(&(pool->queue_lock));
/* 唤醒所有等待线程,线程池要销毁了 */
pthread_cond_broadcast(&(pool->queue_ready));
/* 阻塞等待线程退出,否则就成僵尸了 */
for (i = 0; i < pool->threads_limit; i++)
pthread_join(pool->threadid[i], NULL);
free(pool->threadid);
/* 销毁等待队列 */
pthread_mutex_lock(&(pool->queue_lock));
while(pool->queue_head != NULL){
head = pool->queue_head;
pool->queue_head = pool->queue_head->next;
free(head);
}
pthread_mutex_unlock(&(pool->queue_lock));
/*条件变量和互斥量也别忘了销毁*/
pthread_mutex_destroy(&(pool->queue_lock));
pthread_cond_destroy(&(pool->queue_ready));
return 0;
}
/*********************************************************************
*功能: 向任务队列中添加一个任务
*参数:
pool:线程池句柄
process:任务处理函数
arg:任务参数
*返回值: 无
*********************************************************************/
static void enqueue_task(pool_t *pool, pool_task_f process, void *arg)
{
pool_task *task = NULL;
pool_task *member = NULL;
pthread_mutex_lock(&(pool->queue_lock));
if(pool->task_in_queue >= pool->threads_limit){
printf("task_in_queue > threads_limit!\n");
pthread_mutex_unlock (&(pool->queue_lock));
return;
}
task = (pool_task *)calloc(1, sizeof(pool_task));
assert(task != NULL);
task->process = process;
task->arg = arg;
task->next = NULL;
pool->task_in_queue++;
member = pool->queue_head;
if(member != NULL){
while(member->next != NULL) /* 将任务加入到任务链连的最后位置. */
member = member->next;
member->next = task;
}else{
pool->queue_head = task; /* 如果是第一个任务的话,就指向头 */
}
printf("\ttasks %d\n", pool->task_in_queue);
/* 等待队列中有任务了,唤醒一个等待线程 */
pthread_cond_signal (&(pool->queue_ready));
pthread_mutex_unlock (&(pool->queue_lock));
}
/*********************************************************************
*功能: 从任务队列中取出一个任务
*参数: 线程池句柄
*返回值: 任务句柄
*********************************************************************/
static pool_task *dequeue_task(pool_t *pool)
{
pool_task *task = NULL;
pthread_mutex_lock(&(pool->queue_lock));
/* 判断线程池是否要销毁了 */
if(pool->destroy_flag){
pthread_mutex_unlock(&(pool->queue_lock));
printf("thread 0x%lx will be destroyed\n", pthread_self());
pthread_exit(NULL);
}
/* 如果等待队列为0并且不销毁线程池,则处于阻塞状态 */
if(pool->task_in_queue == 0){
while((pool->task_in_queue == 0) && (!pool->destroy_flag)){
printf("thread 0x%lx is leisure\n", pthread_self());
/* 注意:pthread_cond_wait是一个原子操作,等待前会解锁,唤醒后会加锁 */
pthread_cond_wait(&(pool->queue_ready), &(pool->queue_lock));
}
}else{
/* 等待队列长度减去1,并取出队列中的第一个元素 */
pool->task_in_queue--;
task = pool->queue_head;
pool->queue_head = task->next;
printf("thread 0x%lx received a task\n", pthread_self());
}
pthread_mutex_unlock(&(pool->queue_lock));
return task;
}
/*********************************************************************
*功能: 向线程池中添加一个任务
*参数:
pool:线程池句柄
process:任务处理函数
arg:任务参数
*返回值: 0
*********************************************************************/
int pool_add_task(pool_t *pool, pool_task_f process, void *arg)
{
enqueue_task(pool, process, arg);
return 0;
}
/*********************************************************************
*功能: 线程池服务程序
*参数: 略
*返回值: 略
*********************************************************************/
static void *pool_thread_server(void *arg)
{
pool_t *pool = NULL;
pool = (pool_t *)arg;
while(1){
pool_task *task = NULL;
task = dequeue_task(pool);
/*调用回调函数,执行任务*/
if(task != NULL){
printf ("thread 0x%lx is busy\n", pthread_self());
task->process(task->arg);
free(task);
task = NULL;
}
}
/*这一句应该是不可达的*/
pthread_exit(NULL);
return NULL;
}
运行结果:
1、Makefile编译后执行./wbserver
2、客服端http请求:(根据自己的服务端进行IP修改,端口号可以在主函数里更改)
http://172.16.1.112:13000/html/index.html
下载
https://download.csdn.net/download/u010872301/10432781