Move the RSA key to the beginning of file in python bindings
[deb_shairplay.git] / src / bindings / python / Shairplay.py
index f9dc919bf0881d89395bb48d3e64a006c61f81d9..4400bd9003f2251e934c06fabba0ca00ad10b05b 100644 (file)
@@ -1,16 +1,25 @@
 # coding=utf-8
 '''
-       Copyright (C) 2012  Juho Vähä-Herttua
-
-       This library is free software; you can redistribute it and/or
-       modify it under the terms of the GNU Lesser General Public
-       License as published by the Free Software Foundation; either
-       version 2.1 of the License, or (at your option) any later version.
-
-       This library 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
-       Lesser General Public License for more details.
+Copyright (C) 2012  Juho Vähä-Herttua
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be included
+in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 '''
 
 import os
@@ -19,65 +28,6 @@ import platform
 
 from ctypes import *
 
-audio_init_prototype =        CFUNCTYPE(py_object, c_void_p, c_int, c_int, c_int)
-audio_set_volume_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_float)
-audio_process_prototype =     CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
-audio_flush_prototype =       CFUNCTYPE(None, c_void_p, c_void_p)
-audio_destroy_prototype =     CFUNCTYPE(None, c_void_p, c_void_p)
-
-class RaopNativeCallbacks(Structure):
-       _fields_ = [("cls",               py_object),
-                   ("audio_init",        audio_init_prototype),
-                   ("audio_set_volume",  audio_set_volume_prototype),
-                   ("audio_process",     audio_process_prototype),
-                   ("audio_flush",       audio_flush_prototype),
-                   ("audio_destroy",     audio_destroy_prototype)]
-
-def LoadShairplay(path):
-       if sys.maxsize < 2**32:
-               libname = "shairplay32"
-       else:
-               libname = "shairplay64"
-
-       if platform.system() == "Windows":
-               libname = libname + ".dll"
-       elif platform.system() == "Darwin":
-               libname = "lib" + libname + ".dylib"
-       else:
-               libname = "lib" + libname + ".so"
-
-       try:
-               fullpath = os.path.join(path, libname)
-               libshairplay = cdll.LoadLibrary(fullpath)
-       except:
-               raise RuntimeError("Couldn't load shairplay library " + libname)
-
-       # Initialize dnssd related functions
-       libshairplay.dnssd_init.restype = c_void_p
-       libshairplay.dnssd_init.argtypes = [POINTER(c_int)]
-       libshairplay.dnssd_register_raop.restype = c_int
-       libshairplay.dnssd_register_raop.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int]
-       libshairplay.dnssd_register_airplay.restype = c_int
-       libshairplay.dnssd_register_airplay.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int]
-       libshairplay.dnssd_unregister_raop.restype = None
-       libshairplay.dnssd_unregister_raop.argtypes = [c_void_p]
-       libshairplay.dnssd_unregister_airplay.restype = None
-       libshairplay.dnssd_unregister_airplay.argtypes = [c_void_p]
-       libshairplay.dnssd_destroy.restype = None
-       libshairplay.dnssd_destroy.argtypes = [c_void_p]
-
-       # Initialize raop related functions
-       libshairplay.raop_init.restype = c_void_p
-       libshairplay.raop_init.argtypes = [POINTER(RaopNativeCallbacks), c_char_p]
-       libshairplay.raop_start.restype = c_int
-       libshairplay.raop_start.argtypes = [c_void_p, POINTER(c_ushort), POINTER(c_char), c_int]
-       libshairplay.raop_stop.restype = None
-       libshairplay.raop_stop.argtypes = [c_void_p]
-       libshairplay.raop_destroy.restype = None
-       libshairplay.raop_destroy.argtypes = [c_void_p]
-
-       return libshairplay
-
 RSA_KEY = """
 -----BEGIN RSA PRIVATE KEY-----
 MIIEpQIBAAKCAQEA59dE8qLieItsH1WgjrcFRKj6eUWqi+bGLOX1HL3U3GhC/j0Qg90u3sG/1CUt
@@ -104,20 +54,110 @@ LAuE4Pu13aKiJnfft7hIjbK+5kyb3TysZvoyDnb3HOKvInK7vXbKuU4ISgxB2bB3HcYzQMGsz1qJ
 -----END RSA PRIVATE KEY-----
 """
 
+class RaopLogLevel:
+       EMERG   = 0
+       ALERT   = 1
+       CRIT    = 2
+       ERR     = 3
+       WARNING = 4
+       NOTICE  = 5
+       INFO    = 6
+       DEBUG   = 7
+       
+raop_log_callback_prototype =   CFUNCTYPE(None, c_void_p, c_int, c_char_p)
+
+audio_init_prototype =          CFUNCTYPE(py_object, c_void_p, c_int, c_int, c_int)
+audio_process_prototype =       CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
+audio_destroy_prototype =       CFUNCTYPE(None, c_void_p, c_void_p)
+
+audio_flush_prototype =         CFUNCTYPE(None, c_void_p, c_void_p)
+audio_set_volume_prototype =    CFUNCTYPE(None, c_void_p, c_void_p, c_float)
+audio_set_metadata_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
+audio_set_coverart_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
+
+class RaopNativeCallbacks(Structure):
+       _fields_ = [("cls",                 py_object),
+                   ("audio_init",          audio_init_prototype),
+                   ("audio_process",       audio_process_prototype),
+                   ("audio_destroy",       audio_destroy_prototype),
+                   ("audio_flush",         audio_flush_prototype),
+                   ("audio_set_volume",    audio_set_volume_prototype),
+                   ("audio_set_metadata",  audio_set_metadata_prototype),
+                   ("audio_set_coverart",  audio_set_coverart_prototype)]
+
+def InitShairplay(libshairplay):
+       # Initialize dnssd related functions
+       libshairplay.dnssd_init.restype = c_void_p
+       libshairplay.dnssd_init.argtypes = [POINTER(c_int)]
+       libshairplay.dnssd_register_raop.restype = c_int
+       libshairplay.dnssd_register_raop.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int, c_int]
+       libshairplay.dnssd_register_airplay.restype = c_int
+       libshairplay.dnssd_register_airplay.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int]
+       libshairplay.dnssd_unregister_raop.restype = None
+       libshairplay.dnssd_unregister_raop.argtypes = [c_void_p]
+       libshairplay.dnssd_unregister_airplay.restype = None
+       libshairplay.dnssd_unregister_airplay.argtypes = [c_void_p]
+       libshairplay.dnssd_destroy.restype = None
+       libshairplay.dnssd_destroy.argtypes = [c_void_p]
+
+       # Initialize raop related functions
+       libshairplay.raop_init.restype = c_void_p
+       libshairplay.raop_init.argtypes = [c_int, POINTER(RaopNativeCallbacks), c_char_p, POINTER(c_int)]
+       libshairplay.raop_set_log_level.restype = None
+       libshairplay.raop_set_log_level.argtypes = [c_void_p, c_int]
+       libshairplay.raop_set_log_callback.restype = None
+       libshairplay.raop_set_log_callback.argtypes = [c_void_p, raop_log_callback_prototype, c_void_p]
+       libshairplay.raop_is_running.restype = c_int
+       libshairplay.raop_is_running.argtypes = [c_void_p]
+       libshairplay.raop_start.restype = c_int
+       libshairplay.raop_start.argtypes = [c_void_p, POINTER(c_ushort), POINTER(c_char), c_int, c_char_p]
+       libshairplay.raop_stop.restype = None
+       libshairplay.raop_stop.argtypes = [c_void_p]
+       libshairplay.raop_destroy.restype = None
+       libshairplay.raop_destroy.argtypes = [c_void_p]
+
+def LoadShairplay(path):
+       if sys.maxsize < 2**32:
+               libname = "shairplay32"
+       else:
+               libname = "shairplay64"
+
+       if platform.system() == "Windows":
+               libname = libname + ".dll"
+       elif platform.system() == "Darwin":
+               libname = "lib" + libname + ".dylib"
+       else:
+               libname = "lib" + libname + ".so"
+
+       try:
+               fullpath = os.path.join(path, libname)
+               libshairplay = cdll.LoadLibrary(fullpath)
+       except:
+               raise RuntimeError("Couldn't load shairplay library " + libname)
+
+       InitShairplay(libshairplay)
+       return libshairplay
+
 class RaopCallbacks:
        def audio_init(self, bits, channels, samplerate):
                raise NotImplementedError()
 
-       def audio_set_volume(self, session, volume):
-               pass
-
        def audio_process(self, session, buffer):
                raise NotImplementedError()
 
+       def audio_destroy(self, session):
+               raise NotImplementedError()
+
+       def audio_set_volume(self, session, volume):
+               pass
+
        def audio_flush(self, session):
                pass
 
-       def audio_destroy(self, session):
+       def audio_set_metadata(self, session, buffer):
+               pass
+
+       def audio_set_coverart(self, session, buffer):
                pass
 
 class RaopService:
@@ -126,27 +166,37 @@ class RaopService:
                self.sessions.append(session)
                return session
 
-       def audio_set_volume_cb(self, cls, sessionptr, volume):
-               session = cast(sessionptr, py_object).value
-               self.callbacks.audio_set_volume(session, volume)
-
        def audio_process_cb(self, cls, sessionptr, buffer, buflen):
                session = cast(sessionptr, py_object).value
                strbuffer = string_at(buffer, buflen)
                self.callbacks.audio_process(session, strbuffer)
 
-       def audio_flush_cb(self, cls, sessionptr):
-               session = cast(sessionptr, py_object).value
-               self.callbacks.audio_flush(session)
-
        def audio_destroy_cb(self, cls, sessionptr):
                session = cast(sessionptr, py_object).value
                self.callbacks.audio_destroy(session)
                if session in self.sessions:
                        self.sessions.remove(session)
 
+       def audio_flush_cb(self, cls, sessionptr):
+               session = cast(sessionptr, py_object).value
+               self.callbacks.audio_flush(session)
 
-       def __init__(self, libshairplay, callbacks):
+       def audio_set_volume_cb(self, cls, sessionptr, volume):
+               session = cast(sessionptr, py_object).value
+               self.callbacks.audio_set_volume(session, volume)
+
+       def audio_set_metadata_cb(self, cls, sessionptr, buffer, buflen):
+               session = cast(sessionptr, py_object).value
+               strbuffer = string_at(buffer, buflen)
+               self.callbacks.audio_set_metadata(session, strbuffer)
+
+       def audio_set_coverart_cb(self, cls, sessionptr, buffer, buflen):
+               session = cast(sessionptr, py_object).value
+               strbuffer = string_at(buffer, buflen)
+               self.callbacks.audio_set_coverart(session, strbuffer)
+
+
+       def __init__(self, libshairplay, max_clients, callbacks):
                self.libshairplay = libshairplay
                self.callbacks = callbacks
                self.sessions = []
@@ -155,13 +205,15 @@ class RaopService:
                # We need to hold a reference to native_callbacks
                self.native_callbacks = RaopNativeCallbacks()
                self.native_callbacks.audio_init = audio_init_prototype(self.audio_init_cb)
-               self.native_callbacks.audio_set_volume = audio_set_volume_prototype(self.audio_set_volume_cb)
                self.native_callbacks.audio_process = audio_process_prototype(self.audio_process_cb)
-               self.native_callbacks.audio_flush = audio_flush_prototype(self.audio_flush_cb)
                self.native_callbacks.audio_destroy = audio_destroy_prototype(self.audio_destroy_cb)
+               self.native_callbacks.audio_flush = audio_flush_prototype(self.audio_flush_cb)
+               self.native_callbacks.audio_set_volume = audio_set_volume_prototype(self.audio_set_volume_cb)
+               self.native_callbacks.audio_set_metadata = audio_set_metadata_prototype(self.audio_set_metadata_cb)
+               self.native_callbacks.audio_set_coverart = audio_set_coverart_prototype(self.audio_set_coverart_cb)
 
                # Initialize the raop instance with our callbacks
-               self.instance = self.libshairplay.raop_init(pointer(self.native_callbacks), RSA_KEY)
+               self.instance = self.libshairplay.raop_init(max_clients, pointer(self.native_callbacks), RSA_KEY, None)
                if self.instance == None:
                        raise RuntimeError("Initializing library failed")
 
@@ -170,11 +222,30 @@ class RaopService:
                        self.libshairplay.raop_destroy(self.instance)
                self.instance = None
 
-       def start(self, port, hwaddrstr):
+       def is_running(self):
+               if self.libshairplay.raop_is_running(self.instance):
+                       return True
+               else:
+                       return False
+
+       def set_log_level(self, level):
+               self.libshairplay.raop_set_log_level(self.instance, level)
+
+       def set_log_callback(self, log_callback):
+               # Create a new callback function for thread safety
+               def log_callback_cb(cls, level, message):
+                       log_callback(level, message)
+
+               # We need to hold a reference to the log callback instance
+               log_callback_ptr = raop_log_callback_prototype(log_callback_cb)
+               self.libshairplay.raop_set_log_callback(self.instance, log_callback_ptr, None)
+               self.log_callback = log_callback_ptr
+
+       def start(self, port, hwaddrstr, password=None):
                port = c_ushort(port)
                hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
 
-               ret = self.libshairplay.raop_start(self.instance, pointer(port), hwaddr, c_int(len(hwaddr)))
+               ret = self.libshairplay.raop_start(self.instance, pointer(port), hwaddr, c_int(len(hwaddr)), password)
                if ret < 0:
                        raise RuntimeError("Starting RAOP instance failed")
                return port.value
@@ -198,9 +269,12 @@ class DnssdService:
                        self.libshairplay.dnssd_destroy(self.instance)
                self.instance = None
 
-       def register_raop(self, name, port, hwaddrstr):
+       def register_raop(self, name, port, hwaddrstr, password=False):
                hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
-               self.libshairplay.dnssd_register_raop(self.instance, name, c_ushort(port), hwaddr, len(hwaddr))
+               use_pw = c_int(0)
+               if password:
+                       use_pw = c_int(1)
+               self.libshairplay.dnssd_register_raop(self.instance, name, c_ushort(port), hwaddr, len(hwaddr), use_pw)
 
        def unregister_raop(self):
                self.libshairplay.dnssd_unregister_raop(self.instance)