102 lines
3.0 KiB
C
102 lines
3.0 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "output.h"
|
|
#include "util/http/http.h"
|
|
#include "util/opts/log.h"
|
|
#include "device/buffer.h"
|
|
#include "device/buffer_lock.h"
|
|
|
|
#define SNAPSHOT_TIMEOUT_MS 3000
|
|
#define SNAPSHOT_DEFAULT_DELAY_PARAM 300
|
|
|
|
#define PART_BOUNDARY "123456789000000000000987654321"
|
|
#define CONTENT_TYPE "image/jpeg"
|
|
#define CONTENT_LENGTH "Content-Length"
|
|
|
|
static const char *const STREAM_HEADER = "HTTP/1.0 200 OK\r\n"
|
|
"Access-Control-Allow-Origin: *\r\n"
|
|
"Connection: close\r\n"
|
|
"Content-Type: multipart/x-mixed-replace;boundary=" PART_BOUNDARY "\r\n"
|
|
"\r\n"
|
|
"--" PART_BOUNDARY "\r\n";
|
|
static const char *const STREAM_PART = "Content-Type: " CONTENT_TYPE "\r\n" CONTENT_LENGTH ": %u\r\n\r\n";
|
|
static const char *const STREAM_BOUNDARY = "\r\n"
|
|
"--" PART_BOUNDARY "\r\n";
|
|
|
|
typedef struct
|
|
{
|
|
FILE *stream;
|
|
uint64_t start_time_us;
|
|
} http_snapshot_t;
|
|
|
|
int http_snapshot_buf_part(buffer_lock_t *buf_lock, buffer_t *buf, int frame, http_snapshot_t *snapshot)
|
|
{
|
|
// Ignore frames that are captured
|
|
if (buf->captured_time_us < snapshot->start_time_us) {
|
|
return 0;
|
|
}
|
|
|
|
fprintf(snapshot->stream, "HTTP/1.1 200 OK\r\n");
|
|
fprintf(snapshot->stream, "Content-Type: image/jpeg\r\n");
|
|
fprintf(snapshot->stream, "Content-Length: %zu\r\n", buf->used);
|
|
fprintf(snapshot->stream, "\r\n");
|
|
fwrite(buf->start, buf->used, 1, snapshot->stream);
|
|
return 1;
|
|
}
|
|
|
|
void http_snapshot(http_worker_t *worker, FILE *stream)
|
|
{
|
|
int max_delay_value = SNAPSHOT_DEFAULT_DELAY_PARAM;
|
|
|
|
// passing the max_delay=0 will ensure that frame is capture at this exact moment
|
|
char *max_delay = http_get_param(worker, "max_delay");
|
|
if (max_delay) {
|
|
max_delay_value = atoi(max_delay);
|
|
free(max_delay);
|
|
}
|
|
|
|
http_snapshot_t snapshot = {
|
|
.stream = stream,
|
|
.start_time_us = get_monotonic_time_us(NULL, NULL) - max_delay_value * 1000
|
|
};
|
|
|
|
int n = buffer_lock_write_loop(&snapshot_lock, 1, SNAPSHOT_TIMEOUT_MS,
|
|
(buffer_write_fn)http_snapshot_buf_part, &snapshot);
|
|
|
|
if (n <= 0) {
|
|
http_500(stream, NULL);
|
|
fprintf(stream, "No snapshot captured yet.\r\n");
|
|
}
|
|
}
|
|
|
|
int http_stream_buf_part(buffer_lock_t *buf_lock, buffer_t *buf, int frame, FILE *stream)
|
|
{
|
|
if (!frame && !fputs(STREAM_HEADER, stream)) {
|
|
return -1;
|
|
}
|
|
if (!fprintf(stream, STREAM_PART, buf->used)) {
|
|
return -1;
|
|
}
|
|
if (!fwrite(buf->start, buf->used, 1, stream)) {
|
|
return -1;
|
|
}
|
|
if (!fputs(STREAM_BOUNDARY, stream)) {
|
|
return -1;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
void http_stream(http_worker_t *worker, FILE *stream)
|
|
{
|
|
int n = buffer_lock_write_loop(&stream_lock, 0, 0, (buffer_write_fn)http_stream_buf_part, stream);
|
|
|
|
if (n == 0) {
|
|
http_500(stream, NULL);
|
|
fprintf(stream, "No frames.\n");
|
|
} else if (n < 0) {
|
|
fprintf(stream, "Interrupted. Received %d frames", -n);
|
|
}
|
|
}
|