330 lines
10 KiB
C++
330 lines
10 KiB
C++
//
|
|
// ZoneMinder Streaming Server, $Date$, $Revision$
|
|
// Copyright (C) 2001-2008 Philip Coombes
|
|
//
|
|
// 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 2
|
|
// of the License, or (at your option) any later version.
|
|
//
|
|
// 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, write to the Free Software
|
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
//
|
|
|
|
#include "zm.h"
|
|
#include "zm_db.h"
|
|
#include "zm_user.h"
|
|
#include "zm_signal.h"
|
|
#include "zm_monitorstream.h"
|
|
#include "zm_eventstream.h"
|
|
#include "zm_fifo_stream.h"
|
|
#include <string>
|
|
#include <unistd.h>
|
|
|
|
bool ValidateAccess(User *user, int mon_id) {
|
|
bool allowed = true;
|
|
|
|
if ( mon_id > 0 ) {
|
|
if ( user->getStream() < User::PERM_VIEW )
|
|
allowed = false;
|
|
if ( !user->canAccess(mon_id) )
|
|
allowed = false;
|
|
} else {
|
|
if ( user->getEvents() < User::PERM_VIEW )
|
|
allowed = false;
|
|
}
|
|
if ( !allowed ) {
|
|
Error("Insufficient privileges for request user %d %s for monitor %d",
|
|
user->Id(), user->getUsername(), mon_id);
|
|
}
|
|
return allowed;
|
|
}
|
|
|
|
int main(int argc, const char *argv[], char **envp) {
|
|
self = argv[0];
|
|
|
|
srand(getpid() * time(nullptr));
|
|
|
|
enum { ZMS_UNKNOWN, ZMS_MONITOR, ZMS_EVENT, ZMS_FIFO } source = ZMS_UNKNOWN;
|
|
enum { ZMS_JPEG, ZMS_MPEG, ZMS_RAW, ZMS_ZIP, ZMS_SINGLE } mode = ZMS_JPEG;
|
|
char format[32] = "";
|
|
int monitor_id = 0;
|
|
time_t event_time = 0;
|
|
uint64_t event_id = 0;
|
|
unsigned int frame_id = 1;
|
|
unsigned int scale = 100;
|
|
unsigned int rate = 100;
|
|
double maxfps = 10.0;
|
|
unsigned int bitrate = 100000;
|
|
unsigned int ttl = 0;
|
|
EventStream::StreamMode replay = EventStream::MODE_NONE;
|
|
std::string username;
|
|
std::string password;
|
|
char auth[64] = "";
|
|
std::string jwt_token_str = "";
|
|
unsigned int connkey = 0;
|
|
unsigned int playback_buffer = 0;
|
|
|
|
bool nph = false;
|
|
const char *basename = strrchr(argv[0], '/');
|
|
if ( basename ) // if we found a / lets skip past it
|
|
basename++;
|
|
else // argv[0] might not contain the full path, but just the script name
|
|
basename = argv[0];
|
|
const char *nph_prefix = "nph-";
|
|
if ( basename && !strncmp(basename, nph_prefix, strlen(nph_prefix)) ) {
|
|
nph = true;
|
|
}
|
|
|
|
char log_id_string[32] = "zms";
|
|
logInit(log_id_string);
|
|
zmLoadStaticConfig();
|
|
zmDbConnect();
|
|
zmLoadDBConfig();
|
|
logInit(log_id_string);
|
|
|
|
for (char **env = envp; *env != 0; env++) {
|
|
char *thisEnv = *env;
|
|
Debug(1, "env: %s", thisEnv);
|
|
}
|
|
|
|
const char *query = getenv("QUERY_STRING");
|
|
if ( query == nullptr ) {
|
|
Fatal("No query string.");
|
|
return 0;
|
|
} // end if query
|
|
|
|
Debug(1, "Query: %s", query);
|
|
|
|
char *q_ptr = (char *)query;
|
|
char *parms[16]; // Shouldn't be more than this
|
|
int parm_no = 0;
|
|
while ( (parm_no < 16) && (parms[parm_no] = strtok(q_ptr, "&")) ) {
|
|
parm_no++;
|
|
q_ptr = nullptr;
|
|
}
|
|
|
|
for ( int p = 0; p < parm_no; p++ ) {
|
|
char *name = strtok(parms[p], "=");
|
|
char const *value = strtok(nullptr, "=");
|
|
if ( !value )
|
|
value = "";
|
|
if ( !strcmp(name, "source") ) {
|
|
if ( !strcmp(value, "event") ) {
|
|
source = ZMS_EVENT;
|
|
} else if ( !strcmp(value, "fifo") ) {
|
|
source = ZMS_FIFO;
|
|
} else {
|
|
source = ZMS_MONITOR;
|
|
}
|
|
} else if ( !strcmp(name, "mode") ) {
|
|
mode = !strcmp(value, "jpeg")?ZMS_JPEG:ZMS_MPEG;
|
|
mode = !strcmp(value, "raw")?ZMS_RAW:mode;
|
|
mode = !strcmp(value, "zip")?ZMS_ZIP:mode;
|
|
mode = !strcmp(value, "single")?ZMS_SINGLE:mode;
|
|
} else if ( !strcmp(name, "format") ) {
|
|
strncpy(format, value, sizeof(format)-1);
|
|
} else if ( !strcmp(name, "monitor") ) {
|
|
monitor_id = atoi(value);
|
|
if ( source == ZMS_UNKNOWN )
|
|
source = ZMS_MONITOR;
|
|
} else if ( !strcmp(name, "time") ) {
|
|
event_time = atoi(value);
|
|
} else if ( !strcmp(name, "event") ) {
|
|
event_id = strtoull(value, nullptr, 10);
|
|
source = ZMS_EVENT;
|
|
} else if ( !strcmp(name, "frame") ) {
|
|
frame_id = strtoull(value, nullptr, 10);
|
|
source = ZMS_EVENT;
|
|
} else if ( !strcmp(name, "scale") ) {
|
|
scale = atoi(value);
|
|
} else if ( !strcmp(name, "rate") ) {
|
|
rate = atoi(value);
|
|
} else if ( !strcmp(name, "maxfps") ) {
|
|
maxfps = atof(value);
|
|
} else if ( !strcmp(name, "bitrate") ) {
|
|
bitrate = atoi(value);
|
|
} else if ( !strcmp(name, "ttl") ) {
|
|
ttl = atoi(value);
|
|
} else if ( !strcmp(name, "replay") ) {
|
|
if ( !strcmp(value, "gapless") ) {
|
|
replay = EventStream::MODE_ALL_GAPLESS;
|
|
} else if ( !strcmp(value, "all") ) {
|
|
replay = EventStream::MODE_ALL;
|
|
} else if ( !strcmp(value, "none") ) {
|
|
replay = EventStream::MODE_NONE;
|
|
} else if ( !strcmp(value, "single") ) {
|
|
replay = EventStream::MODE_SINGLE;
|
|
} else {
|
|
Error("Unsupported value %s for replay, defaulting to none", value);
|
|
}
|
|
} else if ( !strcmp(name, "connkey") ) {
|
|
connkey = atoi(value);
|
|
} else if ( !strcmp(name, "buffer") ) {
|
|
playback_buffer = atoi(value);
|
|
} else if ( !strcmp(name, "auth") ) {
|
|
strncpy(auth, value, sizeof(auth)-1);
|
|
} else if ( !strcmp(name, "token") ) {
|
|
jwt_token_str = value;
|
|
Debug(1, "ZMS: JWT token found: %s", jwt_token_str.c_str());
|
|
} else if ( !strcmp(name, "user") ) {
|
|
username = UriDecode(value);
|
|
} else if ( !strcmp(name, "pass") ) {
|
|
password = UriDecode(value);
|
|
Debug(1, "Have %s for password", password.c_str());
|
|
} else {
|
|
Debug(1, "Unknown parameter passed to zms %s=%s", name, value);
|
|
} // end if possible parameter names
|
|
} // end foreach parm
|
|
|
|
if ( monitor_id ) {
|
|
snprintf(log_id_string, sizeof(log_id_string), "zms_m%d", monitor_id);
|
|
} else {
|
|
snprintf(log_id_string, sizeof(log_id_string), "zms_e%" PRIu64, event_id);
|
|
}
|
|
logInit(log_id_string);
|
|
|
|
if ( config.opt_use_auth ) {
|
|
User *user = nullptr;
|
|
|
|
if ( jwt_token_str != "" ) {
|
|
// user = zmLoadTokenUser(jwt_token_str, config.auth_hash_ips);
|
|
user = zmLoadTokenUser(jwt_token_str, false);
|
|
} else if ( strcmp(config.auth_relay, "none") == 0 ) {
|
|
if ( checkUser(username.c_str()) ) {
|
|
user = zmLoadUser(username.c_str());
|
|
} else {
|
|
Error("Bad username");
|
|
}
|
|
} else {
|
|
if ( *auth ) {
|
|
user = zmLoadAuthUser(auth, config.auth_hash_ips);
|
|
} else if ( username.length() && password.length() ) {
|
|
user = zmLoadUser(username.c_str(), password.c_str());
|
|
}
|
|
}
|
|
if ( !user ) {
|
|
fputs("HTTP/1.0 403 Forbidden\r\n\r\n", stdout);
|
|
|
|
const char *referer = getenv("HTTP_REFERER");
|
|
Error("Unable to authenticate user from %s", referer);
|
|
logTerm();
|
|
zmDbClose();
|
|
return 0;
|
|
}
|
|
if ( !ValidateAccess(user, monitor_id) ) {
|
|
delete user;
|
|
user = nullptr;
|
|
fputs("HTTP/1.0 403 Forbidden\r\n\r\n", stdout);
|
|
logTerm();
|
|
zmDbClose();
|
|
return 0;
|
|
}
|
|
delete user;
|
|
user = nullptr;
|
|
} // end if config.opt_use_auth
|
|
|
|
HwCapsDetect();
|
|
Image::Initialise();
|
|
zmSetDefaultTermHandler();
|
|
zmSetDefaultDieHandler();
|
|
|
|
setbuf(stdout, nullptr);
|
|
if ( nph ) {
|
|
fputs("HTTP/1.0 200 OK\r\n", stdout);
|
|
}
|
|
fprintf(stdout, "Server: ZoneMinder Video Server/%s\r\n", ZM_VERSION);
|
|
|
|
time_t now = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
|
|
char date_string[64];
|
|
tm now_tm = {};
|
|
strftime(date_string, sizeof(date_string)-1,
|
|
"%a, %d %b %Y %H:%M:%S GMT", gmtime_r(&now, &now_tm));
|
|
|
|
fputs("Last-Modified: ", stdout);
|
|
fputs(date_string, stdout);
|
|
fputs(
|
|
"\r\nExpires: Mon, 26 Jul 1997 05:00:00 GMT\r\n"
|
|
"Cache-Control: no-store, no-cache, must-revalidate\r\n"
|
|
"Cache-Control: post-check=0, pre-check=0\r\n"
|
|
"Pragma: no-cache\r\n",
|
|
stdout);
|
|
|
|
if ( source == ZMS_MONITOR ) {
|
|
MonitorStream stream;
|
|
stream.setStreamScale(scale);
|
|
stream.setStreamReplayRate(rate);
|
|
stream.setStreamMaxFPS(maxfps);
|
|
stream.setStreamTTL(ttl);
|
|
stream.setStreamQueue(connkey);
|
|
stream.setStreamBuffer(playback_buffer);
|
|
if ( !stream.setStreamStart(monitor_id) ) {
|
|
fputs("Content-Type: multipart/x-mixed-replace; boundary=" BOUNDARY "\r\n\r\n", stdout);
|
|
stream.sendTextFrame("Unable to connect to monitor");
|
|
logTerm();
|
|
zmDbClose();
|
|
return -1;
|
|
}
|
|
|
|
if ( mode == ZMS_JPEG ) {
|
|
stream.setStreamType(MonitorStream::STREAM_JPEG);
|
|
} else if ( mode == ZMS_RAW ) {
|
|
stream.setStreamType(MonitorStream::STREAM_RAW);
|
|
} else if ( mode == ZMS_ZIP ) {
|
|
stream.setStreamType(MonitorStream::STREAM_ZIP);
|
|
} else if ( mode == ZMS_SINGLE ) {
|
|
stream.setStreamType(MonitorStream::STREAM_SINGLE);
|
|
} else {
|
|
stream.setStreamFormat(format);
|
|
stream.setStreamBitrate(bitrate);
|
|
stream.setStreamType(MonitorStream::STREAM_MPEG);
|
|
}
|
|
stream.runStream();
|
|
} else if ( source == ZMS_FIFO ) {
|
|
FifoStream stream;
|
|
stream.setStreamMaxFPS(maxfps);
|
|
stream.setStreamStart(monitor_id, format);
|
|
stream.runStream();
|
|
} else if ( source == ZMS_EVENT ) {
|
|
if ( !event_id ) {
|
|
Fatal("Can't view an event without specifying an event_id.");
|
|
}
|
|
EventStream stream;
|
|
stream.setStreamScale(scale);
|
|
stream.setStreamReplayRate(rate);
|
|
stream.setStreamMaxFPS(maxfps);
|
|
stream.setStreamMode(replay);
|
|
stream.setStreamQueue(connkey);
|
|
if ( monitor_id && event_time ) {
|
|
stream.setStreamStart(monitor_id, event_time);
|
|
} else {
|
|
Debug(3, "Setting stream start to frame (%d)", frame_id);
|
|
stream.setStreamStart(event_id, frame_id);
|
|
}
|
|
if ( mode == ZMS_JPEG ) {
|
|
stream.setStreamType(EventStream::STREAM_JPEG);
|
|
} else {
|
|
stream.setStreamFormat(format);
|
|
stream.setStreamBitrate(bitrate);
|
|
stream.setStreamType(EventStream::STREAM_MPEG);
|
|
} // end if jpeg or mpeg
|
|
stream.runStream();
|
|
} else {
|
|
Error("Neither a monitor or event was specified.");
|
|
} // end if monitor or event
|
|
|
|
Debug(1, "Terminating");
|
|
Image::Deinitialise();
|
|
logTerm();
|
|
dbQueue.stop();
|
|
zmDbClose();
|
|
|
|
return 0;
|
|
}
|