C++实现简易WebServer

#include <winsock.h>
#include <sys/stat.h>
#include <iostream>
using namespace std;
#define SERVER_PORT 10000         //自定义的服务端口
#define HOSTLEN 256           //主机名长度
#define BACKLOG 10           //同时等待的连接个数
/**************************************
该方法包装了send()
通过该方法发送数据能够全部发出
没有遗漏
**************************************/
int sendall(int s, char *buf, int *len) {
int total = 0;           // 已经发送字节数
int bytesleft = *len;                                   //还剩余多少字节
int n;
while(total < *len) {
  n = send(s, buf+total, bytesleft, 0);
  if (n == -1) { break; }
  total += n;
  bytesleft -= n;
}
*len = total;           // 返回实际发送出去的字节数
return n==-1?-1:0;          // 成功发送返回0 失败-1
}
/**************************************
该方法处理错误请求
并向客户端发送错误信息
**************************************/
void wrong_req(int sock) {
char* error_head = "HTTP/1.0 501 Not Implemented\r\n"; //输出501错误
int len = strlen(error_head);
if (sendall(sock, error_head, &len) == -1) {   //向客户发送
  printf("Sending failed!");
  return;
}
char* error_type = "Content-type: text/plain\r\n";
len = strlen(error_type);
if (sendall(sock, error_type, &len) == -1) {
  printf("Sending failed!");
  return;
}
char* error_end = "\r\n";
len = strlen(error_end);
if (sendall(sock, error_end, &len) == -1) {
  printf("Sending failed!");
  return;
}
char* prompt_info = "The command is not yet completed\r\n";
len = strlen(prompt_info);
if (sendall(sock, prompt_info, &len) == -1) {
  printf("Sending failed!");
  return;
}
}
/**********************************
该方法判断用户请求的文件是否存在
不存在返回true 存在返回false
***********************************/
bool not_exit(char* arguments) {
struct stat dir_info;
return (stat(arguments, &dir_info) == -1);
}
/*************************************
所请求的文件不存在
*************************************/
void file_not_found(char* arguments, int sock) {
char* error_head = "HTTP/1.0 404 Not Found\r\n";   //构造404错误head
int len = strlen(error_head);
if (sendall(sock, error_head, &len) == -1) {    //向客户端发送
  printf("Sending error!");
  return;
}
char* error_type = "Content-type: text/plain\r\n";
len = strlen(error_type);
if (sendall(sock, error_type, &len) == -1) {
  printf("Sending error!");
  return;
}
char* error_end = "\r\n";
len = strlen(error_end);
if (sendall(sock, error_end, &len) == -1) {
  printf("Sending error!");
  return;
}
char prompt_info[50] = "Not found:  ";
strcat(prompt_info, arguments);
len = strlen(prompt_info);
if (sendall(sock, prompt_info, &len) == -1) {    //输出未找到的文件
  printf("Sending error!");
  return;
}
}
/*************************************
发送Http协议头部信息
其中包括响应类型和Content Type
*************************************/
void send_header(int send_to, char* content_type) {

char* head = "HTTP/1.0 200 OK\r\n";     //正确的头部信息
int len = strlen(head);
if (sendall(send_to, head, &len) == -1) {   //向连接的客户端发送数据
  printf("Sending error");
  return;
}
if (content_type) {         //content_type不为空
  char temp_1[30] = "Content-type: ";    //准备好要连接的字串
  strcat(temp_1, content_type);     //构造content_type
  strcat(temp_1, "\r\n");
  len = strlen(temp_1);
  if (sendall(send_to, temp_1, &len) == -1) {
   printf("Sending error!");
   return;
  }
}
}
/***********************************
取得用户所请求的文件类型
即文件后缀 (.html .jpg .gif)
************************************/
char* file_type(char* arg) {
char * temp;          //临时字符串指针
if ((temp=strrchr(arg,'.')) != NULL) {    //取得后缀
  return temp+1;
}
return "";           //如果请求的文件名中没有. 则返回空串
}
/*************************************
该方法为程序核心
负责真正发送文件如*.html *.jpg等
*************************************/
void send_file(char* arguments, int sock) {
char* extension = file_type(arguments);    //获得文件后缀名
char* content_type = "text/plain";     //初始化type='text/plain'
FILE* read_from;         //本地文件指针从该文件中读取.html .jpg等
int readed = -1;         //每次读得的字节数

if (strcmp(extension, "html") == 0) {    //发送内容为html
  content_type = "text/html";
}
if (strcmp(extension, "gif") == 0) {    //发送内容为gif
  content_type = "image/gif";
}
if (strcmp(extension, "jpg") == 0) {    //发送内容为jpg
  content_type = "image/jpg";
}
read_from = fopen(arguments, "r");     //打开用户指定的文件准备读取
if(read_from != NULL) {        //指针不为空
  char read_buf[128];        //读文件时的字节缓存数组
  send_header(sock, content_type);    //发送协议头
  send(sock, "\r\n", 2, 0);      //再加一个"\r\n" 不能缺少格式要求
  while(!feof(read_from)) {      //判断文件是否已经结束
   fgets(read_buf, 128, read_from);   //读取
   int len = strlen(read_buf);
   if (sendall(sock, read_buf, &len) == -1) { //发送数据
    printf("Sending error!");    //出现发送错误显示到控制台继续发送
    continue;
   }
  }
}
}
/***********************************
解析并处理用户请求
***********************************/
void handle_req(char* request, int client_sock) {
char command[BUFSIZ];        //保存解析到的命令字段 GET PUT
char arguments[BUFSIZ];        //保存解析到的请求的文件
/*
  * 在用户请求前加上当前目录符号
  */
strcpy(arguments, "./");       //注意该符号在不同操作系统的区别
/*
  * 解析请求
  */
if (sscanf(request, "%s%s", command, arguments+2) != 2) {
  return;           //解析出错在返回
}

printf("handle_cmd:    %s\n",command);    //向控制台输出此时的命令
printf("handle_path:   %s\n",arguments);   //向控制台输出此时的请求路径

if (strcmp(command, "GET") != 0) {     //请求命令格式是否正确
  wrong_req(client_sock);
  return;
}
if (not_exit(arguments)) {       //请求的文件是否存在
  file_not_found(arguments, client_sock);
  return;
}
send_file(arguments, client_sock);     //命令格式及请求路径正确则发送数据

return;
}
/*************************************
该方法构造服务器端的SOCKET
返回构造好的socket描述符
*************************************/
int make_server_socket() {
struct sockaddr_in server_addr;       //服务器地址结构体
int tempSockId;           //临时存储socket描述符
tempSockId = socket(PF_INET, SOCK_STREAM, 0);

if (tempSockId == -1) {         //如果返回值为－1 则出错
  return -1;
}
/*
  * 填充服务器连接信息
  */
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(SERVER_PORT);
server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");  //本地地址
memset(&(server_addr.sin_zero), '\0', 8);
if (bind(tempSockId, (struct sockaddr *)&server_addr,
  sizeof(server_addr)) == -1) {       //绑定服务如果出错则返回－1
  printf("bind error!\n");
  return -1;
}
if (listen(tempSockId, BACKLOG) == -1 ) {     //开始监听
  printf("listen error!\n");
  return -1;
}
return tempSockId;           //返回取得的SOCKET
}
/***********************
主函数main()
程序入口
***********************/
void main(int argc, char * argv[]) {
/*
  * 调用WSAStartup() 便于访问sockets library
  */
WSADATA wsaData;
if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {
  fprintf(stderr, "WSAStartup failed.\n");
  exit(1);
}
printf("My web server started...\n");
int server_socket;        //服务器的socket
int acc_socket;         //接收到的用户连接的socket
int sock_size = sizeof(struct sockaddr_in);
struct sockaddr_in user_socket;     //客户连接信息
server_socket = make_server_socket();   //创建服务器端的socket
if (server_socket == -1) {      //创建socket出错
  printf("Server exception!\n");
  exit(2);
}
/*
  * 主循环
  */
while(true) {
  acc_socket = accept(server_socket, (struct sockaddr *)&user_socket, &sock_size); //接收连接

  //cout << inet_ntoa(user_socket.sin_addr) << endl;    //测试用:-)//

  /*
   * 读取客户请求
   */
  int numbytes;
  char buf[100];
  if ((numbytes=recv(acc_socket, buf, 99, 0)) == -1) {
   perror("recv");
   exit(1);
  }

  //printf("buf ... %s", buf);      //测试用
  /*
   * 处理用户请求
   */
  handle_req(buf, acc_socket);
}
}
/**************程序结束Server.cpp******************/
提示一直下载---- 是不是实体头里的Content-Length设的不对
SINA的博客很多都不能在FF里面打开打开就是一片空白

调试易

C++实现简易WebServer

解决方案 »