Fix horrid memory corruptions.
[dcpomatic.git] / src / lib / util.cc
index 1bd8f43180169571a5a95fc7b27fe48b523c22c0..c2b24944d79ada39b4f1f844cd3f0a1537c1b486 100644 (file)
 #ifdef DVDOMATIC_POSIX
 #include <execinfo.h>
 #include <cxxabi.h>
-#include <sys/types.h> 
-#include <sys/socket.h>
 #endif
 #include <libssh/libssh.h>
 #include <signal.h>
 #include <boost/algorithm/string.hpp>
+#include <boost/bind.hpp>
+#include <boost/lambda/lambda.hpp>
+#include <boost/lexical_cast.hpp>
 #include <openjpeg.h>
 #include <openssl/md5.h>
 #include <magick/MagickCore.h>
@@ -56,14 +57,11 @@ extern "C" {
 #include "filter.h"
 #include "screen.h"
 #include "film_state.h"
+#include "sound_processor.h"
 #ifndef DVDOMATIC_DISABLE_PLAYER
 #include "player_manager.h"
 #endif
 
-#ifdef DEBUG_HASH
-#include <mhash.h>
-#endif
-
 using namespace std;
 using namespace boost;
 
@@ -287,12 +285,246 @@ seconds (struct timeval t)
        return t.tv_sec + (double (t.tv_usec) / 1e6);
 }
 
-/** @param socket Socket to read from */
-SocketReader::SocketReader (shared_ptr<asio::ip::tcp::socket> socket)
-       : _socket (socket)
+
+#ifdef DVDOMATIC_POSIX
+void
+sigchld_handler (int, siginfo_t* info, void *)
+{
+#ifndef DVDOMATIC_DISABLE_PLAYER       
+       PlayerManager::instance()->child_exited (info->si_pid);
+#endif 
+}
+#endif
+
+/** Call the required functions to set up DVD-o-matic's static arrays, etc. */
+void
+dvdomatic_setup ()
+{
+       Format::setup_formats ();
+       DCPContentType::setup_dcp_content_types ();
+       Scaler::setup_scalers ();
+       Filter::setup_filters ();
+       SoundProcessor::setup_sound_processors ();
+
+#ifdef DVDOMATIC_POSIX 
+       struct sigaction sa;
+       sa.sa_flags = SA_SIGINFO;
+       sigemptyset (&sa.sa_mask);
+       sa.sa_sigaction = sigchld_handler;
+       sigaction (SIGCHLD, &sa, 0);
+#endif 
+}
+
+string
+crop_string (Position start, Size size)
+{
+       stringstream s;
+       s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
+       return s.str ();
+}
+
+vector<string>
+split_at_spaces_considering_quotes (string s)
+{
+       vector<string> out;
+       bool in_quotes = false;
+       string c;
+       for (string::size_type i = 0; i < s.length(); ++i) {
+               if (s[i] == ' ' && !in_quotes) {
+                       out.push_back (c);
+                       c = "";
+               } else if (s[i] == '"') {
+                       in_quotes = !in_quotes;
+               } else {
+                       c += s[i];
+               }
+       }
+
+       out.push_back (c);
+       return out;
+}
+
+string
+md5_digest (void const * data, int size)
+{
+       MD5_CTX md5_context;
+       MD5_Init (&md5_context);
+       MD5_Update (&md5_context, data, size);
+       unsigned char digest[MD5_DIGEST_LENGTH];
+       MD5_Final (digest, &md5_context);
+       
+       stringstream s;
+       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
+               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
+       }
+
+       return s.str ();
+}
+
+/** @param file File name.
+ *  @return MD5 digest of file's contents.
+ */
+string
+md5_digest (string file)
+{
+       ifstream f (file.c_str(), ios::binary);
+       if (!f.good ()) {
+               throw OpenFileError (file);
+       }
+       
+       f.seekg (0, ios::end);
+       int bytes = f.tellg ();
+       f.seekg (0, ios::beg);
+
+       int const buffer_size = 64 * 1024;
+       char buffer[buffer_size];
+
+       MD5_CTX md5_context;
+       MD5_Init (&md5_context);
+       while (bytes > 0) {
+               int const t = min (bytes, buffer_size);
+               f.read (buffer, t);
+               MD5_Update (&md5_context, buffer, t);
+               bytes -= t;
+       }
+
+       unsigned char digest[MD5_DIGEST_LENGTH];
+       MD5_Final (digest, &md5_context);
+
+       stringstream s;
+       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
+               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
+       }
+
+       return s.str ();
+}
+
+/** @param An arbitrary sampling rate.
+ *  @return The appropriate DCP-approved sampling rate (48kHz or 96kHz).
+ */
+int
+dcp_audio_sample_rate (int fs)
+{
+       if (fs <= 48000) {
+               return 48000;
+       }
+
+       return 96000;
+}
+
+bool operator== (Crop const & a, Crop const & b)
+{
+       return (a.left == b.left && a.right == b.right && a.top == b.top && a.bottom == b.bottom);
+}
+
+bool operator!= (Crop const & a, Crop const & b)
+{
+       return !(a == b);
+}
+
+/** @param index Colour LUT index.
+ *  @return Human-readable name.
+ */
+string
+colour_lut_index_to_name (int index)
+{
+       switch (index) {
+       case 0:
+               return "sRGB";
+       case 1:
+               return "Rec 709";
+       }
+
+       assert (false);
+       return "";
+}
+
+Socket::Socket ()
+       : _deadline (_io_service)
+       , _socket (_io_service)
        , _buffer_data (0)
 {
+       _deadline.expires_at (posix_time::pos_infin);
+       check ();
+}
+
+void
+Socket::check ()
+{
+       if (_deadline.expires_at() <= asio::deadline_timer::traits_type::now ()) {
+               _socket.close ();
+               _deadline.expires_at (posix_time::pos_infin);
+       }
 
+       _deadline.async_wait (boost::bind (&Socket::check, this));
+}
+
+/** Blocking connect with timeout.
+ *  @param endpoint End-point to connect to.
+ *  @param timeout Time-out in seconds.
+ */
+void
+Socket::connect (asio::ip::basic_resolver_entry<asio::ip::tcp> const & endpoint, int timeout)
+{
+       system::error_code ec = asio::error::would_block;
+       _socket.async_connect (endpoint, lambda::var(ec) = lambda::_1);
+       do {
+               _io_service.run_one();
+       } while (ec == asio::error::would_block);
+
+       if (ec || !_socket.is_open ()) {
+               throw NetworkError ("connect timed out");
+       }
+}
+
+/** Blocking write with timeout.
+ *  @param data Buffer to write.
+ *  @param size Number of bytes to write.
+ *  @param timeout Time-out, in seconds.
+ */
+void
+Socket::write (uint8_t const * data, int size, int timeout)
+{
+       _deadline.expires_from_now (posix_time::seconds (timeout));
+       system::error_code ec = asio::error::would_block;
+
+       asio::async_write (_socket, asio::buffer (data, size), lambda::var(ec) = lambda::_1);
+       do {
+               _io_service.run_one ();
+       } while (ec == asio::error::would_block);
+
+       if (ec) {
+               throw NetworkError ("write timed out");
+       }
+}
+
+/** Blocking read with timeout.
+ *  @param data Buffer to read to.
+ *  @param size Number of bytes to read.
+ *  @param timeout Time-out, in seconds.
+ */
+int
+Socket::read (uint8_t* data, int size, int timeout)
+{
+       _deadline.expires_from_now (posix_time::seconds (timeout));
+       system::error_code ec = asio::error::would_block;
+
+       int amount_read = 0;
+
+       _socket.async_read_some (
+               asio::buffer (data, size),
+               (lambda::var(ec) = lambda::_1, lambda::var(amount_read) = lambda::_2)
+               );
+
+       do {
+               _io_service.run_one ();
+       } while (ec == asio::error::would_block);
+       
+       if (ec) {
+               amount_read = 0;
+       }
+
+       return amount_read;
 }
 
 /** Mark some data as being `consumed', so that it will not be returned
@@ -300,7 +532,7 @@ SocketReader::SocketReader (shared_ptr<asio::ip::tcp::socket> socket)
  *  @param size Amount of data to consume, in bytes.
  */
 void
-SocketReader::consume (int size)
+Socket::consume (int size)
 {
        assert (_buffer_data >= size);
        
@@ -317,7 +549,7 @@ SocketReader::consume (int size)
  *  @param size Number of bytes to read.
  */
 void
-SocketReader::read_definite_and_consume (uint8_t* data, int size)
+Socket::read_definite_and_consume (uint8_t* data, int size, int timeout)
 {
        int const from_buffer = min (_buffer_data, size);
        if (from_buffer > 0) {
@@ -331,7 +563,7 @@ SocketReader::read_definite_and_consume (uint8_t* data, int size)
 
        /* read() the rest */
        while (size > 0) {
-               int const n = asio::read (*_socket, asio::buffer (data, size));
+               int const n = read (data, size, timeout);
                if (n <= 0) {
                        throw NetworkError ("could not read");
                }
@@ -346,7 +578,7 @@ SocketReader::read_definite_and_consume (uint8_t* data, int size)
  *  @param size Maximum amount of data to read.
  */
 void
-SocketReader::read_indefinite (uint8_t* data, int size)
+Socket::read_indefinite (uint8_t* data, int size, int timeout)
 {
        assert (size < int (sizeof (_buffer)));
 
@@ -354,7 +586,7 @@ SocketReader::read_indefinite (uint8_t* data, int size)
        int to_read = size - _buffer_data;
        while (to_read > 0) {
                /* read as much of it as we can (into our buffer) */
-               int const n = asio::read (*_socket, asio::buffer (_buffer + _buffer_data, to_read));
+               int const n = read (_buffer + _buffer_data, to_read, timeout);
                if (n <= 0) {
                        throw NetworkError ("could not read");
                }
@@ -370,126 +602,105 @@ SocketReader::read_indefinite (uint8_t* data, int size)
        memcpy (data, _buffer, size);
 }
 
-#ifdef DVDOMATIC_POSIX
-void
-sigchld_handler (int, siginfo_t* info, void *)
+Rectangle
+Rectangle::intersection (Rectangle const & other) const
 {
-#ifndef DVDOMATIC_DISABLE_PLAYER       
-       PlayerManager::instance()->child_exited (info->si_pid);
-#endif 
+       int const tx = max (x, other.x);
+       int const ty = max (y, other.y);
+       
+       return Rectangle (
+               tx, ty,
+               min (x + width, other.x + other.width) - tx,
+               min (y + height, other.y + other.height) - ty
+               );
 }
-#endif
 
-/** Call the required functions to set up DVD-o-matic's static arrays, etc. */
-void
-dvdomatic_setup ()
+int
+round_up (int a, int t)
 {
-       Format::setup_formats ();
-       DCPContentType::setup_dcp_content_types ();
-       Scaler::setup_scalers ();
-       Filter::setup_filters ();
-
-#ifdef DVDOMATIC_POSIX 
-       struct sigaction sa;
-       sa.sa_flags = SA_SIGINFO;
-       sigemptyset (&sa.sa_mask);
-       sa.sa_sigaction = sigchld_handler;
-       sigaction (SIGCHLD, &sa, 0);
-#endif 
+       a += (t - 1);
+       return a - (a % t);
 }
 
-string
-crop_string (Position start, Size size)
+multimap<string, string>
+read_key_value (istream &s) 
 {
-       stringstream s;
-       s << "crop=" << size.width << ":" << size.height << ":" << start.x << ":" << start.y;
-       return s.str ();
-}
+       multimap<string, string> kv;
+       
+       string line;
+       while (getline (s, line)) {
+               if (line.empty ()) {
+                       continue;
+               }
+               
+               if (line[0] == '#') {
+                       continue;
+               }
 
-vector<string>
-split_at_spaces_considering_quotes (string s)
-{
-       vector<string> out;
-       bool in_quotes = false;
-       string c;
-       for (string::size_type i = 0; i < s.length(); ++i) {
-               if (s[i] == ' ' && !in_quotes) {
-                       out.push_back (c);
-                       c = "";
-               } else if (s[i] == '"') {
-                       in_quotes = !in_quotes;
-               } else {
-                       c += s[i];
+               if (line[line.size() - 1] == '\r') {
+                       line = line.substr (0, line.size() - 1);
+               }
+
+               size_t const s = line.find (' ');
+               if (s == string::npos) {
+                       continue;
                }
+
+               kv.insert (make_pair (line.substr (0, s), line.substr (s + 1)));
        }
 
-       out.push_back (c);
-       return out;
+       return kv;
 }
 
-#ifdef DEBUG_HASH
-void
-md5_data (string title, void const * data, int size)
+string
+get_required_string (multimap<string, string> const & kv, string k)
 {
-       MHASH ht = mhash_init (MHASH_MD5);
-       if (ht == MHASH_FAILED) {
-               throw EncodeError ("could not create hash thread");
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
 
-       mhash (ht, data, size);
+       multimap<string, string>::const_iterator i = kv.find (k);
        
-       uint8_t hash[16];
-       mhash_deinit (ht, hash);
-       
-       printf ("%s [%d]: ", title.c_str (), size);
-       for (int i = 0; i < int (mhash_get_block_size (MHASH_MD5)); ++i) {
-               printf ("%.2x", hash[i]);
+       if (i == kv.end ()) {
+               throw StringError (String::compose ("missing key %1 in key-value set", k));
        }
-       printf ("\n");
+
+       return i->second;
 }
-#endif
 
-string
-md5_digest (string file)
+int
+get_required_int (multimap<string, string> const & kv, string k)
 {
-       ifstream f (file.c_str(), ios::binary);
-       if (!f.good ()) {
-               throw OpenFileError (file);
-       }
-       
-       f.seekg (0, ios::end);
-       int bytes = f.tellg ();
-       f.seekg (0, ios::beg);
-
-       int const buffer_size = 64 * 1024;
-       char buffer[buffer_size];
+       string const v = get_required_string (kv, k);
+       return lexical_cast<int> (v);
+}
 
-       MD5_CTX md5_context;
-       MD5_Init (&md5_context);
-       while (bytes > 0) {
-               int const t = min (bytes, buffer_size);
-               f.read (buffer, t);
-               MD5_Update (&md5_context, buffer, t);
-               bytes -= t;
+string
+get_optional_string (multimap<string, string> const & kv, string k)
+{
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
 
-       unsigned char digest[MD5_DIGEST_LENGTH];
-       MD5_Final (digest, &md5_context);
-
-       stringstream s;
-       for (int i = 0; i < MD5_DIGEST_LENGTH; ++i) {
-               s << hex << setfill('0') << setw(2) << ((int) digest[i]);
+       multimap<string, string>::const_iterator i = kv.find (k);
+       if (i == kv.end ()) {
+               return "";
        }
 
-       return s.str ();
+       return i->second;
 }
 
 int
-dcp_audio_sample_rate (int fs)
+get_optional_int (multimap<string, string> const & kv, string k)
 {
-       if (fs <= 48000) {
-               return 48000;
+       if (kv.count (k) > 1) {
+               throw StringError ("unexpected multiple keys in key-value set");
        }
 
-       return 96000;
+       multimap<string, string>::const_iterator i = kv.find (k);
+       if (i == kv.end ()) {
+               return 0;
+       }
+
+       return lexical_cast<int> (i->second);
 }