mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
Refine SrsHttpUri.
This commit is contained in:
winlin
parent
b285029e15
commit
6e9bba4220
2 changed files with 120 additions and 142 deletions
|
|
@ -819,6 +819,96 @@ char* ISrsHttpMessage::http_ts_send_buffer()
|
||||||
return _http_ts_send_buffer;
|
return _http_ts_send_buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
SrsHttpUri::SrsHttpUri()
|
||||||
|
{
|
||||||
|
port = SRS_DEFAULT_HTTP_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsHttpUri::~SrsHttpUri()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t SrsHttpUri::initialize(string _url)
|
||||||
|
{
|
||||||
|
srs_error_t err = srs_success;
|
||||||
|
|
||||||
|
schema = host = path = query = "";
|
||||||
|
|
||||||
|
url = _url;
|
||||||
|
const char* purl = url.c_str();
|
||||||
|
|
||||||
|
http_parser_url hp_u;
|
||||||
|
int r0;
|
||||||
|
if((r0 = http_parser_parse_url(purl, url.length(), 0, &hp_u)) != 0){
|
||||||
|
return srs_error_new(ERROR_HTTP_PARSE_URI, "parse url %s failed, code=%d", purl, r0);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string field = get_uri_field(url, &hp_u, UF_SCHEMA);
|
||||||
|
if(!field.empty()){
|
||||||
|
schema = field;
|
||||||
|
}
|
||||||
|
|
||||||
|
host = get_uri_field(url, &hp_u, UF_HOST);
|
||||||
|
|
||||||
|
field = get_uri_field(url, &hp_u, UF_PORT);
|
||||||
|
if(!field.empty()){
|
||||||
|
port = atoi(field.c_str());
|
||||||
|
}
|
||||||
|
if(port<=0){
|
||||||
|
port = SRS_DEFAULT_HTTP_PORT;
|
||||||
|
}
|
||||||
|
|
||||||
|
path = get_uri_field(url, &hp_u, UF_PATH);
|
||||||
|
query = get_uri_field(url, &hp_u, UF_QUERY);
|
||||||
|
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_url()
|
||||||
|
{
|
||||||
|
return url;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_schema()
|
||||||
|
{
|
||||||
|
return schema;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_host()
|
||||||
|
{
|
||||||
|
return host;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpUri::get_port()
|
||||||
|
{
|
||||||
|
return port;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_path()
|
||||||
|
{
|
||||||
|
return path;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_query()
|
||||||
|
{
|
||||||
|
return query;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpUri::get_uri_field(string uri, void* php_u, int ifield)
|
||||||
|
{
|
||||||
|
http_parser_url* hp_u = (http_parser_url*)php_u;
|
||||||
|
http_parser_url_fields field = (http_parser_url_fields)ifield;
|
||||||
|
|
||||||
|
if((hp_u->field_set & (1 << field)) == 0){
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
int offset = hp_u->field_data[field].off;
|
||||||
|
int len = hp_u->field_data[field].len;
|
||||||
|
|
||||||
|
return uri.substr(offset, len);
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
/* Based on src/http/ngx_http_parse.c from NGINX copyright Igor Sysoev
|
/* Based on src/http/ngx_http_parse.c from NGINX copyright Igor Sysoev
|
||||||
|
|
@ -2997,116 +3087,3 @@ http_body_is_final(const struct http_parser *parser) {
|
||||||
return parser->state == s_message_done;
|
return parser->state == s_message_done;
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
The MIT License (MIT)
|
|
||||||
|
|
||||||
Copyright (c) 2013-2019 Winlin
|
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
||||||
this software and associated documentation files (the "Software"), to deal in
|
|
||||||
the Software without restriction, including without limitation the rights to
|
|
||||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
||||||
the Software, and to permit persons to whom the Software is furnished to do so,
|
|
||||||
subject to the following conditions:
|
|
||||||
|
|
||||||
The above copyright notice and this permission notice shall be included in all
|
|
||||||
copies or substantial portions of the Software.
|
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
||||||
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
||||||
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
||||||
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
||||||
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
||||||
*/
|
|
||||||
SrsHttpUri::SrsHttpUri()
|
|
||||||
{
|
|
||||||
port = SRS_DEFAULT_HTTP_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsHttpUri::~SrsHttpUri()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t SrsHttpUri::initialize(string _url)
|
|
||||||
{
|
|
||||||
srs_error_t err = srs_success;
|
|
||||||
|
|
||||||
schema = host = path = query = "";
|
|
||||||
|
|
||||||
url = _url;
|
|
||||||
const char* purl = url.c_str();
|
|
||||||
|
|
||||||
http_parser_url hp_u;
|
|
||||||
int r0;
|
|
||||||
if((r0 = http_parser_parse_url(purl, url.length(), 0, &hp_u)) != 0){
|
|
||||||
return srs_error_new(ERROR_HTTP_PARSE_URI, "parse url %s failed, code=%d", purl, r0);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string field = get_uri_field(url, &hp_u, UF_SCHEMA);
|
|
||||||
if(!field.empty()){
|
|
||||||
schema = field;
|
|
||||||
}
|
|
||||||
|
|
||||||
host = get_uri_field(url, &hp_u, UF_HOST);
|
|
||||||
|
|
||||||
field = get_uri_field(url, &hp_u, UF_PORT);
|
|
||||||
if(!field.empty()){
|
|
||||||
port = atoi(field.c_str());
|
|
||||||
}
|
|
||||||
if(port<=0){
|
|
||||||
port = SRS_DEFAULT_HTTP_PORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
path = get_uri_field(url, &hp_u, UF_PATH);
|
|
||||||
query = get_uri_field(url, &hp_u, UF_QUERY);
|
|
||||||
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_url()
|
|
||||||
{
|
|
||||||
return url;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_schema()
|
|
||||||
{
|
|
||||||
return schema;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_host()
|
|
||||||
{
|
|
||||||
return host;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpUri::get_port()
|
|
||||||
{
|
|
||||||
return port;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_path()
|
|
||||||
{
|
|
||||||
return path;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_query()
|
|
||||||
{
|
|
||||||
return query;
|
|
||||||
}
|
|
||||||
|
|
||||||
string SrsHttpUri::get_uri_field(string uri, void* php_u, int ifield)
|
|
||||||
{
|
|
||||||
http_parser_url* hp_u = (http_parser_url*)php_u;
|
|
||||||
http_parser_url_fields field = (http_parser_url_fields)ifield;
|
|
||||||
|
|
||||||
if((hp_u->field_set & (1 << field)) == 0){
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
int offset = hp_u->field_data[field].off;
|
|
||||||
int len = hp_u->field_data[field].len;
|
|
||||||
|
|
||||||
return uri.substr(offset, len);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -508,10 +508,40 @@ public:
|
||||||
virtual bool is_jsonp() = 0;
|
virtual bool is_jsonp() = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Used to resolve the http uri.
|
||||||
|
class SrsHttpUri
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
std::string url;
|
||||||
|
std::string schema;
|
||||||
|
std::string host;
|
||||||
|
int port;
|
||||||
|
std::string path;
|
||||||
|
std::string query;
|
||||||
|
public:
|
||||||
|
SrsHttpUri();
|
||||||
|
virtual ~SrsHttpUri();
|
||||||
|
public:
|
||||||
|
// Initialize the http uri.
|
||||||
|
virtual srs_error_t initialize(std::string _url);
|
||||||
|
public:
|
||||||
|
virtual std::string get_url();
|
||||||
|
virtual std::string get_schema();
|
||||||
|
virtual std::string get_host();
|
||||||
|
virtual int get_port();
|
||||||
|
virtual std::string get_path();
|
||||||
|
virtual std::string get_query();
|
||||||
|
private:
|
||||||
|
// Get the parsed url field.
|
||||||
|
// @return return empty string if not set.
|
||||||
|
virtual std::string get_uri_field(std::string uri, void* hp_u, int field);
|
||||||
|
};
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// The http-parser is license under MIT at https://github.com/nodejs/http-parser/blob/master/LICENSE-MIT
|
// The http-parser is license under MIT at https://github.com/nodejs/http-parser/blob/master/LICENSE-MIT
|
||||||
// Version: 2.1 from https://github.com/nodejs/http-parser/releases/tag/v2.1
|
// Version: 2.1 from https://github.com/nodejs/http-parser/releases/tag/v2.1
|
||||||
|
// File: https://github.com/nodejs/http-parser/blob/80819384450b5511a3d1c424dd92a5843c891364/http_parser.h
|
||||||
|
|
||||||
//Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
//Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
||||||
//
|
//
|
||||||
|
|
@ -809,34 +839,5 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Used to resolve the http uri.
|
|
||||||
class SrsHttpUri
|
|
||||||
{
|
|
||||||
private:
|
|
||||||
std::string url;
|
|
||||||
std::string schema;
|
|
||||||
std::string host;
|
|
||||||
int port;
|
|
||||||
std::string path;
|
|
||||||
std::string query;
|
|
||||||
public:
|
|
||||||
SrsHttpUri();
|
|
||||||
virtual ~SrsHttpUri();
|
|
||||||
public:
|
|
||||||
// Initialize the http uri.
|
|
||||||
virtual srs_error_t initialize(std::string _url);
|
|
||||||
public:
|
|
||||||
virtual std::string get_url();
|
|
||||||
virtual std::string get_schema();
|
|
||||||
virtual std::string get_host();
|
|
||||||
virtual int get_port();
|
|
||||||
virtual std::string get_path();
|
|
||||||
virtual std::string get_query();
|
|
||||||
private:
|
|
||||||
// Get the parsed url field.
|
|
||||||
// @return return empty string if not set.
|
|
||||||
virtual std::string get_uri_field(std::string uri, void* hp_u, int field);
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue