此前微软的 Edge 浏览器被曝光存在一个严重的漏洞，可使得黑客通过该漏洞获取用户的电脑权限，进而掌控系统。漏洞的编号被标记为 CVE-2018-8629，微软也在漏洞曝光不久后发布了紧急修复补丁。
按照微软的说法，当 Edge 浏览器所使用的 JavaScript 引擎 Chakra 不能正确处理内存中的对象即会触发该问题，导致计算机出现远程代码执行漏洞，黑客可架设一个恶意网站或是渗透正当网站再诱导 Edge 用户进行访问，用户“中招”后，黑可即可取得使用者权限，进而掌控被黑的系统。
现在，软件开发商 Phoenix Technologies 的研究人员 Bruno Keith 公布了导致 Microsoft Edge 漏洞的攻击程序演示代码，概念验证代码有 71 行，会导致 out-of-bounds (OOB) 内存泄漏，虽然不会造成有害的影响，但攻击者可以修改演示漏洞代码以获得更有害的结果。
CVE-2018-8629 漏洞影响了 Windows 10、Windows Server 2016 及 Windows Server 2019 上的 Edge 浏览器，而且在 Windows Server 2016/2019 平台上被列为中度漏洞，而在 Windows 10 上则属于重大漏洞，在概念验证代码发布之后，呼吁还没进行修复的用户或企业应尽快部署相关的安全更新。
本文经授权转自开源中国。
热 文 推 荐
☞ 下一次 IT 变革：边缘计算（Edge computing）
年度重磅：《AI聚变：2018年优秀AI应用案例TOP 20》正式发布
print_r('点个好看吧！');
var_dump('点个好看吧！');
NSLog(@"点个好看吧！");
System.out.println("点个好看吧！");
console.log("点个好看吧！");
print("点个好看吧！");
printf("点个好看吧！\n");
cout << "点个好看吧！" << endl;
Console.WriteLine("点个好看吧！");
fmt.Println("点个好看吧！");
Response.Write("点个好看吧！");
alert("点个好看吧！")
echo "点个好看吧！"
点击“阅读原文”，打开 CSDN App 阅读更贴心！
喜欢就点击“好看”吧！
CSDN
微信公众号
成就一亿技术人
微信公众号
1. Util模块
这个模块是一个工具模块，主要提供HTTP协议模块所用到的一些工具函数，比如url编解码，文件读写…等。
#include "server.hpp"
#include 
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
- 133
- 134
- 135
- 136
- 137
- 138
- 139
- 140
- 141
- 142
- 143
- 144
- 145
- 146
- 147
- 148
- 149
- 150
- 151
- 152
- 153
- 154
- 155
- 156
- 157
- 158
- 159
- 160
- 161
- 162
- 163
- 164
- 165
- 166
- 167
- 168
- 169
- 170
- 171
- 172
- 173
- 174
- 175
- 176
- 177
- 178
- 179
- 180
- 181
- 182
- 183
- 184
- 185
- 186
- 187
- 188
- 189
- 190
- 191
- 192
- 193
- 194
- 195
- 196
- 197
- 198
- 199
- 200
- 201
- 202
- 203
- 204
- 205
- 206
- 207
- 208
- 209
- 210
- 211
- 212
- 213
- 214
- 215
- 216
- 217
- 218
- 219
- 220
- 221
- 222
- 223
- 224
- 225
- 226
- 227
- 228
- 229
- 230
- 231
- 232
- 233
- 234
- 235
- 236
- 237
- 238
- 239
- 240
- 241
- 242
- 243
- 244
- 245
- 246
- 247
- 248
- 249
- 250
- 251
- 252
- 253
- 254
- 255
- 256
- 257
- 258
- 259
- 260
- 261
- 262
- 263
- 264
- 265
- 266
- 267
- 268
- 269
- 270
- 271
- 272
- 273
- 274
- 275
- 276
- 277
- 278
- 279
- 280
- 281
- 282
- 283
- 284
- 285
- 286
- 287
- 288
- 289
- 290
- 291
- 292
- 293
- 294
- 295
- 296
- 297
- 298
- 299
- 300
- 301
- 302
- 303
- 304
- 305
- 306
- 307
- 308
- 309
- 310
- 311
- 312
- 313
- 314
- 315
- 316
- 317
- 318
- 319
- 320
- 321
- 322
- 323
- 324
- 325
- 326
- 327
- 328
- 329
- 330
- 331
- 332
- 333
- 334
- 335
- 336
- 337
- 338
- 339
- 340
- 341
- 342
- 343
- 344
- 345
- 346
- 347
- 348
- 349
- 350
- 351
- 352
- 353
- 354
- 355
- 356
- 357
- 358
- 359
- 360
- 361
- 362
- 363
- 364
- 365
- 366
- 367
- 368
- 369
- 370
- 371
- 372
2. HttpRequest模块
这个模块是HTTP请求数据模块，用于保存HTTP请求数据被解析后的各项请求元素信息。
HttpRequest模块:
http请求信息模块：存储HTTP请求信息要素，提供简单的功能性接口
请求信息要素：
请求行：请求方法，URL，协议版本
URL：资源路径，查询字符串
GET /search/1234?word=C++&en=utf8 HTTP/1.1
请求头部：key: value\r\nkey: value\r\n…
Content-Length: 0\r\n
正文
要素：请求方法，资源路径，查询字符串，头部字段，正文，协议版本
std:smatch保存首行使用regex正则进行解析后，所提取的数据，比如提取资源路径中的数字…
功能性接口：
- 将成员变量设置为公有成员，便于直接访问
- 提供查询字符串，以及头部字段的单个查询和获取，插入功能
- 获取正文长度
- 判断长连接&短链接Connection:close/keep-alive
//HttpRequest模块,存储Http请求信息要素,提供简单的功能性接口
class HttpRequest
{
public:
std::string _method;//请求方法
std::string _path;//资源路径
std::string _version;//协议版本
std::string _body;//请求正文
std::smatch _matches;//资源路径的正则提取数据
std::unordered_map<std::string,std::string> _headers;//头部字段
std::unordered_map<std::string,std::string> _params;//查询字符串
public:
HttpRequest():_version("Http/1.1"){}
void ReSet()
{
_method.clear();
_path.clear();
_version = "Http/1.1";
_body.clear();
std::smatch newmatches;
_matches.swap(newmatches);
_headers.clear();
_params.clear();
}
//插入头部字段
void SetHeader(const std::string& key,const std::string& val)
{
_headers.insert({key,val});
}
//判断是否存在指定头部字段
bool HasHeader(const std::string& key) const
{
auto it = _headers.find(key);
if(it == _headers.end())
{
return false;
}
return true;
}
//获取指定头部字段的值
std::string GetHeader(const std::string& key) const
{
auto it = _headers.find(key);
if(it == _headers.end())
{
return "";
}
return it->second;
}
//插入查询字符串
void SetParam(const std::string& key,const std::string& val)
{
_params.insert({key,val});
}
//判断是否有某个指定的查询字符串
bool HasParam(const std::string& key)
{
auto it = _params.find(key);
if(it == _params.end())
{
return false;
}
return true;
}
//获取指定的查询字符串
std::string GetParamr(const std::string& key)
{
auto it = _params.find(key);
if(it == _params.end())
{
return "";
}
return it->second;
}
//获取正文长度
size_t ContentLength()
{
// Content-Length: 1234\r\n
bool ret = HasHeader("Content-Length");
if(ret == false)
{
return 0;
}
return std::stol(GetHeader("Content-Length"));
}
//判断是否是短连接
bool Close() const
{
// 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
if(HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
{
return false;
}
return true;
}
};
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
3. HttpResponse模块
这个模块是HTTP响应数据模块，用于业务处理后设置并保存HTTP响应数据的的各项元素信息，最终会被按照HTTP协议响应格式组织成为响应信息发送给客户端。
HttpResponse模块:
功能：存储HTTP响应信息要素，提供简单的功能性接口
响应信息要素：
- 响应状态码
- 头部字段
- 响应正文
- 重定向信息（是否进行了重定向的标志，重定向的路径）
功能性接口：
- 为了便于成员的访问，因此将成员设置为公有成员
- 头部字段的新增，查询，获取
- 正文的设置
- 重定向的设置
- 长短连接的判断
//HttpResponse模块,存储Http响应信息要素,提供简单的功能性接口
class HttpResponse
{
public:
int _status;//响应码
bool _redirect_flag;//是否重定向
std::string _body;//正文
std::string _redirect_url;//重定向url
std::unordered_map<std::string,std::string> _headers;//头部字段
public:
HttpResponse():_status(200),_redirect_flag(false){}
HttpResponse(int status):_status(status),_redirect_flag(false){}
void ReSet()
{
_status = 200;
_redirect_flag = false;
_body.clear();
_redirect_url.clear();
_headers.clear();
}
//插入头部字段
void SetHeader(const std::string& key,const std::string& val)
{
_headers.insert({key,val});
}
//判断是否存在指定头部字段
bool HasHeader(const std::string& key)
{
auto it = _headers.find(key);
if(it == _headers.end())
{
return false;
}
return true;
}
//获取指定头部字段的值
std::string GetHeader(const std::string& key)
{
auto it = _headers.find(key);
if(it == _headers.end())
{
return "";
}
return it->second;
}
//设置正文以及正文类型
void SetContent(const std::string& body,const std::string& type = "text/html")
{
_body = body;
SetHeader("Content-Type",type);
}
//设置重定向以及重定向状态码
void SetRedirect(const std::string& url,int status = 302)
{
_redirect_flag = true;
_redirect_url = url;
_status = status;
}
//判断是否是短连接
bool Close()
{
// 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
if(HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
{
return false;
}
return true;
}
};
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
4. HttpContext模块
这个模块是一个HTTP请求接收的上下文模块，主要是为了防支在一次接收的数据中，不是一个完整的HTTP请求，则解析过程并未完成，无法进行完整的请求处理，需要在下次接收到新数据后继续根据上下文进行解析，最终得到一个HttpRequest请求信息对象，因此在请求数据的接收以及解析部分需要一个上下文来进行控制接收和处理节奏。
typedef enum{
RECV_HTTP_ERROR,
RECV_HTTP_LINE,
RECV_HTTP_HEAD,
RECV_HTTP_BODY,
RECV_HTTP_OVER
}HttpRecvStatus;
#define MAX_BYTE 8 * 1024
//HttpContext请求接收上下文模块,记录HTTP请求的接收以及处理进度
class HttpContext
{
private:
int _resp_status;//响应状态码,解析请求出错时设置
HttpRecvStatus _recv_status;//当前接收及解析的阶段状态
HttpRequest _request;//已经解析得到的请求信息
private:
//接收请求行
bool RecvHttpLine(Buffer* buf)
{
if (_recv_status != RECV_HTTP_LINE) return false;
//1. 获取一行数据，带有末尾的换行 \n \r\n
std::string line = buf->GetLineAndPop();
//2. 需要考虑的⼀些要素：缓冲区中的数据不足一行， 获取的一行数据超大
if(line.size() == 0)
{
//缓冲区中的数据不足一行，则需要判断缓冲区的可读数据⻓度，如果很长了都不足一行，这是有问题的
if(buf->ReadAbleSize() > MAX_BYTE)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 414 ;//"URI Too Long"
return false;
}
//缓冲区中数据不足一行，但是也不多，就等等新数据的到来
return true;
}
if(line.size() > MAX_BYTE)
{
if(buf->ReadAbleSize() > MAX_BYTE)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 414 ;//"URI Too Long"
return false;
}
}
bool ret = ParseHttpLine(line);
if(ret == false)
{
return false;
}
//首行处理完毕，进⼊头部获取阶段
_recv_status = RECV_HTTP_HEAD;
return true;
}
//解析请求行
bool ParseHttpLine(const std::string& line)
{
std::smatch matches;
//std::regex::icase忽略大小写
std::regex e("(GET|HEAD|POST|PUT|DELETE) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\n|\r\n)?", std::regex::icase);
bool ret = std::regex_match(line, matches, e);
if (ret == false) {
_recv_status = RECV_HTTP_ERROR;
_resp_status = 400 ;//"Bad Request"
return false;
}
//0 : GET /bitejiuyeke/login?user=xiaoming&pass=123123 HTTP/1.1
//1 : GET
//2 : /bitejiuyeke/login
//3 : user=xiaoming&pass=123123
//4 : HTTP/1.1
//请求⽅法的获取
_request._method = matches[1];
//小写转成大写,为了下面HttpServer模块中用大写的请求方法进行判断时不会因为大小写出现不匹配的情况
std::transform(_request._method.begin(), _request._method.end(), _request._method.begin(), ::toupper);
//资源路径的获取，需要进⾏URL解码操作，但是不需要+转空格
_request._path = Until::UrlDecode(matches[2],false);
//协议版本的获取
_request._version = matches[4];
//查询字符串的获取与处理
std::vector<std::string> query_string_arry;
std::string query_string = matches[3];
//查询字符串的格式 key=val&key=val....., 先以 & 符号进⾏分割，得到各个字串
Until::Split(query_string,"&",&query_string_arry);
//针对各个字串，以 = 符号进⾏分割，得到key 和val， 得到之后也需要进⾏URL解码
for(auto& str : query_string_arry)
{
size_t pos = str.find("=");
if(pos == std::string::npos)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 400 ;//"Bad Request"
return false;
}
std::string key = Until::UrlDecode(str.substr(0,pos),true);
std::string val = Until::UrlDecode(str.substr(pos + 1),true);
_request._params.insert({key,val});
}
return true;
}
//接收请求头部
bool RecvHttpHead(Buffer* buf)
{
if(_recv_status != RECV_HTTP_HEAD) return false;
//1. 一⾏一行取出数据，直到遇到空行为⽌， 头部的格式 key: val\r\nkey: val\r\n\r\n
while(1)
{
std::string line = buf->GetLineAndPop();
//2. 需要考虑的⼀些要素：缓冲区中的数据不足一行， 获取的一行数据超大
if(line.size() == 0)
{
//缓冲区中的数据不足一行，则需要判断缓冲区的可读数据⻓度，如果很长了都不足一行，这是有问题的
if(buf->ReadAbleSize() > MAX_BYTE)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 414;//"URI Too Long"
return false;
}
//缓冲区中数据不足一行，但是也不多，就等等新数据的到来
return true;
}
if(line.size() > MAX_BYTE)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 414;
return false;
}
//遇到空行头部提取结束
if(line == "\n" || line == "\r\n")
{
break;
}
bool ret = ParseHttpHead(line);
if(ret == false)
{
return false;
}
}
//头部处理完毕,进入正文获取阶段
_recv_status = RECV_HTTP_BODY;
return true;
}
//解析请求头部
bool ParseHttpHead(std::string& line)
{
//key: val\r\n
//末尾是换行则去掉换行字符
if(line.back() == '\n') line.pop_back();
//末尾是回⻋则去掉回⻋字符
if(line.back() == '\r') line.pop_back();
size_t pos = line.find(": ");
if(pos == std::string::npos)
{
_recv_status = RECV_HTTP_ERROR;
_resp_status = 400;
return false;
}
std::string key = line.substr(0,pos);
std::string val = line.substr(pos + 2);
_request.SetHeader(key,val);
return true;
}
//接收请求正文
bool RecvHttpBody(Buffer* buf)
{
if(_recv_status != RECV_HTTP_BODY) return false;
//1. 获取正文长度
size_t content_length = _request.ContentLength();
if(content_length == 0)
{
//没有正⽂，则请求接收解析完毕
_recv_status = RECV_HTTP_OVER;
return true;
}
//2. 当前已经接收了多少正文,其实就是往 _request._body 中放了多少数据了
size_t real_len = content_length - _request._body.size();//实际还需要接收的正⽂长度
//3. 接收正文放到body中，但是也要考虑当前缓冲区中的数据，是否是全部的正⽂
// 3.1 缓冲区中数据，包含了当前请求的所有正文，则取出所需的数据
if(buf->ReadAbleSize() >= real_len)
{
_request._body.append(buf->ReadPosition(),real_len);
buf->MoveReadOffset(real_len);
_recv_status = RECV_HTTP_OVER;
return true;
}
// 3.2 缓冲区中数据，⽆法满⾜当前正文的需要，数据不足，取出数据，后续等待新数据到来
_request._body.append(buf->ReadPosition(),buf->ReadAbleSize());
buf->MoveReadOffset(buf->ReadAbleSize());
return true;
}
public:
HttpContext():_recv_status(RECV_HTTP_LINE),_resp_status(200){}
void ReSet()
{
_recv_status = RECV_HTTP_LINE;
_resp_status = 200;
_request.ReSet();
}
//接收解析遇到错误时返回对应的错误码
int RespStatus()
{
return _resp_status;
}
//当前请求接收解析到那个阶段
HttpRecvStatus RecvStatus()
{
return _recv_status;
}
//返回请求解析后的HttpRequest对象
HttpRequest &Request()
{
return _request;
}
//接收并解析HTTP请求
void RecvHttpRequest(Buffer* buf)
{
//不同的状态，做不同的事情，但是这⾥不要break， 因为处理完请求⾏后，应该⽴即处理头部，⽽不是退出等新数据
switch(_recv_status)
{
case RECV_HTTP_LINE: RecvHttpLine(buf);
case RECV_HTTP_HEAD: RecvHttpHead(buf);
case RECV_HTTP_BODY: RecvHttpBody(buf);
}
return;
}
};
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
- 133
- 134
- 135
- 136
- 137
- 138
- 139
- 140
- 141
- 142
- 143
- 144
- 145
- 146
- 147
- 148
- 149
- 150
- 151
- 152
- 153
- 154
- 155
- 156
- 157
- 158
- 159
- 160
- 161
- 162
- 163
- 164
- 165
- 166
- 167
- 168
- 169
- 170
- 171
- 172
- 173
- 174
- 175
- 176
- 177
- 178
- 179
- 180
- 181
- 182
- 183
- 184
- 185
- 186
- 187
- 188
- 189
- 190
- 191
- 192
- 193
- 194
- 195
- 196
- 197
- 198
- 199
- 200
- 201
- 202
- 203
- 204
- 205
- 206
- 207
- 208
- 209
- 210
- 211
- 212
- 213
- 214
- 215
- 216
- 217
- 218
- 219
- 220
- 221
- 222
- 223
- 224
- 225
- 226
- 227
- 228
- 229
- 230
- 231
- 232
- 233
5. HttpServer模块
这个模块是最终给组件使用者提供的HTTP服务器模块了，用于以简单的接口实现HTTP服务器的搭建。
HttpServer模块内部包含有一个TcpServer对象：TcpServer对象实现服务器的搭建
HttpServer模块内部包含有两个提供给TcpServer对象的接口：连接建立成功设置上下文接口，数据处理接口。
HttpServer模块内部包含有一个hash-map表存储请求与处理函数的映射表：组件使用者向HttpServer设置哪些请求应该使用哪些函数进行处理，等TcpServer收到对应的请求就会使用对应的函数进行处理。
HttpServer模块：用于实现Http服务器的搭建
首先要给不同的请求方法的请求路径设置对应的回调函数。也就是设计一张请求路由表。
设计一张请求路由表：
表中记录了针对哪个请求，应该使用哪个函数来进行业务处理的映射关系。
当服务器收到了一个请求，就在请求路由表中，查找有没有对应请求的处理函数，如果有，则执行对应的处理函数即可。说白了，什么请求，怎么处理，由用户来设定，服务器收到了请求只需要执行函数即可。
这样做的好处：用户只需要实现业务处理函数，然后将请求与处理函数的映射关系，添加到服务器中。
而服务器只需要接收数据，解析数据，查找路由表映射关系，执行业务处理函数。
要实现简便的搭建HTTP服务器，所需要的要素和提供的功能。
要素:
- GET请求的路由映射表
- POST请求的路由映射表
- PUT请求的路由映射表
- DELETE请求的路由映射表—路由映射表记录对应请求方法的请求资源路径与对应业务处理函数映射关系–更多是功能性请求的处理
- 静态资源相对根目录-实现静态资源请求的处理
- 高性能TCP服务器—进行连接的IO操作
公有接口:
- 添加请求—处理函数映射信息(GET/POST/PUT/DELETE)
- 设置静态资源根目录
- 设置是否启动超时连接关闭
- 设置线程池中线程数量
- 启动服务器
私有接口：
- OnConnected—用于给TcpServer设置协议上下文
- OnMessage----用于进行缓冲区数据解析处理
- 请求的路由查找
- 静态资源请求查找和处理
- 功能性请求的查找和处理
- 组织响应进行回复
服务器处理流程：
- 从socket接收数据，放到接收缓冲区
- 调用OnMessage回调函数进行业务处理
- 对请求进行解析，得到了一个HttpRequest结构，包含了所有的请求要素
- 进行请求的路由查找-找到对应请求的处理方法
a. 静态资源请求–一些实体文件资源的请求，html,image.…
将静态资源文件的数据读取出来，填充到HttpResponse结构中
b. 功能性请求—在请求路由映射表中查找处理函数，找到了则执行函数
具体的业务处理，并进行HttpResponse结构的数据填充 - 对静态资源请求/功能性请求进行处理完毕后，得到了一个填充了响应信息的HttpResponse对象，组织http格式响应，进行发送。
//HttpServer模块,用于实现HttpServer服务器的搭建
const std::string html_404 = "/404.html";
const std::string home_page = "index.html";
#define DEFALT_TIMEOUT 10
class HttpServer
{
private:
//请求路由表
using Handler = std::function<void(const HttpRequest&,HttpResponse*)>;
//请求资源路径我们用正则表达式
//比如 /number/1、/number/2、/number/3 这样的资源路径
//对应的业务处理函数都是一样的。如果一个路径给没有必要。
//因此请求路径用正则表达式，比如number/d+ 这个匹配上面三个
//正则表达式可以用来判断字符串中有没有某个子串
using Handlers = std::vector<std::pair<std::regex,Handler>>;
Handlers _get_route;
Handlers _post_route;
Handlers _put_route;
Handlers _delete_route;
std::string _basedir;//静态资源根⽬录
TcpServer _server;
private:
void ErrHandler(const HttpRequest& req,HttpResponse* rsp)
{
//1. 组织⼀个错误展示页⾯
std::string body;
body += "";
body += "";
body += "";
body += "";
body += ""
;
body += "";
body += std::to_string(rsp->_status);
body += " ";
body += Until::StatusDesc(rsp->_status);
body += "
";
body += "";
body += "";
//2. 将页⾯数据，当作响应正⽂，放⼊rsp中
rsp->SetContent(body, "text/html");
}
//将HttpResponse中的要素按照http协议格式进行组织,发送
void WriteResponse(const PtrConnection& conn,const HttpRequest& req,HttpResponse* rsp)
{
//1. 先完善头部字段
if(req.Close() == true)
{
rsp->SetHeader("Connection","close");
}
else
{
rsp->SetHeader("Connection","keep-alive");
}
if(!rsp->_body.empty() && rsp->HasHeader("Content-Length") == false)
{
rsp->SetHeader("Content-Length",std::to_string(rsp->_body.size()));
}
if(!rsp->_body.empty() && rsp->HasHeader("Content-Typy") == false)
{
rsp->SetHeader("Content-Type","application/octet-stream");
}
if(rsp->_redirect_flag == true)
{
rsp->SetHeader("Location",rsp->_redirect_url);
}
//2. 将rsp中的要素，按照http协议格式进⾏组织
std::stringstream rsp_string;
rsp_string << req._version << " " << std::to_string(rsp->_status) << " " << Until::StatusDesc(rsp->_status) << "\r\n";
for(auto& header : rsp->_headers)
{
rsp_string << header.first << ": " << header.second << "\r\n";
}
rsp_string << "\r\n";
rsp_string << rsp->_body;
//3. 发送数据
conn->Send(rsp_string.str().c_str(),rsp_string.str().size());
}
bool IsFileHandler(const HttpRequest& req)
{
// 1. 必须设置了静态资源根目录
if(_basedir.empty())
{
return false;
}
// 2. 请求⽅法，必须是GET / HEAD请求⽅法
if(req._method != "GET" && req._method != "HEAD")
{
return false;
}
// 3. 请求的资源路径必须是一个合法路径
if(Until::ValidPath(req._path) == false)
{
return false;
}
// 4. 请求的资源必须存在,且是⼀个普通⽂件
// 有⼀种请求⽐较特殊 -- ⽬录：/, 这种情况给后边默认追加⼀个index.html
// /index.html /image/a.png
// 不要忘了前缀的相对根⽬录,也就是将请求路径转换为实际存在的路径 /image/a.png -> ./wwwroot/image/a.png
std::string req_path = _basedir + req._path;//为了避免直接修改请求的资源路径，因此定义⼀个临时对象
if(req._path.back() == '/')
{
//默认首页
req_path += home_page;
}
if (Until::IsRegular(req_path) == false)
{
return false;
}
return true;
}
//静态资源的请求处理 --- 将静态资源⽂件的数据读取出来，放到rsp的_body中, 并设置mime
void FileHandler(const HttpRequest& req,HttpResponse* rsp)
{
std::string req_path = _basedir + req._path;
if(req._path.back() == '/')
{
//默认首页
req_path += home_page;
}
bool ret = Until::ReadFile(req_path,&rsp->_body);
if(ret == false)//进来这里已经判断过资源肯定是存在的,否则就不会进入静态资源的处理
{
return;
}
std::string mime = Until::ExMime(req_path);
rsp->SetHeader("Content-Type",mime);
return;
}
//功能性请求的的分类处理
void Dispatcher(HttpRequest& req,HttpResponse* rsp,const Handlers& handlers)
{
//在对应请求⽅法的路由表中，查找是否含有对应资源的对应请求的处理函数，有则调⽤，没有则返回404
//思想：路由表存储的时键值对 -- 正则表达式 & 处理函数
//使⽤正则表达式，对请求的资源路径进⾏正则匹配，匹配成功就使⽤对应函数进⾏处理
// /numbers/(\d+) /numbers/12345
for(auto& handler : handlers)
{
const std::regex& re = handler.first;
const Handler& functor = handler.second;
bool ret = std::regex_match(req._path,req._matches,re);
if(ret == false)
{
continue;
}
//传⼊请求信息，和空的rsp，执⾏处理函数
return functor(req,rsp);
}
//返回错误码404页面
rsp->_status = 404;
// std::string path = _basedir + html_404;
// bool ret = Until::ReadFile(path,&rsp->_body);
// if(ret == false)//错误路径是我们自己设置肯定是存在的
// {
// return;
// }
// return rsp->SetHeader("Content-Type","text/html");
}
//路由选择
void Route(HttpRequest& req,HttpResponse* rsp)
{
//1. 对请求进⾏分辨，是⼀个静态资源请求，还是⼀个功能性请求
// 静态资源请求，则进⾏静态资源的处理
// 功能性请求，则需要通过⼏个请求路由表来确定是否有处理函数
// 既不是静态资源请求，也没有设置对应的功能性请求处理函数，就返回405
if(IsFileHandler(req))
{
//是⼀个静态资源请求, 则进⾏静态资源请求的处理
return FileHandler(req,rsp);
}
if(req._method == "GET" || req._method == "HEAD")
{
return Dispatcher(req,rsp,_get_route);
}
else if(req._method == "POST")
{
return Dispatcher(req,rsp,_post_route);
}
else if(req._method == "PUT")
{
return Dispatcher(req,rsp,_put_route);
}
else if(req._method == "DELETE")
{
return Dispatcher(req,rsp,_delete_route);
}
rsp->_status = 405;// Method Not Allowed
return;
}
//连接建立后给对应Connection对象设置一个协议上下文
void OnConnected(const PtrConnection& conn)
{
conn->SetContent(HttpContext());
}
//缓存区数据解析+处理
void OnMessage(const PtrConnection& conn,Buffer* buf)
{
while(buf->ReadAbleSize() > 0)
{
//1. 获取上下⽂
//获取在连接建立好就给每个Connection设置HttpContext上下文
HttpContext* context = conn->GetContent()->Get<HttpContext>();
//2. 通过上下⽂对缓冲区数据进⾏解析，得到HttpRequest对象
// 2.1 如果缓冲区的数据解析出错，就直接回复出错响应
// 2.2 如果解析正常，且请求已经获取完毕，才开始去进⾏处理
context->RecvHttpRequest(buf);
HttpRequest& rep = context->Request();
HttpResponse rsp(context->RespStatus());
//if(context->RecvStatus() == RECV_HTTP_ERROR)
if (context->RespStatus() >= 400)
{
//进⾏错误响应，关闭连接
//填充⼀个错误显⽰页⾯数据到rsp中
ErrHandler(rep,&rsp);
//组织响应发送给客户端
WriteResponse(conn,rep,&rsp);
//一定要做下面两步,不然出错了,关闭连接时，接收缓存区还有数据关闭连接的时候先去先处理接收缓存区数据
//但是当前上下文状态一直是RECV_HTTP_ERROR,因此每次去接收缓存区根本拿不到数据,所有在这里死循环
//造成内存资源不足,服务器奔溃退出
//因此在这里把上下文状态重置RECV_HTTP_LINE可以每次都从接收缓存区拿到数据
//直到最后接收缓存区数据不足一行,从下面退出,然后真正的去关闭连接
context->ReSet();
//这里也可以,出错了就把接收缓冲区数据清空,也就不会在多次调用了
buf->MoveReadOffset(buf->ReadAbleSize());
//关闭连接
conn->Shutdown();
return;
}
//当前请求还没有接收完整,则退出，等新数据到来再重新继续处理
if(context->RecvStatus() != RECV_HTTP_OVER)
{
return;
}
//3. 请求路由 + 业务处理
Route(rep,&rsp);
//4. 对HttpResponse进⾏组织发送
WriteResponse(conn,rep,&rsp);
//5. 重置上下⽂，避免影响下次解析
context->ReSet();
//6. 根据⻓短连接判断是否关闭连接或者继续处理
if(rsp.Close() == true)
{
//短链接则直接关闭
return conn->Shutdown();
}
}
return;
}
public:
HttpServer(uint16_t port,int timeout = DEFALT_TIMEOUT):_server(port)
{
_server.EnableInactiveRelease(timeout);
_server.SetConnectedCallback(std::bind(&HttpServer::OnConnected,this,std::placeholders::_1));
_server.SetMessageCallback(std::bind(&HttpServer::OnMessage,this,std::placeholders::_1,std::placeholders::_2));
}
void SetBaseDir(const std::string& path)
{
assert(Until::IsDirectory(path) == true);
_basedir = path;
}
void Get(const std::string& parttern,const Handler& handler)
{
_get_route.push_back(std::make_pair(std::regex(parttern),handler));
}
void Post(const std::string& parttern,const Handler& handler)
{
_post_route.push_back(std::make_pair(std::regex(parttern),handler));
}
void Put(const std::string& parttern,const Handler& handler)
{
_put_route.push_back(std::make_pair(std::regex(parttern),handler));
}
void Delete(const std::string& parttern,const Handler& handler)
{
_delete_route.push_back(std::make_pair(std::regex(parttern),handler));
}
void SetThreadCount(int count)
{
_server.SetThreadCount(count);
}
void Start()
{
_server.Start();
}
};
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
- 133
- 134
- 135
- 136
- 137
- 138
- 139
- 140
- 141
- 142
- 143
- 144
- 145
- 146
- 147
- 148
- 149
- 150
- 151
- 152
- 153
- 154
- 155
- 156
- 157
- 158
- 159
- 160
- 161
- 162
- 163
- 164
- 165
- 166
- 167
- 168
- 169
- 170
- 171
- 172
- 173
- 174
- 175
- 176
- 177
- 178
- 179
- 180
- 181
- 182
- 183
- 184
- 185
- 186
- 187
- 188
- 189
- 190
- 191
- 192
- 193
- 194
- 195
- 196
- 197
- 198
- 199
- 200
- 201
- 202
- 203
- 204
- 205
- 206
- 207
- 208
- 209
- 210
- 211
- 212
- 213
- 214
- 215
- 216
- 217
- 218
- 219
- 220
- 221
- 222
- 223
- 224
- 225
- 226
- 227
- 228
- 229
- 230
- 231
- 232
- 233
- 234
- 235
- 236
- 237
- 238
- 239
- 240
- 241
- 242
- 243
- 244
- 245
- 246
- 247
- 248
- 249
- 250
- 251
- 252
- 253
- 254
- 255
- 256
- 257
- 258
- 259
- 260
- 261
- 262
- 263
- 264
- 265
- 266
- 267
- 268
- 269
- 270
- 271
- 272
- 273
- 274
- 275
- 276
- 277
- 278
- 279
- 280
- 281
- 282
- 283
- 284
- 285
- 286
- 287
- 288
- 289
- 290
- 291
- 292
- 293
- 294
评论记录：
回复评论：