mirror of https://github.com/ossrs/srs.git
merge from wenjie, support banwidth test.
winlin
11 years ago
parent
df96b237b8
commit
9455181790
@ -0,0 +1,444 @@
|
|||||||
|
/*
|
||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2013 wenjiegit
|
||||||
|
|
||||||
|
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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <srs_core_bandwidth.hpp>
|
||||||
|
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
#include <srs_core_rtmp.hpp>
|
||||||
|
#include <srs_core_error.hpp>
|
||||||
|
#include <srs_core_amf0.hpp>
|
||||||
|
#include <srs_core_protocol.hpp>
|
||||||
|
#include <srs_core_config.hpp>
|
||||||
|
#include <srs_core_autofree.hpp>
|
||||||
|
|
||||||
|
SrsBandwidth::SrsBandwidth()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsBandwidth::~SrsBandwidth()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsBandwidth::bandwidth_test(SrsRequest* _req, st_netfd_t stfd, SrsRtmp* _rtmp)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
rtmp = _rtmp;
|
||||||
|
req = _req;
|
||||||
|
|
||||||
|
if (!config->get_bw_check_enabled(req->vhost)) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// validate the bandwidth check key
|
||||||
|
std::string key = "key=" + config->get_bw_check_key(req->vhost);
|
||||||
|
if (req->tcUrl.find(key) == std::string::npos) {
|
||||||
|
ret = ERROR_SYSTEM_BANDWIDTH_KEY;
|
||||||
|
srs_error("check the vhost=%s %s failed, tcUrl=%s, ret=%d",
|
||||||
|
req->vhost.c_str(), key.c_str(), req->tcUrl.c_str(), ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// shared global last check time,
|
||||||
|
// to avoid attach by bandwidth check,
|
||||||
|
// if client request check in the window(specifeid by interval),
|
||||||
|
// directly reject the request.
|
||||||
|
static int64_t last_check_time = 0;
|
||||||
|
int interval_ms = config->get_bw_check_interval_ms(req->vhost);
|
||||||
|
|
||||||
|
int64_t time_now = srs_get_system_time_ms();
|
||||||
|
// reject the connection in the interval window.
|
||||||
|
if (last_check_time > 0 && time_now - last_check_time < interval_ms) {
|
||||||
|
ret = ERROR_SYSTEM_BANDWIDTH_DENIED;
|
||||||
|
srs_trace("bandcheck denied, "
|
||||||
|
"last_check=%"PRId64", now=%"PRId64", interval=%d",
|
||||||
|
last_check_time, time_now, interval_ms);
|
||||||
|
|
||||||
|
rtmp->response_connect_reject(req, "bandcheck rejected");
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// accept and do bandwidth check.
|
||||||
|
last_check_time = time_now;
|
||||||
|
|
||||||
|
char* local_ip = 0;
|
||||||
|
if ((ret = get_local_ip(stfd, local_ip)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("get local ip failed. ret = %d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((ret = rtmp->response_connect_app(req, local_ip)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("response connect app failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
return do_bandwidth_check();
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsBandwidth::get_local_ip(st_netfd_t stfd, char *&local_ip)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
int fd = st_netfd_fileno(stfd);
|
||||||
|
|
||||||
|
// discovery client information
|
||||||
|
sockaddr_in addr;
|
||||||
|
socklen_t addrlen = sizeof(addr);
|
||||||
|
if (getsockname(fd, (sockaddr*)&addr, &addrlen) == -1) {
|
||||||
|
ret = ERROR_SOCKET_GET_LOCAL_IP;
|
||||||
|
srs_error("discovery local ip information failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
srs_verbose("get local ip success.");
|
||||||
|
|
||||||
|
// ip v4 or v6
|
||||||
|
char buf[INET6_ADDRSTRLEN];
|
||||||
|
memset(buf, 0, sizeof(buf));
|
||||||
|
|
||||||
|
if ((inet_ntop(addr.sin_family, &addr.sin_addr, buf, sizeof(buf))) == NULL) {
|
||||||
|
ret = ERROR_SOCKET_GET_LOCAL_IP;
|
||||||
|
srs_error("convert local ip information failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
local_ip = new char[strlen(buf) + 1];
|
||||||
|
strcpy(local_ip, buf);
|
||||||
|
|
||||||
|
srs_verbose("get local ip of client ip=%s, fd=%d", buf, fd);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsBandwidth::do_bandwidth_check()
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
SrsProtocol* protocol = rtmp->get_protocol();
|
||||||
|
|
||||||
|
int play_duration_ms = 3000;
|
||||||
|
int play_interval_ms = 0;
|
||||||
|
int play_actual_duration_ms = 0;
|
||||||
|
int play_bytes = 0;
|
||||||
|
|
||||||
|
int publish_duration_ms = 3000;
|
||||||
|
int publish_interval_ms = 0;
|
||||||
|
int publish_actual_duration_ms = 0;
|
||||||
|
int publish_bytes = 0;
|
||||||
|
|
||||||
|
int limit_kbps = config->get_bw_check_limit_kbps(req->vhost);
|
||||||
|
|
||||||
|
int64_t start_time = srs_get_system_time_ms();
|
||||||
|
|
||||||
|
ret = check_play(play_duration_ms,
|
||||||
|
play_interval_ms, play_actual_duration_ms, play_bytes, limit_kbps);
|
||||||
|
if (ret != ERROR_SUCCESS) {
|
||||||
|
srs_error("band width play check failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = check_publish(publish_duration_ms,
|
||||||
|
publish_interval_ms, publish_actual_duration_ms, publish_bytes, limit_kbps);
|
||||||
|
if (ret != ERROR_SUCCESS) {
|
||||||
|
srs_error("band width publish check failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int64_t end_time = srs_get_system_time_ms();
|
||||||
|
int play_kbps = play_bytes * 8 / play_actual_duration_ms;
|
||||||
|
int publish_kbps = publish_bytes * 8 / publish_actual_duration_ms;
|
||||||
|
|
||||||
|
// send finished msg
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_finish();
|
||||||
|
pkt->data->set("code", new SrsAmf0Number(ERROR_SUCCESS));
|
||||||
|
pkt->data->set("start_time", new SrsAmf0Number(start_time));
|
||||||
|
pkt->data->set("end_time", new SrsAmf0Number(end_time));
|
||||||
|
pkt->data->set("play_kbps", new SrsAmf0Number(play_kbps));
|
||||||
|
pkt->data->set("publish_kbps", new SrsAmf0Number(publish_kbps));
|
||||||
|
pkt->data->set("play_bytes", new SrsAmf0Number(play_bytes));
|
||||||
|
pkt->data->set("play_time", new SrsAmf0Number(play_actual_duration_ms));
|
||||||
|
pkt->data->set("publish_bytes", new SrsAmf0Number(publish_bytes));
|
||||||
|
pkt->data->set("publish_time", new SrsAmf0Number(publish_actual_duration_ms));
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check finish message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// if flash, we notice the result, and expect a final packet.
|
||||||
|
while (true) {
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
SrsBandwidthPacket* pkt = NULL;
|
||||||
|
if ((ret = srs_rtmp_expect_message<SrsBandwidthPacket>(protocol, &msg, &pkt)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("expect final message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
srs_info("get final message succes.");
|
||||||
|
|
||||||
|
if (pkt->is_flash_final()) {
|
||||||
|
srs_trace("BW check recv flash final response.");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_trace("BW check finished.");
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsBandwidth::check_play(
|
||||||
|
int duration_ms, int interval_ms, int& actual_duration_ms,
|
||||||
|
int& play_bytes, int max_play_kbps)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
SrsProtocol* protocol = rtmp->get_protocol();
|
||||||
|
|
||||||
|
if (true) {
|
||||||
|
// send start play command to client
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_start_play();
|
||||||
|
|
||||||
|
pkt->data->set("duration_ms", new SrsAmf0Number(duration_ms));
|
||||||
|
pkt->data->set("interval_ms", new SrsAmf0Number(interval_ms));
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check start play message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
srs_trace("BW check begin.");
|
||||||
|
}
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
// recv client's starting play response
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
SrsBandwidthPacket* pkt = NULL;
|
||||||
|
if ((ret = srs_rtmp_expect_message<SrsBandwidthPacket>(protocol, &msg, &pkt)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("expect bandwidth message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
srs_info("get bandwidth message succes.");
|
||||||
|
|
||||||
|
if (pkt->is_starting_play()) {
|
||||||
|
srs_trace("BW check recv play begin response.");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// send play data to client
|
||||||
|
int64_t current_time = srs_get_system_time_ms();
|
||||||
|
int size = 1024; // TODO: FIXME: magic number
|
||||||
|
char random_data[size];
|
||||||
|
memset(random_data, 0x01, size);
|
||||||
|
|
||||||
|
int interval = 0;
|
||||||
|
while ( (srs_get_system_time_ms() - current_time) < duration_ms ) {
|
||||||
|
st_usleep(interval);
|
||||||
|
|
||||||
|
// TODO: FIXME: use shared ptr message.
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_playing();
|
||||||
|
|
||||||
|
// TODO: FIXME: magic number
|
||||||
|
for (int i = 0; i < 100; ++i) {
|
||||||
|
char buf[32]; // TODO: FIXME: magic number
|
||||||
|
sprintf(buf, "%d", i);
|
||||||
|
pkt->data->set(buf, new SrsAmf0String(random_data));
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: FIXME: get length from the rtmp protocol stack.
|
||||||
|
play_bytes += pkt->get_payload_length();
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check play messages failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// sleep while current kbps <= max_play_kbps
|
||||||
|
int kbps = 0;
|
||||||
|
while (true) {
|
||||||
|
if(srs_get_system_time_ms() - current_time != 0)
|
||||||
|
kbps = play_bytes * 8 / (srs_get_system_time_ms() - current_time);
|
||||||
|
|
||||||
|
if (kbps > max_play_kbps) {
|
||||||
|
st_usleep(500);
|
||||||
|
} else {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
actual_duration_ms = srs_get_system_time_ms() - current_time;
|
||||||
|
srs_trace("BW check send play bytes over.");
|
||||||
|
|
||||||
|
if (true) {
|
||||||
|
// notify client to stop play
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_stop_play();
|
||||||
|
pkt->data->set("duration_ms", new SrsAmf0Number(duration_ms));
|
||||||
|
pkt->data->set("interval_ms", new SrsAmf0Number(interval_ms));
|
||||||
|
pkt->data->set("duration_delta", new SrsAmf0Number(actual_duration_ms));
|
||||||
|
pkt->data->set("bytes_delta", new SrsAmf0Number(play_bytes));
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check stop play message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
srs_trace("BW check stop play bytes.");
|
||||||
|
}
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
// recv client's stop play response.
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
SrsBandwidthPacket* pkt = NULL;
|
||||||
|
if ((ret = srs_rtmp_expect_message<SrsBandwidthPacket>(protocol, &msg, &pkt)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("expect bandwidth message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
srs_info("get bandwidth message succes.");
|
||||||
|
|
||||||
|
if (pkt->is_stopped_play()) {
|
||||||
|
srs_trace("BW check recv stop play response.");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsBandwidth::check_publish(
|
||||||
|
int duration_ms, int interval_ms, int& actual_duration_ms,
|
||||||
|
int& publish_bytes, int max_pub_kbps)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
SrsProtocol* protocol = rtmp->get_protocol();
|
||||||
|
|
||||||
|
if (true) {
|
||||||
|
// notify client to start publish
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_start_publish();
|
||||||
|
|
||||||
|
pkt->data->set("duration_ms", new SrsAmf0Number(duration_ms));
|
||||||
|
pkt->data->set("interval_ms", new SrsAmf0Number(interval_ms));
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check start publish message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
srs_trace("BW check publish begin.");
|
||||||
|
}
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
// read client's notification of starting publish
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
SrsBandwidthPacket* pkt = NULL;
|
||||||
|
if ((ret = srs_rtmp_expect_message<SrsBandwidthPacket>(protocol, &msg, &pkt)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("expect bandwidth message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
srs_info("get bandwidth message succes.");
|
||||||
|
|
||||||
|
if (pkt->is_starting_publish()) {
|
||||||
|
srs_trace("BW check recv publish begin response.");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// recv publish msgs until @duration_ms ms
|
||||||
|
int64_t current_time = srs_get_system_time_ms();
|
||||||
|
while ( (srs_get_system_time_ms() - current_time) < duration_ms ) {
|
||||||
|
st_usleep(0);
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
if ((ret = rtmp->recv_message(&msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("recv message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
|
||||||
|
// TODO: FIXME.
|
||||||
|
publish_bytes += msg->header.payload_length;
|
||||||
|
|
||||||
|
int kbps = 0;
|
||||||
|
while (true) {
|
||||||
|
if(srs_get_system_time_ms() - current_time != 0)
|
||||||
|
kbps = publish_bytes * 8 / (srs_get_system_time_ms() - current_time);
|
||||||
|
|
||||||
|
if (kbps > max_pub_kbps) {
|
||||||
|
st_usleep(500);
|
||||||
|
} else {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
actual_duration_ms = srs_get_system_time_ms() - current_time;
|
||||||
|
srs_trace("BW check recv publish data over.");
|
||||||
|
|
||||||
|
if (true) {
|
||||||
|
// notify client to stop publish
|
||||||
|
SrsBandwidthPacket* pkt = SrsBandwidthPacket::create_stop_publish();
|
||||||
|
pkt->data->set("duration_ms", new SrsAmf0Number(duration_ms));
|
||||||
|
pkt->data->set("interval_ms", new SrsAmf0Number(interval_ms));
|
||||||
|
pkt->data->set("duration_delta", new SrsAmf0Number(actual_duration_ms));
|
||||||
|
pkt->data->set("bytes_delta", new SrsAmf0Number(publish_bytes));
|
||||||
|
|
||||||
|
SrsCommonMessage* msg = (new SrsCommonMessage())->set_packet(pkt, 0);
|
||||||
|
if ((ret = rtmp->send_message(msg)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("send bandwidth check stop publish message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
srs_trace("BW check stop publish bytes.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// expect client to stop publish
|
||||||
|
// if flash client, we never expect the client stop publish bytes,
|
||||||
|
// for the flash send call packet to test publish bandwidth,
|
||||||
|
// there are many many packets in the queue.
|
||||||
|
// we just ignore the packet and send the bandwidth test data.
|
||||||
|
// TODO: FIXME: check whether flash client.
|
||||||
|
while (false) {
|
||||||
|
// recv client's stop publish response.
|
||||||
|
SrsCommonMessage* msg = NULL;
|
||||||
|
SrsBandwidthPacket* pkt = NULL;
|
||||||
|
if ((ret = srs_rtmp_expect_message<SrsBandwidthPacket>(protocol, &msg, &pkt)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("expect bandwidth message failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
SrsAutoFree(SrsCommonMessage, msg, false);
|
||||||
|
srs_info("get bandwidth message succes.");
|
||||||
|
|
||||||
|
if (pkt->is_stopped_publish()) {
|
||||||
|
srs_trace("BW check recv stop publish response.");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
@ -0,0 +1,90 @@
|
|||||||
|
/*
|
||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2013 wenjiegit
|
||||||
|
|
||||||
|
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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef SRS_CORE_BANDWIDTH_HPP
|
||||||
|
#define SRS_CORE_BANDWIDTH_HPP
|
||||||
|
|
||||||
|
/*
|
||||||
|
#include <srs_core_bandwidth.hpp>
|
||||||
|
*/
|
||||||
|
#include <srs_core.hpp>
|
||||||
|
|
||||||
|
class SrsRequest;
|
||||||
|
class SrsRtmp;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* bandwidth test agent which provides the interfaces for bandwidth check.
|
||||||
|
* 1. if vhost disabled bandwidth check, ignore.
|
||||||
|
* 2. otherwise, check the key, error if verify failed.
|
||||||
|
* 3. check the interval limit, error if bandwidth in the interval window.
|
||||||
|
* 4. check the bandwidth under the max kbps.
|
||||||
|
* 5. send the bandwidth data to client.
|
||||||
|
* bandwidth workflow:
|
||||||
|
* +------------+ +----------+
|
||||||
|
* | Client | | Server |
|
||||||
|
* +-----+------+ +-----+----+
|
||||||
|
* | |
|
||||||
|
* | connect vhost------> | if vhost enable bandwidth,
|
||||||
|
* | <-----result(success) | do bandwidth check.
|
||||||
|
* | |
|
||||||
|
* | <----call(start play) | onSrsBandCheckStartPlayBytes
|
||||||
|
* | result(playing)-----> | onSrsBandCheckStartingPlayBytes
|
||||||
|
* | <-------data(playing) | onSrsBandCheckStartingPlayBytes
|
||||||
|
* | <-----call(stop play) | onSrsBandCheckStopPlayBytes
|
||||||
|
* | result(stopped)-----> | onSrsBandCheckStoppedPlayBytes
|
||||||
|
* | |
|
||||||
|
* | <-call(start publish) | onSrsBandCheckStartPublishBytes
|
||||||
|
* | result(publishing)--> | onSrsBandCheckStartingPublishBytes
|
||||||
|
* | data(publishing)----> | onSrsBandCheckStartingPublishBytes
|
||||||
|
* | <--call(stop publish) | onSrsBandCheckStopPublishBytes
|
||||||
|
* | result(stopped)-----> | onSrsBandCheckStoppedPublishBytes(1)
|
||||||
|
* | |
|
||||||
|
* | <--------------report |
|
||||||
|
* | <END> |
|
||||||
|
* 1. when flash client, server ignore the publish stopped result.
|
||||||
|
* and flash client should close connection when got the report.
|
||||||
|
*/
|
||||||
|
class SrsBandwidth
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
SrsRequest* req;
|
||||||
|
SrsRtmp* rtmp;
|
||||||
|
public:
|
||||||
|
SrsBandwidth();
|
||||||
|
virtual ~SrsBandwidth();
|
||||||
|
public:
|
||||||
|
/**
|
||||||
|
* do the bandwidth test.
|
||||||
|
*/
|
||||||
|
virtual int bandwidth_test(SrsRequest* _req, st_netfd_t stfd, SrsRtmp* _rtmp);
|
||||||
|
private:
|
||||||
|
virtual int get_local_ip(st_netfd_t stfd, char *&local_ip);
|
||||||
|
/**
|
||||||
|
* used to process band width check from client.
|
||||||
|
*/
|
||||||
|
virtual int do_bandwidth_check();
|
||||||
|
virtual int check_play(int duration_ms, int interval_ms, int& actual_duration_ms, int& play_bytes, int max_play_kbps);
|
||||||
|
virtual int check_publish(int duration_ms, int interval_ms, int& actual_duration_ms, int& publish_bytes, int max_pub_kbps);
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
||||||
Loading…
Reference in New Issue