# -*- python -*- ''' MIT/X Consortium License Copyright © 2015 Mattias Andrée 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. ''' cimport cython from libc.stdlib cimport malloc, free from posix.types cimport uid_t, gid_t, mode_t cdef extern int bus_create(const char *, int, char **) ''' Create a new bus @param file The pathname of the bus, `NULL` to create a random one @param flags `BUS_EXCL` (if `file` is not `NULL`) to fail if the file already exists, otherwise if the file exists, nothing will happen; `BUS_INTR` to fail if interrupted @param out_file Output parameter for the pathname of the bus @return 0 on success, -1 on error ''' cdef extern int bus_unlink(const char *) ''' Remove a bus @param file The pathname of the bus @return 0 on success, -1 on error ''' cdef extern int bus_open(long, const char *, int) ''' Open an existing bus @param bus Bus information to fill @param file The filename of the bus @param flags `BUS_RDONLY`, `BUS_WRONLY` or `BUS_RDWR`, the value must not be negative @return 0 on success, -1 on error ''' cdef extern int bus_close(long) ''' Close a bus @param bus Bus information @return 0 on success, -1 on error ''' cdef extern int bus_write(long, const char *, int) ''' Broadcast a message a bus @param bus Bus information @param message The message to write, may not be longer than `BUS_MEMORY_SIZE` including the NUL-termination @param flags `BUS_NOWAIT` fail if other process is attempting to write @return 0 on success, -1 on error ''' cdef extern int bus_read(long, int (*)(const char *, void *), void *) ''' Listen (in a loop, forever) for new message on a bus @param bus Bus information @param callback Function to call when a message is received, the input parameters will be the read message and `user_data` from `bus_read`'s parameter with the same name. The message must have been parsed or copied when `callback` returns as it may be over overridden after that time. `callback` should return either of the the values: 0: stop listening 1: continue listening -1: an error has occurred @return 0 on success, -1 on error ''' cdef extern int bus_poll_start(long, int) ''' Announce that the thread is listening on the bus. This is required so the will does not miss any messages due to race conditions. Additionally, not calling this function will cause the bus the misbehave, is `bus_poll` is written to expect this function to have been called. @param bus Bus information @param flags `BUS_NOWAIT` if the bus should fail and set `errno` to `EAGAIN` if there isn't already a message available on the bus when `bus_poll` is called @return 0 on success, -1 on error ''' cdef extern int bus_poll_stop(long) ''' Announce that the thread has stopped listening on the bus. This is required so that the thread does not cause others to wait indefinitely. @param bus Bus information @return 0 on success, -1 on error ''' cdef extern const char *bus_poll(long) ''' Wait for a message to be broadcasted on the bus. The caller should make a copy of the received message, without freeing the original copy, and parse it in a separate thread. When the new thread has started be started, the caller of this function should then either call `bus_poll` again or `bus_poll_stop`. @param bus Bus information @return The received message, `NULL` on error ''' cdef extern int bus_chown(const char *, uid_t, gid_t) ''' Change the ownership of a bus `stat(2)` can be used of the bus's associated file to get the bus's ownership @param file The pathname of the bus @param owner The user ID of the bus's new owner @param group The group ID of the bus's new group @return 0 on success, -1 on error ''' cdef extern int bus_chmod(const char *, mode_t) ''' Change the permissions for a bus `stat(2)` can be used of the bus's associated file to get the bus's permissions @param file The pathname of the bus @param mode The permissions of the bus, any permission for a user implies full permissions for that user, except only the owner may edit the bus's associated file @return 0 on success, -1 on error ''' def bus_allocate() -> int: ''' Allocate memory for a bus @return The address of the allocated memory ''' n = 2 * sizeof(long long) + sizeof(int) + sizeof(char *) return malloc(n) def bus_deallocate(address : int): ''' Deallocate memory for a bus @param address The address of the allocated memory ''' free(address) def bus_create_wrapped(file : str, flags : int) -> str: ''' Create a new bus @param file The pathname of the bus, `None` to create a random one @param flags `BUS_EXCL` (if `file` is not `None`) to fail if the file already exists, otherwise if the file exists, nothing will happen; `BUS_INTR` to fail if interrupted @return The pathname of the bus, `None` on error; `file` is returned unless `file` is `None` ''' cdef const char* cfile cdef char* ofile cdef bytes bs if file is not None: bs = file.encode('utf-8') + bytes([0]) cfile = bs r = bus_create(cfile, flags, NULL) return file if r == 0 else None r = bus_create(NULL, flags, &ofile) if r == 0: bs = ofile return bs.decode('utf-8', 'strict') return None def bus_unlink_wrapped(file : str) -> int: ''' Remove a bus @param file The pathname of the bus @return 0 on success, -1 on error ''' cdef const char* cfile cdef bytes bs bs = file.encode('utf-8') + bytes([0]) cfile = bs return bus_unlink(cfile) def bus_open_wrapped(bus : int, file : str, flags : int) -> int: ''' Open an existing bus @param bus Bus information to fill @param file The filename of the bus @param flags `BUS_RDONLY`, `BUS_WRONLY` or `BUS_RDWR`, the value must not be negative @return 0 on success, -1 on error ''' cdef const char* cfile cdef bytes bs bs = file.encode('utf-8') + bytes([0]) cfile = bs return bus_open(bus, cfile, flags) def bus_close_wrapped(bus : int) -> int: ''' Close a bus @param bus Bus information @return 0 on success, -1 on error ''' return bus_close(bus) def bus_write_wrapped(bus : int, message : str, flags : int) -> int: ''' Broadcast a message a bus @param bus Bus information @param message The message to write, may not be longer than `BUS_MEMORY_SIZE` including the NUL-termination @param flags `BUS_NOWAIT` fail with errno set to `os.errno.EAGAIN` if other process is attempting to write @return 0 on success, -1 on error ''' cdef const char* cmessage cdef bytes bs bs = message.encode('utf-8') + bytes([0]) cmessage = bs return bus_write(bus, cmessage, flags) cdef int bus_callback_wrapper(const char *message, user_data): cdef bytes bs callback, user_data = tuple(user_data) if message is NULL: return callback(None, user_data) else: bs = message return callback(bs, user_data) def bus_read_wrapped(bus : int, callback : callable, user_data) -> int: ''' Listen (in a loop, forever) for new message on a bus @param bus Bus information @param callback Function to call when a message is received, the input parameters will be the read message and `user_data` from `bus_read`'s parameter with the same name. The message must have been parsed or copied when `callback` returns as it may be over overridden after that time. `callback` should return either of the the values: 0: stop listening 1: continue listening -1: an error has occurred @return 0 on success, -1 on error ''' user = (callback, user_data) return bus_read(bus, &bus_callback_wrapper, user) def bus_poll_start_wrapped(bus : int, flags : int) -> int: ''' Announce that the thread is listening on the bus. This is required so the will does not miss any messages due to race conditions. Additionally, not calling this function will cause the bus the misbehave, is `bus_poll_wrapped` is written to expect this function to have been called. @param bus Bus information @param flags `BUS_NOWAIT` if the bus should fail and set `errno` to `os.errno.EAGAIN` if there isn't already a message available on the bus when `bus_poll` is called @return 0 on success, -1 on error ''' return bus_poll_start(bus, flags) def bus_poll_stop_wrapped(bus : int) -> int: ''' Announce that the thread has stopped listening on the bus. This is required so that the thread does not cause others to wait indefinitely. @param bus Bus information @return 0 on success, -1 on error ''' return bus_poll_stop(bus) def bus_poll_wrapped(bus : int) -> bytes: ''' Wait for a message to be broadcasted on the bus. The caller should make a copy of the received message, without freeing the original copy, and parse it in a separate thread. When the new thread has started be started, the caller of this function should then either call `bus_poll_wrapped` again or `bus_poll_stop_wrapped`. @param bus Bus information @return The received message, `None` on error ''' cdef const char* msg cdef bytes bs msg = bus_poll(bus) if msg is NULL: return None bs = msg return bs def bus_chown_wrapped(file : str, owner : int, group : int) -> int: ''' Change the ownership of a bus `os.stat` can be used of the bus's associated file to get the bus's ownership @param file The pathname of the bus @param owner The user ID of the bus's new owner @param group The group ID of the bus's new group @return 0 on success, -1 on error ''' cdef const char* cfile cdef bytes bs bs = file.encode('utf-8') + bytes([0]) cfile = bs return bus_chown(cfile, owner, group) def bus_chmod_wrapped(file : str, mode : int) -> int: ''' Change the permissions for a bus `os.stat` can be used of the bus's associated file to get the bus's permissions @param file The pathname of the bus @param mode The permissions of the bus, any permission for a user implies full permissions for that user, except only the owner may edit the bus's associated file @return 0 on success, -1 on error ''' cdef const char* cfile cdef bytes bs bs = file.encode('utf-8') + bytes([0]) cfile = bs return bus_chmod(cfile, mode)