mirror of
https://git.videolan.org/git/ffmpeg.git
synced 2024-10-04 09:37:53 +02:00
lavf/http: Implement server side network code.
add http_accept, add http_handshake and move handshake logic there, handle connection closing. Signed-off-by: Stephan Holljes <klaxa1337@googlemail.com>
This commit is contained in:
parent
75235a2550
commit
3240e69de6
@ -25,6 +25,7 @@
|
|||||||
#include <zlib.h>
|
#include <zlib.h>
|
||||||
#endif /* CONFIG_ZLIB */
|
#endif /* CONFIG_ZLIB */
|
||||||
|
|
||||||
|
#include "libavutil/avassert.h"
|
||||||
#include "libavutil/avstring.h"
|
#include "libavutil/avstring.h"
|
||||||
#include "libavutil/opt.h"
|
#include "libavutil/opt.h"
|
||||||
|
|
||||||
@ -44,6 +45,14 @@
|
|||||||
* path names). */
|
* path names). */
|
||||||
#define BUFFER_SIZE MAX_URL_SIZE
|
#define BUFFER_SIZE MAX_URL_SIZE
|
||||||
#define MAX_REDIRECTS 8
|
#define MAX_REDIRECTS 8
|
||||||
|
#define HTTP_SINGLE 1
|
||||||
|
#define HTTP_MUTLI 2
|
||||||
|
typedef enum {
|
||||||
|
LOWER_PROTO,
|
||||||
|
READ_HEADERS,
|
||||||
|
WRITE_REPLY_HEADERS,
|
||||||
|
FINISH
|
||||||
|
}HandshakeState;
|
||||||
|
|
||||||
typedef struct HTTPContext {
|
typedef struct HTTPContext {
|
||||||
const AVClass *class;
|
const AVClass *class;
|
||||||
@ -97,6 +106,11 @@ typedef struct HTTPContext {
|
|||||||
char *method;
|
char *method;
|
||||||
int reconnect;
|
int reconnect;
|
||||||
int listen;
|
int listen;
|
||||||
|
char *resource;
|
||||||
|
int reply_code;
|
||||||
|
int is_multi_client;
|
||||||
|
HandshakeState handshake_step;
|
||||||
|
int is_connected_server;
|
||||||
} HTTPContext;
|
} HTTPContext;
|
||||||
|
|
||||||
#define OFFSET(x) offsetof(HTTPContext, x)
|
#define OFFSET(x) offsetof(HTTPContext, x)
|
||||||
@ -128,7 +142,9 @@ static const AVOption options[] = {
|
|||||||
{ "end_offset", "try to limit the request to bytes preceding this offset", OFFSET(end_off), AV_OPT_TYPE_INT64, { .i64 = 0 }, 0, INT64_MAX, D },
|
{ "end_offset", "try to limit the request to bytes preceding this offset", OFFSET(end_off), AV_OPT_TYPE_INT64, { .i64 = 0 }, 0, INT64_MAX, D },
|
||||||
{ "method", "Override the HTTP method or set the expected HTTP method from a client", OFFSET(method), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, D | E },
|
{ "method", "Override the HTTP method or set the expected HTTP method from a client", OFFSET(method), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, D | E },
|
||||||
{ "reconnect", "auto reconnect after disconnect before EOF", OFFSET(reconnect), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, D },
|
{ "reconnect", "auto reconnect after disconnect before EOF", OFFSET(reconnect), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, D },
|
||||||
{ "listen", "listen on HTTP", OFFSET(listen), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, D | E },
|
{ "listen", "listen on HTTP", OFFSET(listen), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 2, D | E },
|
||||||
|
{ "resource", "The resource requested by a client", OFFSET(resource), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, E },
|
||||||
|
{ "reply_code", "The http status code to return to a client", OFFSET(reply_code), AV_OPT_TYPE_INT, { .i64 = 200}, INT_MIN, 599, E},
|
||||||
{ NULL }
|
{ NULL }
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -299,51 +315,143 @@ int ff_http_averror(int status_code, int default_averror)
|
|||||||
return default_averror;
|
return default_averror;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int http_write_reply(URLContext* h, int status_code)
|
||||||
|
{
|
||||||
|
int ret, body = 0, reply_code, message_len;
|
||||||
|
const char *reply_text, *content_type;
|
||||||
|
HTTPContext *s = h->priv_data;
|
||||||
|
char message[BUFFER_SIZE];
|
||||||
|
content_type = "text/plain";
|
||||||
|
|
||||||
|
if (status_code < 0)
|
||||||
|
body = 1;
|
||||||
|
switch (status_code) {
|
||||||
|
case AVERROR_HTTP_BAD_REQUEST:
|
||||||
|
case 400:
|
||||||
|
reply_code = 400;
|
||||||
|
reply_text = "Bad Request";
|
||||||
|
break;
|
||||||
|
case AVERROR_HTTP_FORBIDDEN:
|
||||||
|
case 403:
|
||||||
|
reply_code = 403;
|
||||||
|
reply_text = "Forbidden";
|
||||||
|
break;
|
||||||
|
case AVERROR_HTTP_NOT_FOUND:
|
||||||
|
case 404:
|
||||||
|
reply_code = 404;
|
||||||
|
reply_text = "Not Found";
|
||||||
|
break;
|
||||||
|
case 200:
|
||||||
|
reply_code = 200;
|
||||||
|
reply_text = "OK";
|
||||||
|
content_type = "application/octet-stream";
|
||||||
|
break;
|
||||||
|
case AVERROR_HTTP_SERVER_ERROR:
|
||||||
|
case 500:
|
||||||
|
reply_code = 500;
|
||||||
|
reply_text = "Internal server error";
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
return AVERROR(EINVAL);
|
||||||
|
}
|
||||||
|
if (body) {
|
||||||
|
s->chunked_post = 0;
|
||||||
|
message_len = snprintf(message, sizeof(message),
|
||||||
|
"HTTP/1.1 %03d %s\r\n"
|
||||||
|
"Content-Type: %s\r\n"
|
||||||
|
"Content-Length: %zu\r\n"
|
||||||
|
"\r\n"
|
||||||
|
"%03d %s\r\n",
|
||||||
|
reply_code,
|
||||||
|
reply_text,
|
||||||
|
content_type,
|
||||||
|
strlen(reply_text) + 6, // 3 digit status code + space + \r\n
|
||||||
|
reply_code,
|
||||||
|
reply_text);
|
||||||
|
} else {
|
||||||
|
s->chunked_post = 1;
|
||||||
|
message_len = snprintf(message, sizeof(message),
|
||||||
|
"HTTP/1.1 %03d %s\r\n"
|
||||||
|
"Content-Type: %s\r\n"
|
||||||
|
"Transfer-Encoding: chunked\r\n"
|
||||||
|
"\r\n",
|
||||||
|
reply_code,
|
||||||
|
reply_text,
|
||||||
|
content_type);
|
||||||
|
}
|
||||||
|
av_log(h, AV_LOG_TRACE, "HTTP reply header: \n%s----\n", message);
|
||||||
|
if ((ret = ffurl_write(s->hd, message, message_len)) < 0)
|
||||||
|
return ret;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
static void handle_http_errors(URLContext *h, int error)
|
static void handle_http_errors(URLContext *h, int error)
|
||||||
{
|
{
|
||||||
static const char bad_request[] = "HTTP/1.1 400 Bad Request\r\nContent-Type: text/plain\r\n\r\n400 Bad Request\r\n";
|
av_assert0(error < 0);
|
||||||
static const char internal_server_error[] = "HTTP/1.1 500 Internal server error\r\nContent-Type: text/plain\r\n\r\n500 Internal server error\r\n";
|
http_write_reply(h, error);
|
||||||
HTTPContext *s = h->priv_data;
|
}
|
||||||
if (h->is_connected) {
|
|
||||||
switch(error) {
|
static int http_handshake(URLContext *c)
|
||||||
case AVERROR_HTTP_BAD_REQUEST:
|
{
|
||||||
ffurl_write(s->hd, bad_request, strlen(bad_request));
|
int ret, err, new_location;
|
||||||
break;
|
HTTPContext *ch = c->priv_data;
|
||||||
default:
|
URLContext *cl = ch->hd;
|
||||||
av_log(h, AV_LOG_ERROR, "Unhandled HTTP error.\n");
|
switch (ch->handshake_step) {
|
||||||
ffurl_write(s->hd, internal_server_error, strlen(internal_server_error));
|
case LOWER_PROTO:
|
||||||
|
av_log(c, AV_LOG_TRACE, "Lower protocol\n");
|
||||||
|
if ((ret = ffurl_handshake(cl) > 0))
|
||||||
|
return 2 + ret;
|
||||||
|
if ((ret < 0))
|
||||||
|
return ret;
|
||||||
|
ch->handshake_step = READ_HEADERS;
|
||||||
|
ch->is_connected_server = 1;
|
||||||
|
return 2;
|
||||||
|
case READ_HEADERS:
|
||||||
|
av_log(c, AV_LOG_TRACE, "Read headers\n");
|
||||||
|
if ((err = http_read_header(c, &new_location)) < 0) {
|
||||||
|
handle_http_errors(c, err);
|
||||||
|
return err;
|
||||||
}
|
}
|
||||||
|
ch->handshake_step = WRITE_REPLY_HEADERS;
|
||||||
|
return 1;
|
||||||
|
case WRITE_REPLY_HEADERS:
|
||||||
|
av_log(c, AV_LOG_TRACE, "Reply code: %d\n", ch->reply_code);
|
||||||
|
if ((err = http_write_reply(c, ch->reply_code)) < 0)
|
||||||
|
return err;
|
||||||
|
ch->handshake_step = FINISH;
|
||||||
|
return 1;
|
||||||
|
case FINISH:
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
// this should never be reached.
|
||||||
|
return AVERROR(EINVAL);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int http_listen(URLContext *h, const char *uri, int flags,
|
static int http_listen(URLContext *h, const char *uri, int flags,
|
||||||
AVDictionary **options) {
|
AVDictionary **options) {
|
||||||
HTTPContext *s = h->priv_data;
|
HTTPContext *s = h->priv_data;
|
||||||
int ret;
|
int ret;
|
||||||
static const char header[] = "HTTP/1.1 200 OK\r\nContent-Type: application/octet-stream\r\nTransfer-Encoding: chunked\r\n\r\n";
|
|
||||||
char hostname[1024], proto[10];
|
char hostname[1024], proto[10];
|
||||||
char lower_url[100];
|
char lower_url[100];
|
||||||
const char *lower_proto = "tcp";
|
const char *lower_proto = "tcp";
|
||||||
int port, new_location;
|
int port;
|
||||||
s->chunked_post = 1;
|
|
||||||
av_url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), &port,
|
av_url_split(proto, sizeof(proto), NULL, 0, hostname, sizeof(hostname), &port,
|
||||||
NULL, 0, uri);
|
NULL, 0, uri);
|
||||||
if (!strcmp(proto, "https"))
|
if (!strcmp(proto, "https"))
|
||||||
lower_proto = "tls";
|
lower_proto = "tls";
|
||||||
ff_url_join(lower_url, sizeof(lower_url), lower_proto, NULL, hostname, port,
|
ff_url_join(lower_url, sizeof(lower_url), lower_proto, NULL, hostname, port,
|
||||||
NULL);
|
NULL);
|
||||||
av_dict_set(options, "listen", "1", 0);
|
if ((ret = av_dict_set_int(options, "listen", s->listen, 0)) < 0)
|
||||||
|
goto fail;
|
||||||
if ((ret = ffurl_open(&s->hd, lower_url, AVIO_FLAG_READ_WRITE,
|
if ((ret = ffurl_open(&s->hd, lower_url, AVIO_FLAG_READ_WRITE,
|
||||||
&h->interrupt_callback, options)) < 0)
|
&h->interrupt_callback, options)) < 0)
|
||||||
goto fail;
|
goto fail;
|
||||||
if ((ret = http_read_header(h, &new_location)) < 0)
|
s->handshake_step = LOWER_PROTO;
|
||||||
goto fail;
|
if (s->listen == HTTP_SINGLE) { /* single client */
|
||||||
if ((ret = ffurl_write(s->hd, header, strlen(header))) < 0)
|
s->reply_code = 200;
|
||||||
goto fail;
|
while ((ret = http_handshake(h)) > 0);
|
||||||
return 0;
|
}
|
||||||
|
|
||||||
fail:
|
fail:
|
||||||
handle_http_errors(h, ret);
|
|
||||||
av_dict_free(&s->chained_options);
|
av_dict_free(&s->chained_options);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
@ -389,6 +497,26 @@ static int http_open(URLContext *h, const char *uri, int flags,
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int http_accept(URLContext *s, URLContext **c)
|
||||||
|
{
|
||||||
|
int ret;
|
||||||
|
HTTPContext *sc = s->priv_data;
|
||||||
|
HTTPContext *cc;
|
||||||
|
URLContext *sl = sc->hd;
|
||||||
|
URLContext *cl = NULL;
|
||||||
|
|
||||||
|
av_assert0(sc->listen);
|
||||||
|
if ((ret = ffurl_alloc(c, s->filename, s->flags, &sl->interrupt_callback)) < 0)
|
||||||
|
goto fail;
|
||||||
|
cc = (*c)->priv_data;
|
||||||
|
if ((ret = ffurl_accept(sl, &cl)) < 0)
|
||||||
|
goto fail;
|
||||||
|
cc->hd = cl;
|
||||||
|
cc->is_multi_client = 1;
|
||||||
|
fail:
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
static int http_getc(HTTPContext *s)
|
static int http_getc(HTTPContext *s)
|
||||||
{
|
{
|
||||||
int len;
|
int len;
|
||||||
@ -583,7 +711,7 @@ static int process_line(URLContext *h, char *line, int line_count,
|
|||||||
|
|
||||||
p = line;
|
p = line;
|
||||||
if (line_count == 0) {
|
if (line_count == 0) {
|
||||||
if (s->listen) {
|
if (s->is_connected_server) {
|
||||||
// HTTP method
|
// HTTP method
|
||||||
method = p;
|
method = p;
|
||||||
while (!av_isspace(*p))
|
while (!av_isspace(*p))
|
||||||
@ -604,6 +732,8 @@ static int process_line(URLContext *h, char *line, int line_count,
|
|||||||
"(%s autodetected %s received)\n", auto_method, method);
|
"(%s autodetected %s received)\n", auto_method, method);
|
||||||
return ff_http_averror(400, AVERROR(EIO));
|
return ff_http_averror(400, AVERROR(EIO));
|
||||||
}
|
}
|
||||||
|
if (!(s->method = av_strdup(method)))
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
}
|
}
|
||||||
|
|
||||||
// HTTP resource
|
// HTTP resource
|
||||||
@ -614,6 +744,8 @@ static int process_line(URLContext *h, char *line, int line_count,
|
|||||||
p++;
|
p++;
|
||||||
*(p++) = '\0';
|
*(p++) = '\0';
|
||||||
av_log(h, AV_LOG_TRACE, "Requested resource: %s\n", resource);
|
av_log(h, AV_LOG_TRACE, "Requested resource: %s\n", resource);
|
||||||
|
if (!(s->resource = av_strdup(resource)))
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
|
|
||||||
// HTTP version
|
// HTTP version
|
||||||
while (av_isspace(*p))
|
while (av_isspace(*p))
|
||||||
@ -1353,6 +1485,8 @@ HTTP_CLASS(http);
|
|||||||
URLProtocol ff_http_protocol = {
|
URLProtocol ff_http_protocol = {
|
||||||
.name = "http",
|
.name = "http",
|
||||||
.url_open2 = http_open,
|
.url_open2 = http_open,
|
||||||
|
.url_accept = http_accept,
|
||||||
|
.url_handshake = http_handshake,
|
||||||
.url_read = http_read,
|
.url_read = http_read,
|
||||||
.url_write = http_write,
|
.url_write = http_write,
|
||||||
.url_seek = http_seek,
|
.url_seek = http_seek,
|
||||||
|
Loading…
Reference in New Issue
Block a user