|
|
@@ -0,0 +1,249 @@
|
|
|
+<?php
|
|
|
+
|
|
|
+# Test params
|
|
|
+$_SERVER['PHP_AUTH_USER'] = 'profplump';
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Parameters/Globals
|
|
|
+#=========================================
|
|
|
+$USER_DB = 'sqlite:/var/state/myMyPlex/db.sqlite';
|
|
|
+$DBH = NULL;
|
|
|
+
|
|
|
+# Prevent PHP from whining - the actual timezone you set is irrelevent
|
|
|
+date_default_timezone_set('UTC');
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Error Handler
|
|
|
+#=========================================
|
|
|
+function error($errno, $errstr, $errfile = '', $errline = '') {
|
|
|
+ header('HTTP/1.1 500 myMyPlex Processing Error');
|
|
|
+ $msg = 'myMyPlex Processing Error';
|
|
|
+ if ($errno) {
|
|
|
+ $msg .= ' ' . $errno;
|
|
|
+ }
|
|
|
+ $msg .= ' (' . $errstr . ')';
|
|
|
+ if ($errfile) {
|
|
|
+ $msg .= ' in ' . $errfile;
|
|
|
+ }
|
|
|
+ if ($errline) {
|
|
|
+ $msg .= ' at ' . $errline;
|
|
|
+ }
|
|
|
+ die($msg . "\n");
|
|
|
+}
|
|
|
+set_error_handler('error');
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# DB Conneciton/Table Init
|
|
|
+#=========================================
|
|
|
+function get_dbh() {
|
|
|
+ global $USER_DB;
|
|
|
+ global $DBH;
|
|
|
+
|
|
|
+ # If we haven't already connected in this session
|
|
|
+ if (! $DBH) {
|
|
|
+ # Connect
|
|
|
+ $dbh = new PDO($USER_DB);
|
|
|
+ if (! $dbh) {
|
|
|
+ error(0, 'Unable to connect to DB');
|
|
|
+ }
|
|
|
+ $DBH = $dbh;
|
|
|
+
|
|
|
+ # Create the user table if needed
|
|
|
+ $dbh->query('
|
|
|
+ CREATE TABLE IF NOT EXISTS users
|
|
|
+ (username VARCHAR(64), auth CHAR(32),
|
|
|
+ PRIMARY KEY(username), UNIQUE(auth))
|
|
|
+ ');
|
|
|
+
|
|
|
+ # Create the server table if needed
|
|
|
+ $dbh->query('
|
|
|
+ CREATE TABLE IF NOT EXISTS servers
|
|
|
+ (username VARCHAR(64), name VARCHAR(64),
|
|
|
+ address CHAR(15), port INT, id CHAR(40),
|
|
|
+ version VARCHAR(32),
|
|
|
+ PRIMARY KEY(username, name), UNIQUE(id))
|
|
|
+ ');
|
|
|
+ }
|
|
|
+
|
|
|
+ # Return either a new or existing DB ref
|
|
|
+ return $DBH;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Get User
|
|
|
+#=========================================
|
|
|
+function get_user($username = '', $auth = '') {
|
|
|
+ $user = array(
|
|
|
+ 'username' => '',
|
|
|
+ 'auth' => ''
|
|
|
+ );
|
|
|
+
|
|
|
+ try {
|
|
|
+ # Query for the requested user
|
|
|
+ $dbh = get_dbh();
|
|
|
+ if (strlen($username)) {
|
|
|
+ $stmt = $dbh->prepare('SELECT username, auth FROM users WHERE username = ?');
|
|
|
+ $stmt->execute(array($username));
|
|
|
+ } else {
|
|
|
+ $stmt = $dbh->prepare('SELECT username, auth FROM users WHERE auth = ?');
|
|
|
+ $stmt->execute(array($auth));
|
|
|
+ }
|
|
|
+ $row = $stmt->fetch(PDO::FETCH_ASSOC);
|
|
|
+
|
|
|
+ # If the users exists, map it into our local array
|
|
|
+ if ($row) {
|
|
|
+ foreach (array_keys($user) as $key) {
|
|
|
+ $user[$key] = $row[$key];
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in get_user: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return $user;
|
|
|
+}
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Save User
|
|
|
+#=========================================
|
|
|
+function save_user($user) {
|
|
|
+ try {
|
|
|
+ # Get connected
|
|
|
+ $dbh = get_dbh();
|
|
|
+
|
|
|
+ # Create the user if necessary
|
|
|
+ $stmt = $dbh->prepare('SELECT username FROM users WHERE username = ?');
|
|
|
+ $stmt->execute(array($user['username']));
|
|
|
+ if (! is_array($stmt->fetch())) {
|
|
|
+ $stmt = $dbh->prepare('INSERT INTO users (username) VALUES (?)');
|
|
|
+ $stmt->execute(array($user['username']));
|
|
|
+ }
|
|
|
+
|
|
|
+ # Update the DB from our local array
|
|
|
+ $stmt = $dbh->prepare('UPDATE users SET auth = ? WHERE username = ?');
|
|
|
+ $stmt->execute(array($user['auth'], $user['username']));
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in save_user: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return TRUE;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Get Server
|
|
|
+#=========================================
|
|
|
+function get_server($username, $name) {
|
|
|
+ $server = array(
|
|
|
+ 'username' => '',
|
|
|
+ 'name' => '',
|
|
|
+ 'address' => '',
|
|
|
+ 'port' => '',
|
|
|
+ 'id' => '',
|
|
|
+ 'version' => '',
|
|
|
+ );
|
|
|
+
|
|
|
+ try {
|
|
|
+ # Query for the requested user
|
|
|
+ $dbh = get_dbh();
|
|
|
+ $stmt = $dbh->prepare('
|
|
|
+ SELECT username, name, address, port, id, version
|
|
|
+ FROM servers WHERE username = ? AND name = ?
|
|
|
+ ');
|
|
|
+ $stmt->execute(array($username, $name));
|
|
|
+ $row = $stmt->fetch(PDO::FETCH_ASSOC);
|
|
|
+
|
|
|
+ # If the users exists, map it into our local array
|
|
|
+ if ($row) {
|
|
|
+ foreach (array_keys($server) as $key) {
|
|
|
+ $server[$key] = $row[$key];
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in get_server: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return $server;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Save Server
|
|
|
+#=========================================
|
|
|
+function save_server($server) {
|
|
|
+ try {
|
|
|
+ # Get connected
|
|
|
+ $dbh = get_dbh();
|
|
|
+
|
|
|
+ # Create the server if necessary
|
|
|
+ $stmt = $dbh->prepare('SELECT name FROM servers WHERE username = ? AND name = ?');
|
|
|
+ $stmt->execute(array($server['username'], $server['name']));
|
|
|
+ if (! is_array($stmt->fetch())) {
|
|
|
+ $stmt = $dbh->prepare('INSERT INTO servers (username, name, id) VALUES (?, ?, ?)');
|
|
|
+ $stmt->execute(array($server['username'], $server['name']));
|
|
|
+ }
|
|
|
+
|
|
|
+ # Update the DB from our local array
|
|
|
+ $stmt = $dbh->prepare('UPDATE servers SET address = ?, port = ?, id = ? WHERE username = ? AND name = ?');
|
|
|
+ $stmt->execute(array($server['address'], $server['port'], $server['id'], $server['username'], $server['name']));
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in save_server: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return TRUE;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Delete Server
|
|
|
+#=========================================
|
|
|
+function delete_server($id) {
|
|
|
+ try {
|
|
|
+ $dbh = get_dbh();
|
|
|
+ $stmt = $dbh->prepare('DELETE FROM servers WHERE id = ?');
|
|
|
+ $stmt->execute(array($id));
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in save_server: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return TRUE;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Get Server List
|
|
|
+#=========================================
|
|
|
+function get_server_list($username) {
|
|
|
+ $servers = array();
|
|
|
+
|
|
|
+ try {
|
|
|
+ # Query for the requested user
|
|
|
+ $dbh = get_dbh();
|
|
|
+ $stmt = $dbh->prepare('SELECT name FROM servers WHERE username = ?');
|
|
|
+ $stmt->execute(array($username));
|
|
|
+
|
|
|
+ $rs = $stmt->fetchAll(PDO::FETCH_ASSOC);
|
|
|
+ foreach ($rs as $row) {
|
|
|
+ $servers[] = $row['name'];
|
|
|
+ }
|
|
|
+ } catch (PDOException $e) {
|
|
|
+ error(0, 'DB error in get_server_list: ' . $e->getMessage());
|
|
|
+ }
|
|
|
+
|
|
|
+ return $servers;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+#=========================================
|
|
|
+# Generate Auth Tokens
|
|
|
+#=========================================
|
|
|
+function generate_auth() {
|
|
|
+ return sprintf('%08x%08x%08x%08x',
|
|
|
+ mt_rand(0, 0xffffffff), mt_rand(0, 0xffffffff),
|
|
|
+ mt_rand(0, 0xffffffff), mt_rand(0, 0xffffffff));
|
|
|
+}
|
|
|
+
|
|
|
+?>
|
Zachary Kotlarek
14 лет назад