2016-11-25 14:34:00 -08:00
|
|
|
// This file is part of Moonfire NVR, a security camera digital video recorder.
|
|
|
|
// Copyright (C) 2016 Scott Lamb <slamb@slamb.org>
|
|
|
|
//
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// In addition, as a special exception, the copyright holders give
|
|
|
|
// permission to link the code of portions of this program with the
|
|
|
|
// OpenSSL library under certain conditions as described in each
|
|
|
|
// individual source file, and distribute linked combinations including
|
|
|
|
// the two.
|
|
|
|
//
|
|
|
|
// You must obey the GNU General Public License in all respects for all
|
|
|
|
// of the code used other than OpenSSL. If you modify file(s) with this
|
|
|
|
// exception, you may extend this exception to your version of the
|
|
|
|
// file(s), but you are not obligated to do so. If you do not wish to do
|
|
|
|
// so, delete this exception statement from your version. If you delete
|
|
|
|
// this exception statement from all source files in the program, then
|
|
|
|
// also delete it here.
|
|
|
|
//
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
// GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2018-12-28 21:53:29 -06:00
|
|
|
use base::clock::Clocks;
|
2020-02-28 21:41:31 -08:00
|
|
|
use base::{ErrorKind, bail_t, strutil};
|
2020-01-08 23:04:36 -08:00
|
|
|
use bytes::Bytes;
|
2018-12-28 12:21:49 -06:00
|
|
|
use crate::body::{Body, BoxedError};
|
2018-12-28 21:53:29 -06:00
|
|
|
use crate::json;
|
|
|
|
use crate::mp4;
|
2018-11-25 21:31:50 -08:00
|
|
|
use base64;
|
|
|
|
use bytes::{BufMut, BytesMut};
|
2016-11-25 14:34:00 -08:00
|
|
|
use core::borrow::Borrow;
|
|
|
|
use core::str::FromStr;
|
2018-12-28 21:53:29 -06:00
|
|
|
use db::{auth, recording};
|
|
|
|
use db::dir::SampleFileDir;
|
|
|
|
use failure::{Error, bail, format_err};
|
2018-02-11 22:45:51 -08:00
|
|
|
use fnv::FnvHashMap;
|
2020-02-28 21:41:31 -08:00
|
|
|
use futures::sink::SinkExt;
|
2020-01-08 23:04:36 -08:00
|
|
|
use futures::future::{self, Future, TryFutureExt};
|
2020-02-28 21:41:31 -08:00
|
|
|
use futures::stream::StreamExt;
|
2018-12-28 21:53:29 -06:00
|
|
|
use http::{Request, Response, status::StatusCode};
|
2018-01-23 11:08:21 -08:00
|
|
|
use http_serve;
|
2018-08-29 22:26:19 -07:00
|
|
|
use http::header::{self, HeaderValue};
|
2018-12-28 21:53:29 -06:00
|
|
|
use lazy_static::lazy_static;
|
|
|
|
use log::{debug, info, warn};
|
2016-12-20 22:08:18 -08:00
|
|
|
use regex::Regex;
|
2016-11-25 14:34:00 -08:00
|
|
|
use serde_json;
|
2017-10-21 21:54:27 -07:00
|
|
|
use std::collections::HashMap;
|
2016-12-20 22:08:18 -08:00
|
|
|
use std::cmp;
|
2017-10-21 21:54:27 -07:00
|
|
|
use std::fs;
|
2018-11-28 14:22:30 -08:00
|
|
|
use std::net::IpAddr;
|
2016-12-08 21:28:50 -08:00
|
|
|
use std::ops::Range;
|
2017-10-21 21:54:27 -07:00
|
|
|
use std::path::PathBuf;
|
2020-01-08 23:04:36 -08:00
|
|
|
use std::pin::Pin;
|
2017-02-24 21:33:26 -08:00
|
|
|
use std::sync::Arc;
|
2020-02-28 21:41:31 -08:00
|
|
|
use tokio_tungstenite::tungstenite;
|
2016-11-25 14:34:00 -08:00
|
|
|
use url::form_urlencoded;
|
|
|
|
use uuid::Uuid;
|
|
|
|
|
|
|
|
lazy_static! {
|
2016-12-20 22:08:18 -08:00
|
|
|
/// Regex used to parse the `s` query parameter to `view.mp4`.
|
|
|
|
/// As described in `design/api.md`, this is of the form
|
2018-03-02 11:38:11 -08:00
|
|
|
/// `START_ID[-END_ID][@OPEN_ID][.[REL_START_TIME]-[REL_END_TIME]]`.
|
|
|
|
static ref SEGMENTS_RE: Regex =
|
|
|
|
Regex::new(r"^(\d+)(-\d+)?(@\d+)?(?:\.(\d+)?-(\d+)?)?$").unwrap();
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
type BoxedFuture = Box<dyn Future<Output = Result<Response<Body>, BoxedError>> +
|
|
|
|
Sync + Send + 'static>;
|
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
2016-11-25 14:34:00 -08:00
|
|
|
enum Path {
|
2018-12-29 13:06:44 -06:00
|
|
|
TopLevel, // "/api/"
|
|
|
|
Request, // "/api/request"
|
|
|
|
InitSegment([u8; 20], bool), // "/api/init/<sha1>.mp4{.txt}"
|
|
|
|
Camera(Uuid), // "/api/cameras/<uuid>/"
|
2019-06-06 16:18:13 -07:00
|
|
|
Signals, // "/api/signals"
|
2018-12-29 13:06:44 -06:00
|
|
|
StreamRecordings(Uuid, db::StreamType), // "/api/cameras/<uuid>/<type>/recordings"
|
|
|
|
StreamViewMp4(Uuid, db::StreamType, bool), // "/api/cameras/<uuid>/<type>/view.mp4{.txt}"
|
|
|
|
StreamViewMp4Segment(Uuid, db::StreamType, bool), // "/api/cameras/<uuid>/<type>/view.m4s{.txt}"
|
2019-01-21 15:58:52 -08:00
|
|
|
StreamLiveMp4Segments(Uuid, db::StreamType), // "/api/cameras/<uuid>/<type>/live.m4s"
|
2018-12-29 13:06:44 -06:00
|
|
|
Login, // "/api/login"
|
|
|
|
Logout, // "/api/logout"
|
|
|
|
Static, // (anything that doesn't start with "/api/")
|
2016-11-25 14:34:00 -08:00
|
|
|
NotFound,
|
|
|
|
}
|
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
impl Path {
|
|
|
|
fn decode(path: &str) -> Self {
|
|
|
|
if !path.starts_with("/api/") {
|
|
|
|
return Path::Static;
|
|
|
|
}
|
|
|
|
let path = &path["/api".len()..];
|
|
|
|
if path == "/" {
|
|
|
|
return Path::TopLevel;
|
|
|
|
}
|
|
|
|
match path {
|
|
|
|
"/login" => return Path::Login,
|
|
|
|
"/logout" => return Path::Logout,
|
2019-06-06 16:18:13 -07:00
|
|
|
"/request" => return Path::Request,
|
|
|
|
"/signals" => return Path::Signals,
|
2018-12-29 13:06:44 -06:00
|
|
|
_ => {},
|
|
|
|
};
|
|
|
|
if path.starts_with("/init/") {
|
|
|
|
let (debug, path) = if path.ends_with(".txt") {
|
|
|
|
(true, &path[0 .. path.len() - 4])
|
|
|
|
} else {
|
|
|
|
(false, path)
|
|
|
|
};
|
|
|
|
if path.len() != 50 || !path.ends_with(".mp4") {
|
|
|
|
return Path::NotFound;
|
|
|
|
}
|
|
|
|
if let Ok(sha1) = strutil::dehex(&path.as_bytes()[6..46]) {
|
|
|
|
return Path::InitSegment(sha1, debug);
|
|
|
|
}
|
2017-10-01 15:29:22 -07:00
|
|
|
return Path::NotFound;
|
|
|
|
}
|
2018-12-29 13:06:44 -06:00
|
|
|
if !path.starts_with("/cameras/") {
|
|
|
|
return Path::NotFound;
|
2017-10-01 15:29:22 -07:00
|
|
|
}
|
2018-12-29 13:06:44 -06:00
|
|
|
let path = &path["/cameras/".len()..];
|
|
|
|
let slash = match path.find('/') {
|
|
|
|
None => { return Path::NotFound; },
|
|
|
|
Some(s) => s,
|
|
|
|
};
|
|
|
|
let uuid = &path[0 .. slash];
|
|
|
|
let path = &path[slash+1 .. ];
|
2016-11-25 14:34:00 -08:00
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
// TODO(slamb): require uuid to be in canonical format.
|
|
|
|
let uuid = match Uuid::parse_str(uuid) {
|
|
|
|
Ok(u) => u,
|
|
|
|
Err(_) => { return Path::NotFound },
|
|
|
|
};
|
2018-01-23 11:05:07 -08:00
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
if path.is_empty() {
|
|
|
|
return Path::Camera(uuid);
|
|
|
|
}
|
2018-01-23 11:05:07 -08:00
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
let slash = match path.find('/') {
|
|
|
|
None => { return Path::NotFound; },
|
|
|
|
Some(s) => s,
|
|
|
|
};
|
|
|
|
let (type_, path) = path.split_at(slash);
|
2018-01-23 11:05:07 -08:00
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
let type_ = match db::StreamType::parse(type_) {
|
|
|
|
None => { return Path::NotFound; },
|
|
|
|
Some(t) => t,
|
|
|
|
};
|
|
|
|
match path {
|
|
|
|
"/recordings" => Path::StreamRecordings(uuid, type_),
|
|
|
|
"/view.mp4" => Path::StreamViewMp4(uuid, type_, false),
|
|
|
|
"/view.mp4.txt" => Path::StreamViewMp4(uuid, type_, true),
|
|
|
|
"/view.m4s" => Path::StreamViewMp4Segment(uuid, type_, false),
|
|
|
|
"/view.m4s.txt" => Path::StreamViewMp4Segment(uuid, type_, true),
|
2019-01-21 15:58:52 -08:00
|
|
|
"/live.m4s" => Path::StreamLiveMp4Segments(uuid, type_),
|
2018-12-29 13:06:44 -06:00
|
|
|
_ => Path::NotFound,
|
|
|
|
}
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
fn plain_response<B: Into<Body>>(status: http::StatusCode, body: B) -> Response<Body> {
|
|
|
|
Response::builder()
|
|
|
|
.status(status)
|
|
|
|
.header(header::CONTENT_TYPE, HeaderValue::from_static("text/plain"))
|
|
|
|
.body(body.into()).expect("hardcoded head should be valid")
|
|
|
|
}
|
|
|
|
|
|
|
|
fn not_found<B: Into<Body>>(body: B) -> Response<Body> {
|
|
|
|
plain_response(StatusCode::NOT_FOUND, body)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn bad_req<B: Into<Body>>(body: B) -> Response<Body> {
|
|
|
|
plain_response(StatusCode::BAD_REQUEST, body)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn internal_server_err<E: Into<Error>>(err: E) -> Response<Body> {
|
|
|
|
plain_response(StatusCode::INTERNAL_SERVER_ERROR, err.into().to_string())
|
|
|
|
}
|
|
|
|
|
2018-12-28 17:30:33 -06:00
|
|
|
fn from_base_error(err: base::Error) -> Response<Body> {
|
|
|
|
let status_code = match err.kind() {
|
2019-06-19 15:17:50 -07:00
|
|
|
ErrorKind::PermissionDenied | ErrorKind::Unauthenticated => StatusCode::UNAUTHORIZED,
|
2018-12-28 17:30:33 -06:00
|
|
|
ErrorKind::InvalidArgument => StatusCode::BAD_REQUEST,
|
|
|
|
ErrorKind::NotFound => StatusCode::NOT_FOUND,
|
|
|
|
_ => StatusCode::INTERNAL_SERVER_ERROR,
|
|
|
|
};
|
|
|
|
plain_response(status_code, err.to_string())
|
|
|
|
}
|
|
|
|
|
2016-12-20 22:08:18 -08:00
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
|
|
struct Segments {
|
|
|
|
ids: Range<i32>,
|
2018-03-02 11:38:11 -08:00
|
|
|
open_id: Option<u32>,
|
2016-12-20 22:08:18 -08:00
|
|
|
start_time: i64,
|
|
|
|
end_time: Option<i64>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Segments {
|
|
|
|
pub fn parse(input: &str) -> Result<Segments, ()> {
|
|
|
|
let caps = SEGMENTS_RE.captures(input).ok_or(())?;
|
2017-01-12 23:09:02 -08:00
|
|
|
let ids_start = i32::from_str(caps.get(1).unwrap().as_str()).map_err(|_| ())?;
|
|
|
|
let ids_end = match caps.get(2) {
|
2018-03-02 11:38:11 -08:00
|
|
|
Some(m) => i32::from_str(&m.as_str()[1..]).map_err(|_| ())?,
|
2016-12-20 22:08:18 -08:00
|
|
|
None => ids_start,
|
|
|
|
} + 1;
|
2018-03-02 11:38:11 -08:00
|
|
|
let open_id = match caps.get(3) {
|
|
|
|
Some(m) => Some(u32::from_str(&m.as_str()[1..]).map_err(|_| ())?),
|
|
|
|
None => None,
|
|
|
|
};
|
2016-12-20 22:08:18 -08:00
|
|
|
if ids_start < 0 || ids_end <= ids_start {
|
|
|
|
return Err(());
|
|
|
|
}
|
2018-03-02 11:38:11 -08:00
|
|
|
let start_time = caps.get(4).map_or(Ok(0), |m| i64::from_str(m.as_str())).map_err(|_| ())?;
|
2016-12-20 22:08:18 -08:00
|
|
|
if start_time < 0 {
|
|
|
|
return Err(());
|
|
|
|
}
|
2018-03-02 11:38:11 -08:00
|
|
|
let end_time = match caps.get(5) {
|
2016-12-20 22:08:18 -08:00
|
|
|
Some(v) => {
|
2017-01-12 23:09:02 -08:00
|
|
|
let e = i64::from_str(v.as_str()).map_err(|_| ())?;
|
2016-12-20 22:08:18 -08:00
|
|
|
if e <= start_time {
|
|
|
|
return Err(());
|
|
|
|
}
|
|
|
|
Some(e)
|
|
|
|
},
|
|
|
|
None => None
|
|
|
|
};
|
2018-03-02 11:38:11 -08:00
|
|
|
Ok(Segments {
|
2016-12-20 22:08:18 -08:00
|
|
|
ids: ids_start .. ids_end,
|
2018-03-02 11:38:11 -08:00
|
|
|
open_id,
|
|
|
|
start_time,
|
|
|
|
end_time,
|
2016-12-20 22:08:18 -08:00
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-10-21 21:54:27 -07:00
|
|
|
/// A user interface file (.html, .js, etc).
|
|
|
|
/// The list of files is loaded into the server at startup; this makes path canonicalization easy.
|
|
|
|
/// The files themselves are opened on every request so they can be changed during development.
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct UiFile {
|
2018-08-29 22:26:19 -07:00
|
|
|
mime: HeaderValue,
|
2017-10-21 21:54:27 -07:00
|
|
|
path: PathBuf,
|
|
|
|
}
|
|
|
|
|
2019-06-19 15:17:50 -07:00
|
|
|
struct Caller {
|
|
|
|
permissions: db::Permissions,
|
|
|
|
session: Option<json::Session>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Caller {
|
|
|
|
}
|
|
|
|
|
2017-10-21 21:54:27 -07:00
|
|
|
struct ServiceInner {
|
2017-03-02 19:29:28 -08:00
|
|
|
db: Arc<db::Database>,
|
2018-02-11 22:45:51 -08:00
|
|
|
dirs_by_stream_id: Arc<FnvHashMap<i32, Arc<SampleFileDir>>>,
|
2017-10-21 21:54:27 -07:00
|
|
|
ui_files: HashMap<String, UiFile>,
|
2017-10-21 23:57:13 -07:00
|
|
|
time_zone_name: String,
|
2019-06-19 15:17:50 -07:00
|
|
|
allow_unauthenticated_permissions: Option<db::Permissions>,
|
2018-11-28 14:22:30 -08:00
|
|
|
trust_forward_hdrs: bool,
|
2017-03-02 19:29:28 -08:00
|
|
|
}
|
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
type ResponseResult = Result<Response<Body>, Response<Body>>;
|
2016-11-25 14:34:00 -08:00
|
|
|
|
2019-06-06 16:18:13 -07:00
|
|
|
fn serve_json<T: serde::ser::Serialize>(req: &Request<hyper::Body>, out: &T) -> ResponseResult {
|
|
|
|
let (mut resp, writer) = http_serve::streaming_body(&req).build();
|
|
|
|
resp.headers_mut().insert(header::CONTENT_TYPE,
|
|
|
|
HeaderValue::from_static("application/json"));
|
|
|
|
if let Some(mut w) = writer {
|
|
|
|
serde_json::to_writer(&mut w, out).map_err(internal_server_err)?;
|
|
|
|
}
|
|
|
|
Ok(resp)
|
|
|
|
}
|
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
impl ServiceInner {
|
2019-06-19 15:17:50 -07:00
|
|
|
fn top_level(&self, req: &Request<::hyper::Body>, caller: Caller) -> ResponseResult {
|
2017-10-21 21:54:27 -07:00
|
|
|
let mut days = false;
|
2019-06-19 15:17:50 -07:00
|
|
|
let mut camera_configs = false;
|
2018-01-23 11:22:23 -08:00
|
|
|
if let Some(q) = req.uri().query() {
|
2017-10-21 21:54:27 -07:00
|
|
|
for (key, value) in form_urlencoded::parse(q.as_bytes()) {
|
2018-11-25 21:31:50 -08:00
|
|
|
let (key, value): (_, &str) = (key.borrow(), value.borrow());
|
2017-10-21 21:54:27 -07:00
|
|
|
match key {
|
|
|
|
"days" => days = value == "true",
|
2019-06-19 15:17:50 -07:00
|
|
|
"cameraConfigs" => camera_configs = value == "true",
|
2017-10-21 21:54:27 -07:00
|
|
|
_ => {},
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-19 15:17:50 -07:00
|
|
|
if camera_configs {
|
|
|
|
if !caller.permissions.read_camera_configs {
|
|
|
|
return Err(plain_response(StatusCode::UNAUTHORIZED,
|
|
|
|
"read_camera_configs required"));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-06 16:18:13 -07:00
|
|
|
let db = self.db.lock();
|
|
|
|
serve_json(req, &json::TopLevel {
|
|
|
|
time_zone_name: &self.time_zone_name,
|
2019-06-19 15:17:50 -07:00
|
|
|
cameras: (&db, days, camera_configs),
|
|
|
|
session: caller.session,
|
2019-06-06 16:18:13 -07:00
|
|
|
signals: (&db, days),
|
|
|
|
signal_types: &db,
|
|
|
|
})
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
fn camera(&self, req: &Request<::hyper::Body>, uuid: Uuid) -> ResponseResult {
|
2019-06-06 16:18:13 -07:00
|
|
|
let db = self.db.lock();
|
|
|
|
let camera = db.get_camera(uuid)
|
|
|
|
.ok_or_else(|| not_found(format!("no such camera {}", uuid)))?;
|
2019-06-19 15:17:50 -07:00
|
|
|
serve_json(req, &json::Camera::wrap(camera, &db, true, false).map_err(internal_server_err)?)
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
|
2018-08-29 22:26:19 -07:00
|
|
|
fn stream_recordings(&self, req: &Request<::hyper::Body>, uuid: Uuid, type_: db::StreamType)
|
2018-11-25 21:31:50 -08:00
|
|
|
-> ResponseResult {
|
2017-10-17 09:00:05 -07:00
|
|
|
let (r, split) = {
|
2019-06-13 21:55:15 -07:00
|
|
|
let mut time = recording::Time::min_value() .. recording::Time::max_value();
|
2017-10-17 09:00:05 -07:00
|
|
|
let mut split = recording::Duration(i64::max_value());
|
2018-01-23 11:22:23 -08:00
|
|
|
if let Some(q) = req.uri().query() {
|
2017-10-17 09:00:05 -07:00
|
|
|
for (key, value) in form_urlencoded::parse(q.as_bytes()) {
|
|
|
|
let (key, value) = (key.borrow(), value.borrow());
|
|
|
|
match key {
|
2018-11-25 21:31:50 -08:00
|
|
|
"startTime90k" => {
|
|
|
|
time.start = recording::Time::parse(value)
|
|
|
|
.map_err(|_| bad_req("unparseable startTime90k"))?
|
|
|
|
},
|
|
|
|
"endTime90k" => {
|
|
|
|
time.end = recording::Time::parse(value)
|
|
|
|
.map_err(|_| bad_req("unparseable endTime90k"))?
|
|
|
|
},
|
|
|
|
"split90k" => {
|
|
|
|
split = recording::Duration(i64::from_str(value)
|
|
|
|
.map_err(|_| bad_req("unparseable split90k"))?)
|
|
|
|
},
|
2017-10-17 09:00:05 -07:00
|
|
|
_ => {},
|
|
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
(time, split)
|
|
|
|
};
|
2016-12-08 21:28:50 -08:00
|
|
|
let mut out = json::ListRecordings{recordings: Vec::new()};
|
|
|
|
{
|
|
|
|
let db = self.db.lock();
|
|
|
|
let camera = db.get_camera(uuid)
|
2018-11-25 21:31:50 -08:00
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such camera {}", uuid)))?;
|
2018-01-23 11:05:07 -08:00
|
|
|
let stream_id = camera.streams[type_.index()]
|
2018-11-25 21:31:50 -08:00
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such stream {}/{}", uuid, type_)))?;
|
2018-02-23 09:19:42 -08:00
|
|
|
db.list_aggregated_recordings(stream_id, r, split, &mut |row| {
|
2017-10-04 06:36:30 -07:00
|
|
|
let end = row.ids.end - 1; // in api, ids are inclusive.
|
2018-03-01 20:59:05 -08:00
|
|
|
let vse = db.video_sample_entries_by_id().get(&row.video_sample_entry_id).unwrap();
|
2017-10-04 00:00:56 -07:00
|
|
|
out.recordings.push(json::Recording {
|
|
|
|
start_id: row.ids.start,
|
2018-03-02 11:38:11 -08:00
|
|
|
end_id: if end == row.ids.start { None } else { Some(end) },
|
2016-12-20 22:08:18 -08:00
|
|
|
start_time_90k: row.time.start.0,
|
|
|
|
end_time_90k: row.time.end.0,
|
2016-12-08 21:28:50 -08:00
|
|
|
sample_file_bytes: row.sample_file_bytes,
|
2018-03-02 11:38:11 -08:00
|
|
|
open_id: row.open_id,
|
|
|
|
first_uncommitted: row.first_uncommitted,
|
2016-12-08 21:28:50 -08:00
|
|
|
video_samples: row.video_samples,
|
2018-03-01 20:59:05 -08:00
|
|
|
video_sample_entry_width: vse.width,
|
|
|
|
video_sample_entry_height: vse.height,
|
|
|
|
video_sample_entry_sha1: strutil::hex(&vse.sha1),
|
2018-03-02 15:40:32 -08:00
|
|
|
growing: row.growing,
|
2016-12-08 21:28:50 -08:00
|
|
|
});
|
|
|
|
Ok(())
|
2018-11-25 21:31:50 -08:00
|
|
|
}).map_err(internal_server_err)?;
|
2016-12-08 21:28:50 -08:00
|
|
|
}
|
2019-06-06 16:18:13 -07:00
|
|
|
serve_json(req, &out)
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
fn init_segment(&self, sha1: [u8; 20], debug: bool, req: &Request<::hyper::Body>)
|
|
|
|
-> ResponseResult {
|
2017-10-01 15:29:22 -07:00
|
|
|
let mut builder = mp4::FileBuilder::new(mp4::Type::InitSegment);
|
|
|
|
let db = self.db.lock();
|
2018-03-01 20:59:05 -08:00
|
|
|
for ent in db.video_sample_entries_by_id().values() {
|
2017-10-01 15:29:22 -07:00
|
|
|
if ent.sha1 == sha1 {
|
|
|
|
builder.append_video_sample_entry(ent.clone());
|
2018-11-25 21:31:50 -08:00
|
|
|
let mp4 = builder.build(self.db.clone(), self.dirs_by_stream_id.clone())
|
2018-12-28 17:30:33 -06:00
|
|
|
.map_err(from_base_error)?;
|
2018-12-29 13:06:44 -06:00
|
|
|
if debug {
|
|
|
|
return Ok(plain_response(StatusCode::OK, format!("{:#?}", mp4)));
|
|
|
|
} else {
|
|
|
|
return Ok(http_serve::serve(mp4, req));
|
|
|
|
}
|
2017-10-01 15:29:22 -07:00
|
|
|
}
|
|
|
|
}
|
2018-11-25 21:31:50 -08:00
|
|
|
Err(not_found("no such init segment"))
|
2017-10-01 15:29:22 -07:00
|
|
|
}
|
|
|
|
|
2019-06-19 15:17:50 -07:00
|
|
|
fn stream_view_mp4(&self, req: &Request<::hyper::Body>, caller: Caller, uuid: Uuid,
|
2019-01-21 15:58:52 -08:00
|
|
|
stream_type: db::StreamType, mp4_type: mp4::Type, debug: bool)
|
2018-11-25 21:31:50 -08:00
|
|
|
-> ResponseResult {
|
2019-06-19 15:17:50 -07:00
|
|
|
if !caller.permissions.view_video {
|
|
|
|
return Err(plain_response(StatusCode::UNAUTHORIZED, "view_video required"));
|
|
|
|
}
|
2020-02-16 23:16:19 -08:00
|
|
|
let (stream_id, camera_name);
|
|
|
|
{
|
2016-11-25 14:34:00 -08:00
|
|
|
let db = self.db.lock();
|
2016-12-02 21:46:31 -08:00
|
|
|
let camera = db.get_camera(uuid)
|
2018-11-25 21:31:50 -08:00
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such camera {}", uuid)))?;
|
2020-02-16 23:16:19 -08:00
|
|
|
camera_name = camera.short_name.clone();
|
|
|
|
stream_id = camera.streams[stream_type.index()]
|
2018-11-25 21:31:50 -08:00
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such stream {}/{}", uuid,
|
2020-02-16 23:16:19 -08:00
|
|
|
stream_type)))?;
|
|
|
|
|
2016-11-25 14:34:00 -08:00
|
|
|
};
|
2020-02-16 23:16:19 -08:00
|
|
|
let mut start_time_for_filename = None;
|
2019-01-21 15:58:52 -08:00
|
|
|
let mut builder = mp4::FileBuilder::new(mp4_type);
|
2018-01-23 11:05:07 -08:00
|
|
|
if let Some(q) = req.uri().query() {
|
2017-03-02 19:29:28 -08:00
|
|
|
for (key, value) in form_urlencoded::parse(q.as_bytes()) {
|
|
|
|
let (key, value) = (key.borrow(), value.borrow());
|
|
|
|
match key {
|
|
|
|
"s" => {
|
|
|
|
let s = Segments::parse(value).map_err(
|
2018-11-25 21:31:50 -08:00
|
|
|
|()| plain_response(StatusCode::BAD_REQUEST,
|
|
|
|
format!("invalid s parameter: {}", value)))?;
|
2018-01-23 11:05:07 -08:00
|
|
|
debug!("stream_view_mp4: appending s={:?}", s);
|
2017-03-02 19:29:28 -08:00
|
|
|
let mut est_segments = (s.ids.end - s.ids.start) as usize;
|
|
|
|
if let Some(end) = s.end_time {
|
|
|
|
// There should be roughly ceil((end - start) /
|
|
|
|
// desired_recording_duration) recordings in the desired timespan if
|
|
|
|
// there are no gaps or overlap, possibly another for misalignment of
|
|
|
|
// the requested timespan with the rotate offset and another because
|
|
|
|
// rotation only happens at key frames.
|
|
|
|
let ceil_durations = (end - s.start_time +
|
|
|
|
recording::DESIRED_RECORDING_DURATION - 1) /
|
|
|
|
recording::DESIRED_RECORDING_DURATION;
|
|
|
|
est_segments = cmp::min(est_segments, (ceil_durations + 2) as usize);
|
|
|
|
}
|
|
|
|
builder.reserve(est_segments);
|
|
|
|
let db = self.db.lock();
|
|
|
|
let mut prev = None;
|
|
|
|
let mut cur_off = 0;
|
2018-02-23 09:19:42 -08:00
|
|
|
db.list_recordings_by_id(stream_id, s.ids.clone(), &mut |r| {
|
2018-02-20 10:11:10 -08:00
|
|
|
let recording_id = r.id.recording();
|
|
|
|
|
2018-03-02 11:38:11 -08:00
|
|
|
if let Some(o) = s.open_id {
|
|
|
|
if r.open_id != o {
|
|
|
|
bail!("recording {} has open id {}, requested {}",
|
|
|
|
r.id, r.open_id, o);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-02 19:29:28 -08:00
|
|
|
// Check for missing recordings.
|
|
|
|
match prev {
|
2018-02-20 10:11:10 -08:00
|
|
|
None if recording_id == s.ids.start => {},
|
2018-02-20 22:46:14 -08:00
|
|
|
None => bail!("no such recording {}/{}", stream_id, s.ids.start),
|
2018-02-20 10:11:10 -08:00
|
|
|
Some(id) if r.id.recording() != id + 1 => {
|
2018-02-20 22:46:14 -08:00
|
|
|
bail!("no such recording {}/{}", stream_id, id + 1);
|
2017-03-02 19:29:28 -08:00
|
|
|
},
|
|
|
|
_ => {},
|
|
|
|
};
|
2018-02-20 10:11:10 -08:00
|
|
|
prev = Some(recording_id);
|
2017-03-02 19:29:28 -08:00
|
|
|
|
|
|
|
// Add a segment for the relevant part of the recording, if any.
|
|
|
|
let end_time = s.end_time.unwrap_or(i64::max_value());
|
|
|
|
let d = r.duration_90k as i64;
|
|
|
|
if s.start_time <= cur_off + d && cur_off < end_time {
|
|
|
|
let start = cmp::max(0, s.start_time - cur_off);
|
|
|
|
let end = cmp::min(d, end_time - cur_off);
|
|
|
|
let times = start as i32 .. end as i32;
|
2018-02-20 10:11:10 -08:00
|
|
|
debug!("...appending recording {} with times {:?} \
|
|
|
|
(out of dur {})", r.id, times, d);
|
2020-02-16 23:16:19 -08:00
|
|
|
if start_time_for_filename.is_none() {
|
|
|
|
start_time_for_filename =
|
|
|
|
Some(r.start + recording::Duration(start));
|
|
|
|
}
|
2017-03-02 19:29:28 -08:00
|
|
|
builder.append(&db, r, start as i32 .. end as i32)?;
|
|
|
|
} else {
|
2018-02-20 10:11:10 -08:00
|
|
|
debug!("...skipping recording {} dur {}", r.id, d);
|
2017-03-02 19:29:28 -08:00
|
|
|
}
|
|
|
|
cur_off += d;
|
|
|
|
Ok(())
|
2018-11-25 21:31:50 -08:00
|
|
|
}).map_err(internal_server_err)?;
|
2017-03-02 19:29:28 -08:00
|
|
|
|
2016-12-20 22:08:18 -08:00
|
|
|
// Check for missing recordings.
|
|
|
|
match prev {
|
2017-03-02 19:29:28 -08:00
|
|
|
Some(id) if s.ids.end != id + 1 => {
|
2018-11-25 21:31:50 -08:00
|
|
|
return Err(not_found(format!("no such recording {}/{}",
|
|
|
|
stream_id, s.ids.end - 1)));
|
2017-03-02 19:29:28 -08:00
|
|
|
},
|
|
|
|
None => {
|
2018-11-25 21:31:50 -08:00
|
|
|
return Err(not_found(format!("no such recording {}/{}",
|
|
|
|
stream_id, s.ids.start)));
|
2016-12-20 22:08:18 -08:00
|
|
|
},
|
|
|
|
_ => {},
|
|
|
|
};
|
2017-03-02 19:29:28 -08:00
|
|
|
if let Some(end) = s.end_time {
|
|
|
|
if end > cur_off {
|
2018-11-25 21:31:50 -08:00
|
|
|
return Err(plain_response(
|
|
|
|
StatusCode::BAD_REQUEST,
|
|
|
|
format!("end time {} is beyond specified recordings",
|
|
|
|
end)));
|
2017-03-02 19:29:28 -08:00
|
|
|
}
|
2016-12-20 22:08:18 -08:00
|
|
|
}
|
2017-03-02 19:29:28 -08:00
|
|
|
},
|
|
|
|
"ts" => builder.include_timestamp_subtitle_track(value == "true"),
|
2018-11-25 21:31:50 -08:00
|
|
|
_ => return Err(bad_req(format!("parameter {} not understood", key))),
|
2017-03-02 19:29:28 -08:00
|
|
|
}
|
|
|
|
};
|
|
|
|
}
|
2020-02-16 23:16:19 -08:00
|
|
|
if let Some(start) = start_time_for_filename {
|
|
|
|
let tm = time::at(time::Timespec{sec: start.unix_seconds(), nsec: 0});
|
|
|
|
let stream_abbrev = if stream_type == db::StreamType::MAIN { "main" } else { "sub" };
|
|
|
|
let suffix = if mp4_type == mp4::Type::Normal { "mp4" } else { "m4s" };
|
|
|
|
builder.set_filename(&format!("{}-{}-{}.{}", tm.strftime("%Y%m%d%H%M%S").unwrap(),
|
|
|
|
camera_name, stream_abbrev, suffix))
|
|
|
|
.map_err(from_base_error)?;
|
|
|
|
}
|
2018-11-25 21:31:50 -08:00
|
|
|
let mp4 = builder.build(self.db.clone(), self.dirs_by_stream_id.clone())
|
2018-12-28 17:30:33 -06:00
|
|
|
.map_err(from_base_error)?;
|
2018-12-29 13:06:44 -06:00
|
|
|
if debug {
|
|
|
|
return Ok(plain_response(StatusCode::OK, format!("{:#?}", mp4)));
|
|
|
|
}
|
2018-01-23 11:08:21 -08:00
|
|
|
Ok(http_serve::serve(mp4, req))
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
2017-10-21 21:54:27 -07:00
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
fn static_file(&self, req: &Request<::hyper::Body>, path: &str) -> ResponseResult {
|
|
|
|
let s = self.ui_files.get(path).ok_or_else(|| not_found("no such static file"))?;
|
2020-01-08 23:04:36 -08:00
|
|
|
let f = tokio::task::block_in_place(move || {
|
|
|
|
fs::File::open(&s.path).map_err(internal_server_err)
|
|
|
|
})?;
|
2018-04-06 15:54:52 -07:00
|
|
|
let mut hdrs = http::HeaderMap::new();
|
2018-08-29 22:26:19 -07:00
|
|
|
hdrs.insert(header::CONTENT_TYPE, s.mime.clone());
|
2020-01-08 23:04:36 -08:00
|
|
|
let e = http_serve::ChunkedReadFile::new(f, hdrs).map_err(internal_server_err)?;
|
2018-01-23 11:08:21 -08:00
|
|
|
Ok(http_serve::serve(e, &req))
|
2017-10-21 21:54:27 -07:00
|
|
|
}
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
fn authreq(&self, req: &Request<::hyper::Body>) -> auth::Request {
|
|
|
|
auth::Request {
|
|
|
|
when_sec: Some(self.db.clocks().realtime().sec),
|
2018-11-28 14:22:30 -08:00
|
|
|
addr: if self.trust_forward_hdrs {
|
|
|
|
req.headers().get("X-Real-IP")
|
|
|
|
.and_then(|v| v.to_str().ok())
|
|
|
|
.and_then(|v| IpAddr::from_str(v).ok())
|
|
|
|
} else { None },
|
2018-11-25 21:31:50 -08:00
|
|
|
user_agent: req.headers().get(header::USER_AGENT).map(|ua| ua.as_bytes().to_vec()),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-01 00:44:19 -08:00
|
|
|
fn request(&self, req: &Request<::hyper::Body>) -> ResponseResult {
|
|
|
|
let authreq = self.authreq(req);
|
2018-12-27 16:00:15 -06:00
|
|
|
let host = req.headers().get(header::HOST).map(|h| String::from_utf8_lossy(h.as_bytes()));
|
|
|
|
let agent = authreq.user_agent.as_ref().map(|u| String::from_utf8_lossy(&u[..]));
|
2018-12-01 00:44:19 -08:00
|
|
|
Ok(plain_response(StatusCode::OK, format!(
|
|
|
|
"when: {}\n\
|
2018-12-27 16:00:15 -06:00
|
|
|
host: {:?}\n\
|
2018-12-01 00:44:19 -08:00
|
|
|
addr: {:?}\n\
|
|
|
|
user_agent: {:?}\n\
|
|
|
|
secure: {:?}",
|
|
|
|
time::at(time::Timespec{sec: authreq.when_sec.unwrap(), nsec: 0})
|
|
|
|
.strftime("%FT%T")
|
|
|
|
.map(|f| f.to_string())
|
|
|
|
.unwrap_or_else(|e| e.to_string()),
|
2018-12-27 16:00:15 -06:00
|
|
|
host.as_ref().map(|h| &*h),
|
2018-12-01 00:44:19 -08:00
|
|
|
&authreq.addr,
|
2018-12-27 16:00:15 -06:00
|
|
|
agent.as_ref().map(|a| &*a),
|
2018-12-01 00:44:19 -08:00
|
|
|
self.is_secure(req))))
|
|
|
|
}
|
|
|
|
|
2018-11-28 14:22:30 -08:00
|
|
|
fn is_secure(&self, req: &Request<::hyper::Body>) -> bool {
|
|
|
|
self.trust_forward_hdrs &&
|
|
|
|
req.headers().get("X-Forwarded-Proto")
|
|
|
|
.map(|v| v.as_bytes() == b"https")
|
|
|
|
.unwrap_or(false)
|
|
|
|
}
|
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
fn login(&self, req: &Request<::hyper::Body>, body: Bytes) -> ResponseResult {
|
2020-01-08 23:23:58 -08:00
|
|
|
let r: json::LoginRequest = serde_json::from_slice(&body)
|
|
|
|
.map_err(|e| bad_req(e.to_string()))?;
|
2018-11-25 21:31:50 -08:00
|
|
|
let authreq = self.authreq(req);
|
|
|
|
let host = req.headers().get(header::HOST).ok_or_else(|| bad_req("missing Host header!"))?;
|
|
|
|
let host = host.as_bytes();
|
|
|
|
let domain = match ::memchr::memchr(b':', host) {
|
|
|
|
Some(colon) => &host[0..colon],
|
|
|
|
None => host,
|
|
|
|
}.to_owned();
|
|
|
|
let mut l = self.db.lock();
|
2018-11-28 14:22:30 -08:00
|
|
|
let is_secure = self.is_secure(req);
|
|
|
|
let flags = (auth::SessionFlags::HttpOnly as i32) |
|
|
|
|
(auth::SessionFlags::SameSite as i32) |
|
2018-12-01 22:04:54 -08:00
|
|
|
(auth::SessionFlags::SameSiteStrict as i32) |
|
2020-02-16 23:16:19 -08:00
|
|
|
if is_secure { auth::SessionFlags::Secure as i32 } else { 0 };
|
2020-01-08 23:23:58 -08:00
|
|
|
let (sid, _) = l.login_by_password(authreq, &r.username, r.password, Some(domain),
|
2018-11-25 21:31:50 -08:00
|
|
|
flags)
|
|
|
|
.map_err(|e| plain_response(StatusCode::UNAUTHORIZED, e.to_string()))?;
|
2018-11-28 14:22:30 -08:00
|
|
|
let s_suffix = if is_secure {
|
2020-01-08 23:04:36 -08:00
|
|
|
&b"; HttpOnly; Secure; SameSite=Strict; Max-Age=2147483648; Path=/"[..]
|
2018-11-28 14:22:30 -08:00
|
|
|
} else {
|
2020-01-08 23:04:36 -08:00
|
|
|
&b"; HttpOnly; SameSite=Strict; Max-Age=2147483648; Path=/"[..]
|
2018-11-28 14:22:30 -08:00
|
|
|
};
|
2018-11-25 21:31:50 -08:00
|
|
|
let mut encoded = [0u8; 64];
|
|
|
|
base64::encode_config_slice(&sid, base64::STANDARD_NO_PAD, &mut encoded);
|
|
|
|
let mut cookie = BytesMut::with_capacity("s=".len() + encoded.len() + s_suffix.len());
|
2020-01-08 23:04:36 -08:00
|
|
|
cookie.put(&b"s="[..]);
|
2018-11-25 21:31:50 -08:00
|
|
|
cookie.put(&encoded[..]);
|
|
|
|
cookie.put(s_suffix);
|
|
|
|
Ok(Response::builder()
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(header::SET_COOKIE, HeaderValue::from_maybe_shared(cookie.freeze())
|
|
|
|
.expect("cookie can't have invalid bytes"))
|
2018-11-25 21:31:50 -08:00
|
|
|
.status(StatusCode::NO_CONTENT)
|
|
|
|
.body(b""[..].into()).unwrap())
|
|
|
|
}
|
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
fn logout(&self, req: &Request<hyper::Body>, body: Bytes) -> ResponseResult {
|
2020-01-08 23:23:58 -08:00
|
|
|
let r: json::LogoutRequest = serde_json::from_slice(&body)
|
|
|
|
.map_err(|e| bad_req(e.to_string()))?;
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
let mut res = Response::new(b""[..].into());
|
|
|
|
if let Some(sid) = extract_sid(req) {
|
|
|
|
let authreq = self.authreq(req);
|
|
|
|
let mut l = self.db.lock();
|
|
|
|
let hash = sid.hash();
|
|
|
|
let need_revoke = match l.authenticate_session(authreq.clone(), &hash) {
|
|
|
|
Ok((s, _)) => {
|
2020-01-08 23:23:58 -08:00
|
|
|
if !csrf_matches(r.csrf, s.csrf()) {
|
2018-11-25 21:31:50 -08:00
|
|
|
warn!("logout request with missing/incorrect csrf");
|
|
|
|
return Err(bad_req("logout with incorrect csrf token"));
|
|
|
|
}
|
|
|
|
info!("revoking session");
|
|
|
|
true
|
|
|
|
},
|
|
|
|
Err(e) => {
|
|
|
|
// TODO: distinguish "no such session", "session is no longer valid", and
|
|
|
|
// "user ... is disabled" (which are all client error / bad state) from database
|
|
|
|
// errors.
|
|
|
|
warn!("logout failed: {}", e);
|
|
|
|
false
|
|
|
|
},
|
|
|
|
};
|
|
|
|
if need_revoke {
|
|
|
|
// TODO: inline this above with non-lexical lifetimes.
|
|
|
|
l.revoke_session(auth::RevocationReason::LoggedOut, None, authreq, &hash)
|
|
|
|
.map_err(internal_server_err)?;
|
|
|
|
}
|
|
|
|
|
|
|
|
// By now the session is invalid (whether it was valid to start with or not).
|
|
|
|
// Clear useless cookie.
|
|
|
|
res.headers_mut().append(header::SET_COOKIE,
|
|
|
|
HeaderValue::from_str("s=; Max-Age=0; Path=/").unwrap());
|
|
|
|
}
|
|
|
|
*res.status_mut() = StatusCode::NO_CONTENT;
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
fn post_signals(&self, req: &Request<hyper::Body>, caller: Caller, body: Bytes)
|
2019-06-19 15:17:50 -07:00
|
|
|
-> ResponseResult {
|
|
|
|
if !caller.permissions.update_signals {
|
|
|
|
return Err(plain_response(StatusCode::UNAUTHORIZED, "update_signals required"));
|
|
|
|
}
|
2019-06-14 16:11:12 -07:00
|
|
|
let r: json::PostSignalsRequest = serde_json::from_slice(&body)
|
|
|
|
.map_err(|e| bad_req(e.to_string()))?;
|
|
|
|
let mut l = self.db.lock();
|
|
|
|
let now = recording::Time::new(self.db.clocks().realtime());
|
|
|
|
let start = r.start_time_90k.map(recording::Time).unwrap_or(now);
|
|
|
|
let end = match r.end_base {
|
|
|
|
json::PostSignalsEndBase::Epoch => recording::Time(r.rel_end_time_90k.ok_or_else(
|
|
|
|
|| bad_req("must specify rel_end_time_90k when end_base is epoch"))?),
|
|
|
|
json::PostSignalsEndBase::Now => {
|
|
|
|
now + recording::Duration(r.rel_end_time_90k.unwrap_or(0))
|
|
|
|
},
|
|
|
|
};
|
|
|
|
l.update_signals(start .. end, &r.signal_ids, &r.states).map_err(from_base_error)?;
|
|
|
|
serve_json(req, &json::PostSignalsResponse {
|
|
|
|
time_90k: now.0,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_signals(&self, req: &Request<hyper::Body>) -> ResponseResult {
|
2019-06-13 21:55:15 -07:00
|
|
|
let mut time = recording::Time::min_value() .. recording::Time::max_value();
|
2019-06-06 16:18:13 -07:00
|
|
|
if let Some(q) = req.uri().query() {
|
|
|
|
for (key, value) in form_urlencoded::parse(q.as_bytes()) {
|
|
|
|
let (key, value) = (key.borrow(), value.borrow());
|
|
|
|
match key {
|
|
|
|
"startTime90k" => {
|
|
|
|
time.start = recording::Time::parse(value)
|
|
|
|
.map_err(|_| bad_req("unparseable startTime90k"))?
|
|
|
|
},
|
|
|
|
"endTime90k" => {
|
|
|
|
time.end = recording::Time::parse(value)
|
|
|
|
.map_err(|_| bad_req("unparseable endTime90k"))?
|
|
|
|
},
|
|
|
|
_ => {},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut signals = json::Signals::default();
|
|
|
|
self.db.lock().list_changes_by_time(time, &mut |c: &db::signal::ListStateChangesRow| {
|
|
|
|
signals.times_90k.push(c.when.0);
|
|
|
|
signals.signal_ids.push(c.signal);
|
|
|
|
signals.states.push(c.state);
|
2019-06-13 21:55:15 -07:00
|
|
|
});
|
2019-06-06 16:18:13 -07:00
|
|
|
serve_json(req, &signals)
|
|
|
|
}
|
|
|
|
|
2019-06-19 15:17:50 -07:00
|
|
|
fn authenticate(&self, req: &Request<hyper::Body>, unauth_path: bool)
|
|
|
|
-> Result<Caller, base::Error> {
|
2018-11-25 21:31:50 -08:00
|
|
|
if let Some(sid) = extract_sid(req) {
|
|
|
|
let authreq = self.authreq(req);
|
2019-06-19 15:17:50 -07:00
|
|
|
|
|
|
|
// TODO: real error handling! this assumes all errors are due to lack of
|
|
|
|
// authentication, when they could be logic errors in SQL or such.
|
|
|
|
if let Ok((s, u)) = self.db.lock().authenticate_session(authreq.clone(), &sid.hash()) {
|
|
|
|
return Ok(Caller {
|
|
|
|
permissions: s.permissions.clone(),
|
|
|
|
session: Some(json::Session {
|
2018-11-25 21:31:50 -08:00
|
|
|
username: u.username.clone(),
|
|
|
|
csrf: s.csrf(),
|
2019-06-19 15:17:50 -07:00
|
|
|
}),
|
|
|
|
});
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
2019-06-20 12:10:23 -07:00
|
|
|
info!("authenticate_session failed");
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
2019-06-19 15:17:50 -07:00
|
|
|
|
|
|
|
if let Some(s) = self.allow_unauthenticated_permissions.as_ref() {
|
|
|
|
return Ok(Caller {
|
|
|
|
permissions: s.clone(),
|
|
|
|
session: None,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
if unauth_path {
|
|
|
|
return Ok(Caller {
|
|
|
|
permissions: db::Permissions::default(),
|
|
|
|
session: None,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
bail_t!(Unauthenticated, "unauthenticated");
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn csrf_matches(csrf: &str, session: auth::SessionHash) -> bool {
|
|
|
|
let mut b64 = [0u8; 32];
|
|
|
|
session.encode_base64(&mut b64);
|
|
|
|
::ring::constant_time::verify_slices_are_equal(&b64[..], csrf.as_bytes()).is_ok()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Extracts `s` cookie from the HTTP request. Does not authenticate.
|
|
|
|
fn extract_sid(req: &Request<hyper::Body>) -> Option<auth::RawSessionId> {
|
|
|
|
let hdr = match req.headers().get(header::COOKIE) {
|
|
|
|
None => return None,
|
|
|
|
Some(c) => c,
|
|
|
|
};
|
|
|
|
for mut cookie in hdr.as_bytes().split(|&b| b == b';') {
|
|
|
|
if cookie.starts_with(b" ") {
|
|
|
|
cookie = &cookie[1..];
|
|
|
|
}
|
|
|
|
if cookie.starts_with(b"s=") {
|
|
|
|
let s = &cookie[2..];
|
|
|
|
if let Ok(s) = auth::RawSessionId::decode_base64(s) {
|
|
|
|
return Some(s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
None
|
2017-10-21 21:54:27 -07:00
|
|
|
}
|
|
|
|
|
2020-01-08 23:23:58 -08:00
|
|
|
/// Returns a future separating the request from its JSON body.
|
2019-06-14 16:11:12 -07:00
|
|
|
///
|
|
|
|
/// If this is not a `POST` or the body's `Content-Type` is not
|
2020-01-08 23:23:58 -08:00
|
|
|
/// `application/json`, returns an appropriate error response instead.
|
2019-06-14 16:11:12 -07:00
|
|
|
///
|
|
|
|
/// Use with `and_then` to chain logic which consumes the form body.
|
2020-01-08 23:04:36 -08:00
|
|
|
async fn with_json_body(mut req: Request<hyper::Body>)
|
|
|
|
-> Result<(Request<hyper::Body>, Bytes), Response<Body>> {
|
2019-06-14 16:11:12 -07:00
|
|
|
if *req.method() != http::method::Method::POST {
|
2020-01-08 23:04:36 -08:00
|
|
|
return Err(plain_response(StatusCode::METHOD_NOT_ALLOWED, "POST expected"));
|
2019-06-14 16:11:12 -07:00
|
|
|
}
|
|
|
|
let correct_mime_type = match req.headers().get(header::CONTENT_TYPE) {
|
|
|
|
Some(t) if t == "application/json" => true,
|
|
|
|
Some(t) if t == "application/json; charset=UTF-8" => true,
|
|
|
|
_ => false,
|
|
|
|
};
|
|
|
|
if !correct_mime_type {
|
2020-01-08 23:04:36 -08:00
|
|
|
return Err(bad_req("expected application/json request body"));
|
2019-06-14 16:11:12 -07:00
|
|
|
}
|
|
|
|
let b = ::std::mem::replace(req.body_mut(), hyper::Body::empty());
|
2020-01-08 23:04:36 -08:00
|
|
|
match hyper::body::to_bytes(b).await {
|
|
|
|
Ok(b) => Ok((req, b)),
|
|
|
|
Err(e) => Err(internal_server_err(format_err!("unable to read request body: {}", e))),
|
|
|
|
}
|
2019-06-14 16:11:12 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-11-28 14:22:30 -08:00
|
|
|
pub struct Config<'a> {
|
|
|
|
pub db: Arc<db::Database>,
|
|
|
|
pub ui_dir: Option<&'a str>,
|
|
|
|
pub trust_forward_hdrs: bool,
|
|
|
|
pub time_zone_name: String,
|
2019-06-19 15:17:50 -07:00
|
|
|
pub allow_unauthenticated_permissions: Option<db::Permissions>,
|
2018-11-28 14:22:30 -08:00
|
|
|
}
|
|
|
|
|
2017-10-21 21:54:27 -07:00
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct Service(Arc<ServiceInner>);
|
|
|
|
|
|
|
|
impl Service {
|
2018-11-28 14:22:30 -08:00
|
|
|
pub fn new(config: Config) -> Result<Self, Error> {
|
2017-10-21 21:54:27 -07:00
|
|
|
let mut ui_files = HashMap::new();
|
2018-11-28 14:22:30 -08:00
|
|
|
if let Some(d) = config.ui_dir {
|
2017-10-21 21:54:27 -07:00
|
|
|
Service::fill_ui_files(d, &mut ui_files);
|
|
|
|
}
|
|
|
|
debug!("UI files: {:#?}", ui_files);
|
2018-02-11 22:45:51 -08:00
|
|
|
let dirs_by_stream_id = {
|
2018-11-28 14:22:30 -08:00
|
|
|
let l = config.db.lock();
|
2018-02-11 22:45:51 -08:00
|
|
|
let mut d =
|
|
|
|
FnvHashMap::with_capacity_and_hasher(l.streams_by_id().len(), Default::default());
|
|
|
|
for (&id, s) in l.streams_by_id().iter() {
|
|
|
|
let dir_id = match s.sample_file_dir_id {
|
|
|
|
Some(d) => d,
|
|
|
|
None => continue,
|
|
|
|
};
|
|
|
|
d.insert(id, l.sample_file_dirs_by_id()
|
|
|
|
.get(&dir_id)
|
|
|
|
.unwrap()
|
2018-02-14 23:10:10 -08:00
|
|
|
.get()?);
|
2018-02-11 22:45:51 -08:00
|
|
|
}
|
|
|
|
Arc::new(d)
|
|
|
|
};
|
2018-11-25 21:31:50 -08:00
|
|
|
|
2017-10-21 21:54:27 -07:00
|
|
|
Ok(Service(Arc::new(ServiceInner {
|
2018-11-28 14:22:30 -08:00
|
|
|
db: config.db,
|
2018-02-11 22:45:51 -08:00
|
|
|
dirs_by_stream_id,
|
2017-10-21 21:54:27 -07:00
|
|
|
ui_files,
|
2019-06-19 15:17:50 -07:00
|
|
|
allow_unauthenticated_permissions: config.allow_unauthenticated_permissions,
|
2018-11-28 14:22:30 -08:00
|
|
|
trust_forward_hdrs: config.trust_forward_hdrs,
|
|
|
|
time_zone_name: config.time_zone_name,
|
2017-10-21 21:54:27 -07:00
|
|
|
})))
|
|
|
|
}
|
|
|
|
|
|
|
|
fn fill_ui_files(dir: &str, files: &mut HashMap<String, UiFile>) {
|
|
|
|
let r = match fs::read_dir(dir) {
|
|
|
|
Ok(r) => r,
|
|
|
|
Err(e) => {
|
|
|
|
warn!("Unable to search --ui-dir={}; will serve no static files. Error was: {}",
|
|
|
|
dir, e);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
for e in r {
|
|
|
|
let e = match e {
|
|
|
|
Ok(e) => e,
|
|
|
|
Err(e) => {
|
|
|
|
warn!("Error searching UI directory; may be missing files. Error was: {}", e);
|
|
|
|
continue;
|
|
|
|
},
|
|
|
|
};
|
|
|
|
let (p, mime) = match e.file_name().to_str() {
|
2018-04-06 15:54:52 -07:00
|
|
|
Some(n) if n == "index.html" => ("/".to_owned(), "text/html"),
|
|
|
|
Some(n) if n.ends_with(".html") => (format!("/{}", n), "text/html"),
|
|
|
|
Some(n) if n.ends_with(".ico") => (format!("/{}", n), "image/vnd.microsoft.icon"),
|
|
|
|
Some(n) if n.ends_with(".js") => (format!("/{}", n), "text/javascript"),
|
|
|
|
Some(n) if n.ends_with(".map") => (format!("/{}", n), "text/javascript"),
|
|
|
|
Some(n) if n.ends_with(".png") => (format!("/{}", n), "image/png"),
|
2017-10-21 21:54:27 -07:00
|
|
|
Some(n) => {
|
|
|
|
warn!("UI directory file {:?} has unknown extension; skipping", n);
|
|
|
|
continue;
|
|
|
|
},
|
|
|
|
None => {
|
|
|
|
warn!("UI directory file {:?} is not a valid UTF-8 string; skipping",
|
|
|
|
e.file_name());
|
|
|
|
continue;
|
|
|
|
},
|
|
|
|
};
|
|
|
|
files.insert(p, UiFile {
|
2018-08-29 22:26:19 -07:00
|
|
|
mime: HeaderValue::from_static(mime),
|
2017-10-21 21:54:27 -07:00
|
|
|
path: e.path(),
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
2018-11-25 21:31:50 -08:00
|
|
|
|
2020-02-28 21:41:31 -08:00
|
|
|
fn stream_live_m4s(&self, req: Request<::hyper::Body>, caller: Caller, uuid: Uuid,
|
2019-01-21 15:58:52 -08:00
|
|
|
stream_type: db::StreamType) -> ResponseResult {
|
2019-06-19 15:17:50 -07:00
|
|
|
if !caller.permissions.view_video {
|
|
|
|
return Err(plain_response(StatusCode::UNAUTHORIZED, "view_video required"));
|
|
|
|
}
|
2020-02-28 21:41:31 -08:00
|
|
|
|
2019-01-21 15:58:52 -08:00
|
|
|
let stream_id;
|
|
|
|
let open_id;
|
2020-01-08 23:04:36 -08:00
|
|
|
let (sub_tx, sub_rx) = futures::channel::mpsc::unbounded();
|
2019-01-21 15:58:52 -08:00
|
|
|
{
|
|
|
|
let mut db = self.0.db.lock();
|
|
|
|
open_id = match db.open {
|
|
|
|
None => return Err(plain_response(
|
|
|
|
StatusCode::PRECONDITION_FAILED,
|
|
|
|
"database is read-only; there are no live streams")),
|
|
|
|
Some(o) => o.id,
|
|
|
|
};
|
|
|
|
let camera = db.get_camera(uuid)
|
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such camera {}", uuid)))?;
|
|
|
|
stream_id = camera.streams[stream_type.index()]
|
|
|
|
.ok_or_else(|| plain_response(StatusCode::NOT_FOUND,
|
|
|
|
format!("no such stream {}/{}", uuid,
|
|
|
|
stream_type)))?;
|
|
|
|
db.watch_live(stream_id, Box::new(move |l| sub_tx.unbounded_send(l).is_ok()))
|
|
|
|
.expect("stream_id refed by camera");
|
|
|
|
}
|
2020-02-28 21:41:31 -08:00
|
|
|
|
|
|
|
let (parts, body) = req.into_parts();
|
|
|
|
let req = Request::from_parts(parts, ());
|
|
|
|
let response = tungstenite::handshake::server::create_response(&req)
|
|
|
|
.map_err(|e| bad_req(e.to_string()))?;
|
|
|
|
let (parts, ()) = response.into_parts();
|
|
|
|
|
|
|
|
tokio::spawn(self.clone().stream_live_m4s_ws(stream_id, open_id, body, sub_rx));
|
|
|
|
|
|
|
|
Ok(Response::from_parts(parts, Body::from("")))
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn stream_live_m4s_ws(
|
|
|
|
self, stream_id: i32, open_id: u32, body: hyper::Body,
|
|
|
|
mut sub_rx: futures::channel::mpsc::UnboundedReceiver<db::LiveSegment>) {
|
|
|
|
let upgraded = match body.on_upgrade().await {
|
|
|
|
Ok(u) => u,
|
|
|
|
Err(e) => {
|
|
|
|
warn!("Unable to upgrade stream to websocket: {}", e);
|
|
|
|
return;
|
|
|
|
},
|
|
|
|
};
|
|
|
|
let mut ws = tokio_tungstenite::WebSocketStream::from_raw_socket(
|
|
|
|
upgraded,
|
|
|
|
tungstenite::protocol::Role::Server,
|
|
|
|
None,
|
|
|
|
).await;
|
|
|
|
loop {
|
|
|
|
let live = match sub_rx.next().await {
|
|
|
|
Some(l) => l,
|
|
|
|
None => return,
|
|
|
|
};
|
|
|
|
if let Err(e) = self.stream_live_m4s_chunk(open_id, stream_id, &mut ws, live).await {
|
|
|
|
info!("Dropping WebSocket after error: {}", e);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn stream_live_m4s_chunk(
|
|
|
|
&self, open_id: u32, stream_id: i32,
|
|
|
|
ws: &mut tokio_tungstenite::WebSocketStream<hyper::upgrade::Upgraded>,
|
|
|
|
live: db::LiveSegment) -> Result<(), Error> {
|
|
|
|
let mut builder = mp4::FileBuilder::new(mp4::Type::MediaSegment);
|
|
|
|
let mut vse_id = None;
|
|
|
|
let mut start = None;
|
|
|
|
{
|
|
|
|
let db = self.0.db.lock();
|
|
|
|
let mut rows = 0;
|
|
|
|
db.list_recordings_by_id(stream_id, live.recording .. live.recording+1, &mut |r| {
|
|
|
|
rows += 1;
|
|
|
|
let vse = db.video_sample_entries_by_id().get(&r.video_sample_entry_id)
|
|
|
|
.unwrap();
|
|
|
|
vse_id = Some(strutil::hex(&vse.sha1));
|
|
|
|
start = Some(r.start);
|
|
|
|
builder.append(&db, r, live.off_90k.clone())?;
|
|
|
|
Ok(())
|
|
|
|
})?;
|
|
|
|
if rows != 1 {
|
|
|
|
bail_t!(Internal, "unable to find {:?}", live);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
let vse_id = vse_id.unwrap();
|
|
|
|
let start = start.unwrap();
|
|
|
|
use http_serve::Entity;
|
|
|
|
let mp4 = builder.build(self.0.db.clone(), self.0.dirs_by_stream_id.clone())?;
|
|
|
|
let mut hdrs = http::header::HeaderMap::new();
|
|
|
|
mp4.add_headers(&mut hdrs);
|
|
|
|
let mime_type = hdrs.get(http::header::CONTENT_TYPE).unwrap();
|
|
|
|
let hdr = format!(
|
|
|
|
"Content-Type: {}\r\n\
|
|
|
|
X-Recording-Start: {}\r\n\
|
|
|
|
X-Recording-Id: {}.{}\r\n\
|
|
|
|
X-Time-Range: {}-{}\r\n\
|
|
|
|
X-Video-Sample-Entry-Sha1: {}\r\n\r\n",
|
|
|
|
mime_type.to_str().unwrap(),
|
|
|
|
start.0,
|
|
|
|
open_id,
|
|
|
|
live.recording,
|
|
|
|
live.off_90k.start,
|
|
|
|
live.off_90k.end,
|
|
|
|
&vse_id);
|
|
|
|
let mut v = /*Pin::from(*/hdr.into_bytes()/*)*/;
|
|
|
|
mp4.append_into_vec(&mut v).await?;
|
|
|
|
//let v = Pin::into_inner();
|
|
|
|
ws.send(tungstenite::Message::Binary(v)).await?;
|
|
|
|
Ok(())
|
2019-01-21 15:58:52 -08:00
|
|
|
}
|
2019-06-14 16:11:12 -07:00
|
|
|
|
2019-06-19 15:17:50 -07:00
|
|
|
fn signals(&self, req: Request<hyper::Body>, caller: Caller)
|
2020-01-08 23:04:36 -08:00
|
|
|
-> Box<dyn Future<Output = Result<Response<Body>, Response<Body>>> + Send + Sync + 'static> {
|
2019-06-14 16:11:12 -07:00
|
|
|
use http::method::Method;
|
|
|
|
match *req.method() {
|
|
|
|
Method::POST => Box::new(with_json_body(req)
|
|
|
|
.and_then({
|
|
|
|
let s = self.0.clone();
|
2020-01-08 23:04:36 -08:00
|
|
|
move |(req, b)| future::ready(s.post_signals(&req, caller, b))
|
2019-06-14 16:11:12 -07:00
|
|
|
})),
|
2020-01-08 23:04:36 -08:00
|
|
|
Method::GET | Method::HEAD => Box::new(future::ready(self.0.get_signals(&req))),
|
2019-06-14 16:11:12 -07:00
|
|
|
_ => Box::new(future::err(plain_response(StatusCode::METHOD_NOT_ALLOWED,
|
|
|
|
"POST, GET, or HEAD expected"))),
|
|
|
|
}
|
|
|
|
}
|
2016-11-25 14:34:00 -08:00
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
pub fn serve(&mut self, req: Request<::hyper::Body>) -> BoxedFuture {
|
|
|
|
fn wrap<R>(is_private: bool, r: R) -> BoxedFuture
|
|
|
|
where R: Future<Output = Result<Response<Body>, Response<Body>>> + Send + Sync + 'static {
|
|
|
|
return Box::new(r.or_else(|e| futures::future::ok(e)).map_ok(move |mut r| {
|
2018-12-01 16:55:10 -08:00
|
|
|
if is_private {
|
|
|
|
r.headers_mut().insert("Cache-Control", HeaderValue::from_static("private"));
|
|
|
|
}
|
|
|
|
r
|
|
|
|
}))
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
|
2018-12-01 16:55:10 -08:00
|
|
|
fn wrap_r(is_private: bool, r: ResponseResult)
|
2020-01-08 23:04:36 -08:00
|
|
|
-> Box<dyn Future<Output = Result<Response<Body>, BoxedError>> + Send + Sync + 'static> {
|
|
|
|
return wrap(is_private, future::ready(r))
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
let p = Path::decode(req.uri().path());
|
2019-06-19 15:17:50 -07:00
|
|
|
let always_allow_unauthenticated = match p {
|
|
|
|
Path::NotFound | Path::Request | Path::Login | Path::Logout | Path::Static => true,
|
|
|
|
_ => false,
|
2018-11-25 21:31:50 -08:00
|
|
|
};
|
2019-06-19 15:17:50 -07:00
|
|
|
debug!("request on: {}: {:?}", req.uri(), p);
|
|
|
|
let caller = match self.0.authenticate(&req, always_allow_unauthenticated) {
|
|
|
|
Ok(c) => c,
|
|
|
|
Err(e) => return Box::new(future::ok(from_base_error(e))),
|
2018-11-25 21:31:50 -08:00
|
|
|
};
|
2018-12-01 16:55:10 -08:00
|
|
|
match p {
|
2018-12-29 13:06:44 -06:00
|
|
|
Path::InitSegment(sha1, debug) => wrap_r(true, self.0.init_segment(sha1, debug, &req)),
|
2019-06-19 15:17:50 -07:00
|
|
|
Path::TopLevel => wrap_r(true, self.0.top_level(&req, caller)),
|
2018-12-01 16:55:10 -08:00
|
|
|
Path::Request => wrap_r(true, self.0.request(&req)),
|
|
|
|
Path::Camera(uuid) => wrap_r(true, self.0.camera(&req, uuid)),
|
2018-11-25 21:31:50 -08:00
|
|
|
Path::StreamRecordings(uuid, type_) => {
|
2018-12-01 16:55:10 -08:00
|
|
|
wrap_r(true, self.0.stream_recordings(&req, uuid, type_))
|
2018-11-25 21:31:50 -08:00
|
|
|
},
|
2018-12-29 13:06:44 -06:00
|
|
|
Path::StreamViewMp4(uuid, type_, debug) => {
|
2019-06-19 15:17:50 -07:00
|
|
|
wrap_r(true, self.0.stream_view_mp4(&req, caller, uuid, type_, mp4::Type::Normal,
|
|
|
|
debug))
|
2017-10-01 15:29:22 -07:00
|
|
|
},
|
2018-12-29 13:06:44 -06:00
|
|
|
Path::StreamViewMp4Segment(uuid, type_, debug) => {
|
2019-06-19 15:17:50 -07:00
|
|
|
wrap_r(true, self.0.stream_view_mp4(&req, caller, uuid, type_,
|
|
|
|
mp4::Type::MediaSegment, debug))
|
2017-10-01 15:29:22 -07:00
|
|
|
},
|
2019-01-21 15:58:52 -08:00
|
|
|
Path::StreamLiveMp4Segments(uuid, type_) => {
|
2020-02-28 21:41:31 -08:00
|
|
|
wrap_r(true, self.stream_live_m4s(req, caller, uuid, type_))
|
2019-01-21 15:58:52 -08:00
|
|
|
},
|
2018-12-01 16:55:10 -08:00
|
|
|
Path::NotFound => wrap(true, future::err(not_found("path not understood"))),
|
2020-01-08 23:23:58 -08:00
|
|
|
Path::Login => wrap(true, with_json_body(req).and_then({
|
2018-11-25 21:31:50 -08:00
|
|
|
let s = self.clone();
|
2020-01-08 23:04:36 -08:00
|
|
|
move |(req, b)| future::ready(s.0.login(&req, b))
|
2018-11-25 21:31:50 -08:00
|
|
|
})),
|
2020-01-08 23:23:58 -08:00
|
|
|
Path::Logout => wrap(true, with_json_body(req).and_then({
|
2018-11-25 21:31:50 -08:00
|
|
|
let s = self.clone();
|
2020-01-08 23:04:36 -08:00
|
|
|
move |(req, b)| future::ready(s.0.logout(&req, b))
|
2018-11-25 21:31:50 -08:00
|
|
|
})),
|
2020-01-08 23:04:36 -08:00
|
|
|
Path::Signals => wrap(true, Pin::from(self.signals(req, caller))),
|
2018-12-01 16:55:10 -08:00
|
|
|
Path::Static => wrap_r(false, self.0.static_file(&req, req.uri().path())),
|
2018-08-29 22:26:19 -07:00
|
|
|
}
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
2018-12-28 21:53:29 -06:00
|
|
|
use db::testutil::{self, TestDb};
|
2020-01-08 23:04:36 -08:00
|
|
|
use futures::future::FutureExt;
|
2018-12-28 21:53:29 -06:00
|
|
|
use log::info;
|
2018-11-25 21:31:50 -08:00
|
|
|
use std::collections::HashMap;
|
2017-10-21 21:54:27 -07:00
|
|
|
use super::Segments;
|
2016-11-25 14:34:00 -08:00
|
|
|
|
2018-11-25 21:31:50 -08:00
|
|
|
struct Server {
|
2018-12-28 21:53:29 -06:00
|
|
|
db: TestDb<base::clock::RealClocks>,
|
2018-11-25 21:31:50 -08:00
|
|
|
base_url: String,
|
|
|
|
//test_camera_uuid: Uuid,
|
|
|
|
handle: Option<::std::thread::JoinHandle<()>>,
|
2020-01-08 23:04:36 -08:00
|
|
|
shutdown_tx: Option<futures::channel::oneshot::Sender<()>>,
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Server {
|
2019-06-19 15:17:50 -07:00
|
|
|
fn new(allow_unauthenticated_permissions: Option<db::Permissions>) -> Server {
|
2018-12-28 21:53:29 -06:00
|
|
|
let db = TestDb::new(base::clock::RealClocks {});
|
2020-01-08 23:04:36 -08:00
|
|
|
let (shutdown_tx, shutdown_rx) = futures::channel::oneshot::channel::<()>();
|
2018-11-28 14:22:30 -08:00
|
|
|
let service = super::Service::new(super::Config {
|
|
|
|
db: db.db.clone(),
|
|
|
|
ui_dir: None,
|
2019-06-19 15:17:50 -07:00
|
|
|
allow_unauthenticated_permissions,
|
2018-11-28 14:22:30 -08:00
|
|
|
trust_forward_hdrs: true,
|
|
|
|
time_zone_name: "".to_owned(),
|
|
|
|
}).unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
let make_svc = hyper::service::make_service_fn(move |_conn| {
|
|
|
|
futures::future::ok::<_, std::convert::Infallible>(hyper::service::service_fn({
|
|
|
|
let mut s = service.clone();
|
|
|
|
move |req| std::pin::Pin::from(s.serve(req))
|
|
|
|
}))
|
|
|
|
});
|
2020-01-09 20:06:30 -08:00
|
|
|
let (tx, rx) = std::sync::mpsc::channel();
|
|
|
|
let handle = ::std::thread::spawn(move || {
|
2020-01-08 23:04:36 -08:00
|
|
|
let addr = ([127, 0, 0, 1], 0).into();
|
2020-01-09 20:06:30 -08:00
|
|
|
let mut rt = tokio::runtime::Runtime::new().unwrap();
|
|
|
|
let srv = rt.enter(|| {
|
|
|
|
hyper::server::Server::bind(&addr)
|
2020-01-08 23:04:36 -08:00
|
|
|
.tcp_nodelay(true)
|
|
|
|
.serve(make_svc)
|
2020-01-09 20:06:30 -08:00
|
|
|
});
|
|
|
|
let addr = srv.local_addr(); // resolve port 0 to a real ephemeral port number.
|
|
|
|
tx.send(addr).unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
rt.block_on(srv.with_graceful_shutdown(shutdown_rx.map(|_| ()))).unwrap();
|
2018-11-25 21:31:50 -08:00
|
|
|
});
|
2020-01-09 20:06:30 -08:00
|
|
|
let addr = rx.recv().unwrap();
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
// Create a user.
|
|
|
|
let mut c = db::UserChange::add_user("slamb".to_owned());
|
|
|
|
c.set_password("hunter2".to_owned());
|
|
|
|
db.db.lock().apply_user_change(c).unwrap();
|
|
|
|
|
|
|
|
Server {
|
|
|
|
db,
|
|
|
|
base_url: format!("http://{}:{}", addr.ip(), addr.port()),
|
|
|
|
handle: Some(handle),
|
|
|
|
shutdown_tx: Some(shutdown_tx),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Drop for Server {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
self.shutdown_tx.take().unwrap().send(()).unwrap();
|
|
|
|
self.handle.take().unwrap().join().unwrap()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Debug, Default)]
|
|
|
|
struct SessionCookie(Option<String>);
|
|
|
|
|
|
|
|
impl SessionCookie {
|
2020-01-08 23:04:36 -08:00
|
|
|
pub fn new(headers: &reqwest::header::HeaderMap) -> Self {
|
2018-11-25 21:31:50 -08:00
|
|
|
let mut c = SessionCookie::default();
|
|
|
|
c.update(headers);
|
|
|
|
c
|
|
|
|
}
|
|
|
|
|
2020-01-08 23:04:36 -08:00
|
|
|
pub fn update(&mut self, headers: &reqwest::header::HeaderMap) {
|
|
|
|
for set_cookie in headers.get_all(reqwest::header::SET_COOKIE) {
|
2018-11-25 21:31:50 -08:00
|
|
|
let mut set_cookie = set_cookie.to_str().unwrap().split("; ");
|
|
|
|
let c = set_cookie.next().unwrap();
|
|
|
|
let mut clear = false;
|
|
|
|
for attr in set_cookie {
|
|
|
|
if attr == "Max-Age=0" {
|
|
|
|
clear = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !c.starts_with("s=") {
|
|
|
|
panic!("unrecognized cookie");
|
|
|
|
}
|
|
|
|
self.0 = if clear { None } else { Some(c.to_owned()) };
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Produces a `Cookie` header value.
|
|
|
|
pub fn header(&self) -> String {
|
|
|
|
self.0.as_ref().map(|s| s.as_str()).unwrap_or("").to_owned()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-29 13:06:44 -06:00
|
|
|
#[test]
|
|
|
|
fn paths() {
|
|
|
|
use super::Path;
|
|
|
|
use uuid::Uuid;
|
|
|
|
let cam_uuid = Uuid::parse_str("35144640-ff1e-4619-b0d5-4c74c185741c").unwrap();
|
|
|
|
assert_eq!(Path::decode("/foo"), Path::Static);
|
|
|
|
assert_eq!(Path::decode("/api/"), Path::TopLevel);
|
|
|
|
assert_eq!(Path::decode("/api/init/07cec464126825088ea86a07eddd6a00afa71559.mp4"),
|
|
|
|
Path::InitSegment([0x07, 0xce, 0xc4, 0x64, 0x12, 0x68, 0x25, 0x08, 0x8e, 0xa8,
|
|
|
|
0x6a, 0x07, 0xed, 0xdd, 0x6a, 0x00, 0xaf, 0xa7, 0x15, 0x59],
|
|
|
|
false));
|
|
|
|
assert_eq!(Path::decode("/api/init/07cec464126825088ea86a07eddd6a00afa71559.mp4.txt"),
|
|
|
|
Path::InitSegment([0x07, 0xce, 0xc4, 0x64, 0x12, 0x68, 0x25, 0x08, 0x8e, 0xa8,
|
|
|
|
0x6a, 0x07, 0xed, 0xdd, 0x6a, 0x00, 0xaf, 0xa7, 0x15, 0x59],
|
|
|
|
true));
|
|
|
|
assert_eq!(Path::decode("/api/init/000000000000000000000000000000000000000x.mp4"),
|
|
|
|
Path::NotFound); // non-hexadigit
|
|
|
|
assert_eq!(Path::decode("/api/init/000000000000000000000000000000000000000.mp4"),
|
|
|
|
Path::NotFound); // too short
|
|
|
|
assert_eq!(Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/"),
|
|
|
|
Path::Camera(cam_uuid));
|
|
|
|
assert_eq!(Path::decode("/api/cameras/asdf/"), Path::NotFound);
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/recordings"),
|
|
|
|
Path::StreamRecordings(cam_uuid, db::StreamType::MAIN));
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/sub/recordings"),
|
|
|
|
Path::StreamRecordings(cam_uuid, db::StreamType::SUB));
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/junk/recordings"),
|
|
|
|
Path::NotFound);
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/view.mp4"),
|
|
|
|
Path::StreamViewMp4(cam_uuid, db::StreamType::MAIN, false));
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/view.mp4.txt"),
|
|
|
|
Path::StreamViewMp4(cam_uuid, db::StreamType::MAIN, true));
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/view.m4s"),
|
|
|
|
Path::StreamViewMp4Segment(cam_uuid, db::StreamType::MAIN, false));
|
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/view.m4s.txt"),
|
|
|
|
Path::StreamViewMp4Segment(cam_uuid, db::StreamType::MAIN, true));
|
2019-01-21 15:58:52 -08:00
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/live.m4s"),
|
|
|
|
Path::StreamLiveMp4Segments(cam_uuid, db::StreamType::MAIN));
|
2018-12-29 13:06:44 -06:00
|
|
|
assert_eq!(
|
|
|
|
Path::decode("/api/cameras/35144640-ff1e-4619-b0d5-4c74c185741c/main/junk"),
|
|
|
|
Path::NotFound);
|
|
|
|
assert_eq!(Path::decode("/api/login"), Path::Login);
|
|
|
|
assert_eq!(Path::decode("/api/logout"), Path::Logout);
|
2019-06-06 16:18:13 -07:00
|
|
|
assert_eq!(Path::decode("/api/signals"), Path::Signals);
|
2018-12-29 13:06:44 -06:00
|
|
|
assert_eq!(Path::decode("/api/junk"), Path::NotFound);
|
|
|
|
}
|
|
|
|
|
2016-12-20 22:08:18 -08:00
|
|
|
#[test]
|
|
|
|
fn test_segments() {
|
|
|
|
testutil::init();
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: None, start_time: 0, end_time: None},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: Some(42), start_time: 0, end_time: None},
|
|
|
|
Segments::parse("1@42").unwrap());
|
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: None, start_time: 26, end_time: None},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1.26-").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: Some(42), start_time: 26, end_time: None},
|
|
|
|
Segments::parse("1@42.26-").unwrap());
|
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: None, start_time: 0, end_time: Some(42)},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1.-42").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..2, open_id: None, start_time: 26, end_time: Some(42)},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1.26-42").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..6, open_id: None, start_time: 0, end_time: None},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1-5").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..6, open_id: None, start_time: 26, end_time: None},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1-5.26-").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..6, open_id: None, start_time: 0, end_time: Some(42)},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1-5.-42").unwrap());
|
2018-03-02 11:38:11 -08:00
|
|
|
assert_eq!(Segments{ids: 1..6, open_id: None, start_time: 26, end_time: Some(42)},
|
2016-12-20 22:08:18 -08:00
|
|
|
Segments::parse("1-5.26-42").unwrap());
|
|
|
|
}
|
2018-11-25 21:31:50 -08:00
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
#[tokio::test]
|
|
|
|
async fn unauthorized_without_cookie() {
|
2018-11-25 21:31:50 -08:00
|
|
|
testutil::init();
|
2019-06-19 15:17:50 -07:00
|
|
|
let s = Server::new(None);
|
2018-11-25 21:31:50 -08:00
|
|
|
let cli = reqwest::Client::new();
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.get(&format!("{}/api/", &s.base_url)).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::UNAUTHORIZED);
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
#[tokio::test]
|
|
|
|
async fn login() {
|
2018-11-25 21:31:50 -08:00
|
|
|
testutil::init();
|
2019-06-19 15:17:50 -07:00
|
|
|
let s = Server::new(None);
|
2018-11-25 21:31:50 -08:00
|
|
|
let cli = reqwest::Client::new();
|
|
|
|
let login_url = format!("{}/api/login", &s.base_url);
|
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.get(&login_url).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::METHOD_NOT_ALLOWED);
|
2018-11-25 21:31:50 -08:00
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.post(&login_url).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::BAD_REQUEST);
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
let mut p = HashMap::new();
|
|
|
|
p.insert("username", "slamb");
|
|
|
|
p.insert("password", "asdf");
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.post(&login_url).json(&p).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::UNAUTHORIZED);
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
p.insert("password", "hunter2");
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.post(&login_url).json(&p).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::NO_CONTENT);
|
2018-11-25 21:31:50 -08:00
|
|
|
let cookie = SessionCookie::new(resp.headers());
|
|
|
|
info!("cookie: {:?}", cookie);
|
|
|
|
info!("header: {}", cookie.header());
|
|
|
|
|
|
|
|
let resp = cli.get(&format!("{}/api/", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2018-11-25 21:31:50 -08:00
|
|
|
.send()
|
2020-01-09 20:06:30 -08:00
|
|
|
.await
|
2018-11-25 21:31:50 -08:00
|
|
|
.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::OK);
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
#[tokio::test]
|
|
|
|
async fn logout() {
|
2018-11-25 21:31:50 -08:00
|
|
|
testutil::init();
|
2019-06-19 15:17:50 -07:00
|
|
|
let s = Server::new(None);
|
2018-11-25 21:31:50 -08:00
|
|
|
let cli = reqwest::Client::new();
|
|
|
|
let mut p = HashMap::new();
|
|
|
|
p.insert("username", "slamb");
|
|
|
|
p.insert("password", "hunter2");
|
2020-01-09 20:06:30 -08:00
|
|
|
let resp = cli.post(&format!("{}/api/login", &s.base_url)).json(&p).send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::NO_CONTENT);
|
2018-11-25 21:31:50 -08:00
|
|
|
let cookie = SessionCookie::new(resp.headers());
|
|
|
|
|
|
|
|
// A GET shouldn't work.
|
|
|
|
let resp = cli.get(&format!("{}/api/logout", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2018-11-25 21:31:50 -08:00
|
|
|
.send()
|
2020-01-09 20:06:30 -08:00
|
|
|
.await
|
2018-11-25 21:31:50 -08:00
|
|
|
.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::METHOD_NOT_ALLOWED);
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
// Neither should a POST without a csrf token.
|
|
|
|
let resp = cli.post(&format!("{}/api/logout", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2018-11-25 21:31:50 -08:00
|
|
|
.send()
|
2020-01-09 20:06:30 -08:00
|
|
|
.await
|
2018-11-25 21:31:50 -08:00
|
|
|
.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::BAD_REQUEST);
|
2018-11-25 21:31:50 -08:00
|
|
|
|
|
|
|
// But it should work with the csrf token.
|
|
|
|
// Retrieve that from the toplevel API request.
|
|
|
|
let toplevel: serde_json::Value = cli.post(&format!("{}/api/", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2020-01-09 20:06:30 -08:00
|
|
|
.send().await.unwrap()
|
|
|
|
.json().await.unwrap();
|
2018-11-25 21:31:50 -08:00
|
|
|
let csrf = toplevel.get("session").unwrap().get("csrf").unwrap().as_str();
|
|
|
|
let mut p = HashMap::new();
|
|
|
|
p.insert("csrf", csrf);
|
|
|
|
let resp = cli.post(&format!("{}/api/logout", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2020-01-08 23:23:58 -08:00
|
|
|
.json(&p)
|
2018-11-25 21:31:50 -08:00
|
|
|
.send()
|
2020-01-09 20:06:30 -08:00
|
|
|
.await
|
2018-11-25 21:31:50 -08:00
|
|
|
.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::NO_CONTENT);
|
2018-11-25 21:31:50 -08:00
|
|
|
let mut updated_cookie = cookie.clone();
|
|
|
|
updated_cookie.update(resp.headers());
|
|
|
|
|
|
|
|
// The cookie should be cleared client-side.
|
|
|
|
assert!(updated_cookie.0.is_none());
|
|
|
|
|
|
|
|
// It should also be invalidated server-side.
|
|
|
|
let resp = cli.get(&format!("{}/api/", &s.base_url))
|
2020-01-08 23:04:36 -08:00
|
|
|
.header(reqwest::header::COOKIE, cookie.header())
|
2018-11-25 21:31:50 -08:00
|
|
|
.send()
|
2020-01-09 20:06:30 -08:00
|
|
|
.await
|
2018-11-25 21:31:50 -08:00
|
|
|
.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::UNAUTHORIZED);
|
2018-11-25 21:31:50 -08:00
|
|
|
}
|
2018-12-28 17:30:33 -06:00
|
|
|
|
2020-01-09 20:06:30 -08:00
|
|
|
#[tokio::test]
|
|
|
|
async fn view_without_segments() {
|
2018-12-28 17:30:33 -06:00
|
|
|
testutil::init();
|
2019-06-19 15:17:50 -07:00
|
|
|
let mut permissions = db::Permissions::new();
|
|
|
|
permissions.view_video = true;
|
|
|
|
let s = Server::new(Some(permissions));
|
2018-12-28 17:30:33 -06:00
|
|
|
let cli = reqwest::Client::new();
|
|
|
|
let resp = cli.get(
|
|
|
|
&format!("{}/api/cameras/{}/main/view.mp4", &s.base_url, s.db.test_camera_uuid))
|
2020-01-09 20:06:30 -08:00
|
|
|
.send().await.unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::BAD_REQUEST);
|
2018-12-28 17:30:33 -06:00
|
|
|
}
|
2016-11-25 14:34:00 -08:00
|
|
|
}
|
2017-02-12 20:37:03 -08:00
|
|
|
|
|
|
|
#[cfg(all(test, feature="nightly"))]
|
|
|
|
mod bench {
|
|
|
|
extern crate test;
|
|
|
|
|
2018-02-20 23:15:39 -08:00
|
|
|
use db::testutil::{self, TestDb};
|
2017-02-12 20:37:03 -08:00
|
|
|
use hyper;
|
2018-12-28 21:53:29 -06:00
|
|
|
use lazy_static::lazy_static;
|
2018-02-03 21:56:04 -08:00
|
|
|
use uuid::Uuid;
|
2017-02-12 20:37:03 -08:00
|
|
|
|
|
|
|
struct Server {
|
|
|
|
base_url: String,
|
2018-02-03 21:56:04 -08:00
|
|
|
test_camera_uuid: Uuid,
|
2017-02-12 20:37:03 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Server {
|
|
|
|
fn new() -> Server {
|
2018-03-23 15:16:43 -07:00
|
|
|
let db = TestDb::new(::base::clock::RealClocks {});
|
2018-02-03 21:56:04 -08:00
|
|
|
let test_camera_uuid = db.test_camera_uuid;
|
2017-02-12 20:37:03 -08:00
|
|
|
testutil::add_dummy_recordings_to_db(&db.db, 1440);
|
2018-11-28 14:22:30 -08:00
|
|
|
let service = super::Service::new(super::Config {
|
|
|
|
db: db.db.clone(),
|
|
|
|
ui_dir: None,
|
2019-06-20 14:57:13 -07:00
|
|
|
allow_unauthenticated_permissions: Some(db::Permissions::default()),
|
2018-11-28 14:22:30 -08:00
|
|
|
trust_forward_hdrs: false,
|
|
|
|
time_zone_name: "".to_owned(),
|
|
|
|
}).unwrap();
|
2020-01-08 23:04:36 -08:00
|
|
|
let make_svc = hyper::service::make_service_fn(move |_conn| {
|
|
|
|
futures::future::ok::<_, std::convert::Infallible>(hyper::service::service_fn({
|
|
|
|
let mut s = service.clone();
|
|
|
|
move |req| std::pin::Pin::from(s.serve(req))
|
|
|
|
}))
|
|
|
|
});
|
|
|
|
let mut rt = tokio::runtime::Runtime::new().unwrap();
|
|
|
|
let srv = rt.enter(|| {
|
|
|
|
let addr = ([127, 0, 0, 1], 0).into();
|
|
|
|
hyper::server::Server::bind(&addr)
|
|
|
|
.tcp_nodelay(true)
|
|
|
|
.serve(make_svc)
|
|
|
|
});
|
|
|
|
let addr = srv.local_addr(); // resolve port 0 to a real ephemeral port number.
|
2017-02-12 20:37:03 -08:00
|
|
|
::std::thread::spawn(move || {
|
2020-01-08 23:04:36 -08:00
|
|
|
rt.block_on(srv).unwrap();
|
2017-02-12 20:37:03 -08:00
|
|
|
});
|
2018-02-03 21:56:04 -08:00
|
|
|
Server {
|
|
|
|
base_url: format!("http://{}:{}", addr.ip(), addr.port()),
|
|
|
|
test_camera_uuid,
|
|
|
|
}
|
2017-02-12 20:37:03 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
lazy_static! {
|
|
|
|
static ref SERVER: Server = { Server::new() };
|
|
|
|
}
|
|
|
|
|
|
|
|
#[bench]
|
2018-12-28 21:53:29 -06:00
|
|
|
fn serve_stream_recordings(b: &mut test::Bencher) {
|
2017-02-12 20:37:03 -08:00
|
|
|
testutil::init();
|
|
|
|
let server = &*SERVER;
|
2018-01-23 11:05:07 -08:00
|
|
|
let url = reqwest::Url::parse(&format!("{}/api/cameras/{}/main/recordings", server.base_url,
|
2018-02-03 21:56:04 -08:00
|
|
|
server.test_camera_uuid)).unwrap();
|
2017-11-16 23:01:09 -08:00
|
|
|
let client = reqwest::Client::new();
|
2020-02-16 23:58:07 -08:00
|
|
|
let mut rt = tokio::runtime::Runtime::new().unwrap();
|
2017-03-03 22:26:29 -08:00
|
|
|
let mut f = || {
|
2020-02-16 23:58:07 -08:00
|
|
|
rt.block_on(async {
|
|
|
|
let resp = client.get(url.clone()).send().await.unwrap();
|
|
|
|
assert_eq!(resp.status(), reqwest::StatusCode::OK);
|
|
|
|
let _b = resp.bytes().await.unwrap();
|
|
|
|
});
|
2017-03-03 22:26:29 -08:00
|
|
|
};
|
|
|
|
f(); // warm.
|
|
|
|
b.iter(f);
|
2017-02-12 20:37:03 -08:00
|
|
|
}
|
|
|
|
}
|