plugins/XBMCRepeat/xbmcclient.py
#!/usr/bin/python
# -*- coding: utf-8 -*-
# Copyright (C) 2008-2013 Team XBMC
#
# 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.
"""
Implementation of XBMC's UDP based input system.
A set of classes that abstract the various packets that the event server
currently supports. In addition, there's also a class, XBMCClient, that
provides functions that sends the various packets. Use XBMCClient if you
don't need complete control over packet structure.
The basic workflow involves:
1. Send a HELO packet
2. Send x number of valid packets
3. Send a BYE packet
IMPORTANT NOTE ABOUT TIMEOUTS:
A client is considered to be timed out if XBMC doesn't received a packet
at least once every 60 seconds. To "ping" XBMC with an empty packet use
PacketPING or XBMCClient.ping(). See the documentation for details.
"""
__author__ = "d4rk@xbmc.org"
__version__ = "0.0.3"
from struct import pack
from socket import *
import time
MAX_PACKET_SIZE = 1024
HEADER_SIZE = 32
MAX_PAYLOAD_SIZE = MAX_PACKET_SIZE - HEADER_SIZE
UNIQUE_IDENTIFICATION = (int)(time.time())
PT_HELO = 0x01
PT_BYE = 0x02
PT_BUTTON = 0x03
PT_MOUSE = 0x04
PT_PING = 0x05
PT_BROADCAST = 0x06
PT_NOTIFICATION = 0x07
PT_BLOB = 0x08
PT_LOG = 0x09
PT_ACTION = 0x0A
PT_DEBUG = 0xFF
ICON_NONE = 0x00
ICON_JPEG = 0x01
ICON_PNG = 0x02
ICON_GIF = 0x03
BT_USE_NAME = 0x01
BT_DOWN = 0x02
BT_UP = 0x04
BT_USE_AMOUNT = 0x08
BT_QUEUE = 0x10
BT_NO_REPEAT = 0x20
BT_VKEY = 0x40
BT_AXIS = 0x80
BT_AXISSINGLE = 0x100
MS_ABSOLUTE = 0x01
LOGDEBUG = 0x00
LOGINFO = 0x01
LOGNOTICE = 0x02
LOGWARNING = 0x03
LOGERROR = 0x04
LOGSEVERE = 0x05
LOGFATAL = 0x06
LOGNONE = 0x07
ACTION_EXECBUILTIN = 0x01
ACTION_BUTTON = 0x02
######################################################################
# Helper Functions
######################################################################
def format_string(msg):
""" """
return msg + "\0"
def format_uint32(num):
""" """
return pack ("!I", num)
def format_uint16(num):
""" """
if num<0:
num = 0
elif num>65535:
num = 65535
return pack ("!H", num)
######################################################################
# Packet Classes
######################################################################
class Packet:
"""Base class that implements a single event packet.
- Generic packet structure (maximum 1024 bytes per packet)
- Header is 32 bytes long, so 992 bytes available for payload
- large payloads can be split into multiple packets using H4 and H5
H5 should contain total no. of packets in such a case
- H6 contains length of P1, which is limited to 992 bytes
- if H5 is 0 or 1, then H4 will be ignored (single packet msg)
- H7 must be set to zeros for now
-----------------------------
| -H1 Signature ("XBMC") | - 4 x CHAR 4B
| -H2 Version (eg. 2.0) | - 2 x UNSIGNED CHAR 2B
| -H3 PacketType | - 1 x UNSIGNED SHORT 2B
| -H4 Sequence number | - 1 x UNSIGNED LONG 4B
| -H5 No. of packets in msg | - 1 x UNSIGNED LONG 4B
| -H7 Client's unique token | - 1 x UNSIGNED LONG 4B
| -H8 Reserved | - 10 x UNSIGNED CHAR 10B
|---------------------------|
| -P1 payload | -
-----------------------------
"""
def __init__(self):
self.sig = "XBMC"
self.minver = 0
self.majver = 2
self.seq = 1
self.maxseq = 1
self.payloadsize = 0
self.uid = UNIQUE_IDENTIFICATION
self.reserved = "\0" * 10
self.payload = ""
return
def append_payload(self, blob):
"""Append to existing payload
Arguments:
blob -- binary data to append to the current payload
"""
self.set_payload(self.payload + blob)
def set_payload(self, payload):
"""Set the payload for this packet
Arguments:
payload -- binary data that contains the payload
"""
self.payload = payload
self.payloadsize = len(self.payload)
self.maxseq = int((self.payloadsize + (MAX_PAYLOAD_SIZE - 1)) / MAX_PAYLOAD_SIZE)
def num_packets(self):
""" Return the number of packets required for payload """
return self.maxseq
def get_header(self, packettype=-1, seq=1, maxseq=1, payload_size=0):
"""Construct a header and return as string
Keyword arguments:
packettype -- valid packet types are PT_HELO, PT_BYE, PT_BUTTON,
PT_MOUSE, PT_PING, PT_BORADCAST, PT_NOTIFICATION,
PT_BLOB, PT_DEBUG
seq -- the sequence of this packet for a multi packet message
(default 1)
maxseq -- the total number of packets for a multi packet message
(default 1)
payload_size -- the size of the payload of this packet (default 0)
"""
if packettype < 0:
packettype = self.packettype
header = self.sig
header += chr(self.majver)
header += chr(self.minver)
header += format_uint16(packettype)
header += format_uint32(seq)
header += format_uint32(maxseq)
header += format_uint16(payload_size)
header += format_uint32(self.uid)
header += self.reserved
return header
def get_payload_size(self, seq):
"""Returns the calculated payload size for the particular packet
Arguments:
seq -- the sequence number
"""
if self.maxseq == 1:
return self.payloadsize
if seq < self.maxseq:
return MAX_PAYLOAD_SIZE
return self.payloadsize % MAX_PAYLOAD_SIZE
def get_udp_message(self, packetnum=1):
"""Construct the UDP message for the specified packetnum and return
as string
Keyword arguments:
packetnum -- the packet no. for which to construct the message
(default 1)
"""
if packetnum > self.num_packets() or packetnum < 1:
return ""
header = ""
if packetnum==1:
header = self.get_header(self.packettype, packetnum, self.maxseq,
self.get_payload_size(packetnum))
else:
header = self.get_header(PT_BLOB, packetnum, self.maxseq,
self.get_payload_size(packetnum))
payload = self.payload[ (packetnum-1) * MAX_PAYLOAD_SIZE :
(packetnum-1) * MAX_PAYLOAD_SIZE+
self.get_payload_size(packetnum) ]
return header + payload
def send(self, sock, addr, uid=UNIQUE_IDENTIFICATION):
"""Send the entire message to the specified socket and address.
Arguments:
sock -- datagram socket object (socket.socket)
addr -- address, port pair (eg: ("127.0.0.1", 9777) )
uid -- unique identification
"""
self.uid = uid
for a in range ( 0, self.num_packets() ):
try:
sock.sendto(self.get_udp_message(a+1), addr)
except:
return False
return True
class PacketHELO (Packet):
"""A HELO packet
A HELO packet establishes a valid connection to XBMC. It is the
first packet that should be sent.
"""
def __init__(self, devicename=None, icon_type=ICON_NONE, icon_file=None):
"""
Keyword arguments:
devicename -- the string that identifies the client
icon_type -- one of ICON_NONE, ICON_JPEG, ICON_PNG, ICON_GIF
icon_file -- location of icon file with respect to current working
directory if icon_type is not ICON_NONE
"""
Packet.__init__(self)
self.packettype = PT_HELO
self.icontype = icon_type
self.set_payload ( format_string(devicename)[0:128] )
self.append_payload( chr (icon_type) )
self.append_payload( format_uint16 (0) ) # port no
self.append_payload( format_uint32 (0) ) # reserved1
self.append_payload( format_uint32 (0) ) # reserved2
if icon_type != ICON_NONE and icon_file:
self.append_payload( file(icon_file).read() )
class PacketNOTIFICATION (Packet):
"""A NOTIFICATION packet
This packet displays a notification window in XBMC. It can contain
a caption, a message and an icon.
"""
def __init__(self, title, message, icon_type=ICON_NONE, icon_file=None):
"""
Keyword arguments:
title -- the notification caption / title
message -- the main text of the notification
icon_type -- one of ICON_NONE, ICON_JPEG, ICON_PNG, ICON_GIF
icon_file -- location of icon file with respect to current working
directory if icon_type is not ICON_NONE
"""
Packet.__init__(self)
self.packettype = PT_NOTIFICATION
self.title = title
self.message = message
self.set_payload ( format_string(title) )
self.append_payload( format_string(message) )
self.append_payload( chr (icon_type) )
self.append_payload( format_uint32 (0) ) # reserved
if icon_type != ICON_NONE and icon_file:
self.append_payload( file(icon_file).read() )
class PacketBUTTON (Packet):
"""A BUTTON packet
A button packet send a key press or release event to XBMC
"""
def __init__(self, code=0, repeat=1, down=1, queue=0,
map_name="", button_name="", amount=0, axis=0):
"""
Keyword arguments:
code -- raw button code (default: 0)
repeat -- this key press should repeat until released (default: 1)
Note that queued pressed cannot repeat.
down -- if this is 1, it implies a press event, 0 implies a release
event. (default: 1)
queue -- a queued key press means that the button event is
executed just once after which the next key press is
processed. It can be used for macros. Currently there
is no support for time delays between queued presses.
(default: 0)
map_name -- a combination of map_name and button_name refers to a
mapping in the user's Keymap.xml or Lircmap.xml.
map_name can be one of the following:
"KB" => standard keyboard map ( <keyboard> section )
"XG" => xbox gamepad map ( <gamepad> section )
"R1" => xbox remote map ( <remote> section )
"R2" => xbox universal remote map ( <universalremote>
section )
"LI:devicename" => LIRC remote map where 'devicename' is the
actual device's name
button_name -- a button name defined in the map specified in map_name.
For example, if map_name is "KB" refering to the
<keyboard> section in Keymap.xml then, valid
button_names include "printscreen", "minus", "x", etc.
amount -- unimplemented for now; in the future it will be used for
specifying magnitude of analog key press events
"""
Packet.__init__(self)
self.flags = 0
self.packettype = PT_BUTTON
if type (code ) == str:
code = ord(code)
# assign code only if we don't have a map and button name
if not (map_name and button_name):
self.code = code
else:
self.flags |= BT_USE_NAME
self.code = 0
if (amount != None):
self.flags |= BT_USE_AMOUNT
self.amount = int(amount)
else:
self.amount = 0
if down:
self.flags |= BT_DOWN
else:
self.flags |= BT_UP
if not repeat:
self.flags |= BT_NO_REPEAT
if queue:
self.flags |= BT_QUEUE
if axis == 1:
self.flags |= BT_AXISSINGLE
elif axis == 2:
self.flags |= BT_AXIS
self.set_payload ( format_uint16(self.code) )
self.append_payload( format_uint16(self.flags) )
self.append_payload( format_uint16(self.amount) )
self.append_payload( format_string (map_name) )
self.append_payload( format_string (button_name) )
class PacketMOUSE (Packet):
"""A MOUSE packet
A MOUSE packets sets the mouse position in XBMC
"""
def __init__(self, x, y):
"""
Arguments:
x -- horitontal position ranging from 0 to 65535
y -- vertical position ranging from 0 to 65535
The range will be mapped to the screen width and height in XBMC
"""
Packet.__init__(self)
self.packettype = PT_MOUSE
self.flags = MS_ABSOLUTE
self.append_payload( chr (self.flags) )
self.append_payload( format_uint16(x) )
self.append_payload( format_uint16(y) )
class PacketBYE (Packet):
"""A BYE packet
A BYE packet terminates the connection to XBMC.
"""
def __init__(self):
Packet.__init__(self)
self.packettype = PT_BYE
class PacketPING (Packet):
"""A PING packet
A PING packet tells XBMC that the client is still alive. All valid
packets act as ping (not just this one). A client needs to ping
XBMC at least once in 60 seconds or it will time out.
"""
def __init__(self):
Packet.__init__(self)
self.packettype = PT_PING
class PacketLOG (Packet):
"""A LOG packet
A LOG packet tells XBMC to log the message to xbmc.log with the loglevel as specified.
"""
def __init__(self, loglevel=0, logmessage="", autoprint=True):
"""
Keyword arguments:
loglevel -- the loglevel, follows XBMC standard.
logmessage -- the message to log
autoprint -- if the logmessage should automaticly be printed to stdout
"""
Packet.__init__(self)
self.packettype = PT_LOG
self.append_payload( chr (loglevel) )
self.append_payload( format_string(logmessage) )
if (autoprint):
print logmessage
class PacketACTION (Packet):
"""An ACTION packet
An ACTION packet tells XBMC to do the action specified, based on the type it knows were it needs to be sent.
The idea is that this will be as in scripting/skining and keymapping, just triggered from afar.
"""
def __init__(self, actionmessage="", actiontype=ACTION_EXECBUILTIN):
"""
Keyword arguments:
loglevel -- the loglevel, follows XBMC standard.
logmessage -- the message to log
autoprint -- if the logmessage should automaticly be printed to stdout
"""
Packet.__init__(self)
self.packettype = PT_ACTION
self.append_payload( chr (actiontype) )
self.append_payload( format_string(actionmessage) )
######################################################################
# XBMC Client Class
######################################################################
class XBMCClient:
"""An XBMC event client"""
def __init__(self, name ="", icon_file=None, broadcast=False, uid=UNIQUE_IDENTIFICATION,
ip="127.0.0.1"):
"""
Keyword arguments:
name -- Name of the client
icon_file -- location of an icon file, if any (png, jpg or gif)
uid -- unique identification
"""
self.name = str(name)
self.icon_file = icon_file
self.icon_type = self._get_icon_type(icon_file)
self.ip = ip
self.port = 9777
self.sock = socket(AF_INET,SOCK_DGRAM)
if broadcast:
self.sock.setsockopt(SOL_SOCKET, SO_BROADCAST, 1)
self.uid = uid
def connect(self, ip=None, port=None):
"""Initialize connection to XBMC
ip -- IP Address of XBMC
port -- port that the event server on XBMC is listening on
"""
if ip:
self.ip = ip
if port:
self.port = int(port)
self.addr = (self.ip, self.port)
packet = PacketHELO(self.name, self.icon_type, self.icon_file)
return packet.send(self.sock, self.addr, self.uid)
def close(self):
"""Close the current connection"""
packet = PacketBYE()
return packet.send(self.sock, self.addr, self.uid)
def ping(self):
"""Send a PING packet"""
packet = PacketPING()
return packet.send(self.sock, self.addr, self.uid)
def send_notification(self, title="", message="", icon_file=None):
"""Send a notification to the connected XBMC
Keyword Arguments:
title -- The title/heading for the notifcation
message -- The message to be displayed
icon_file -- location of an icon file, if any (png, jpg, gif)
"""
self.connect()
packet = PacketNOTIFICATION(title, message,
self._get_icon_type(icon_file),
icon_file)
return packet.send(self.sock, self.addr, self.uid)
def send_keyboard_button(self, button=None):
"""Send a keyboard event to XBMC
Keyword Arguments:
button -- name of the keyboard button to send (same as in Keymap.xml)
"""
if not button:
return
return self.send_button(map="KB", button=button)
def send_remote_button(self, button=None):
"""Send a remote control event to XBMC
Keyword Arguments:
button -- name of the remote control button to send (same as in Keymap.xml)
"""
if not button:
return
return self.send_button(map="R1", button=button)
def release_button(self):
"""Release all buttons"""
packet = PacketBUTTON(code=0x01, down=0)
return packet.send(self.sock, self.addr, self.uid)
def send_button(self, map="", button="", amount=0):
"""Send a button event to XBMC
Keyword arguments:
map -- a combination of map_name and button_name refers to a
mapping in the user's Keymap.xml or Lircmap.xml.
map_name can be one of the following:
"KB" => standard keyboard map ( <keyboard> section )
"XG" => xbox gamepad map ( <gamepad> section )
"R1" => xbox remote map ( <remote> section )
"R2" => xbox universal remote map ( <universalremote>
section )
"LI:devicename" => LIRC remote map where 'devicename' is the
actual device's name
button -- a button name defined in the map specified in map, above.
For example, if map is "KB" refering to the <keyboard>
section in Keymap.xml then, valid buttons include
"printscreen", "minus", "x", etc.
"""
packet = PacketBUTTON(map_name=str(map), button_name=str(button), amount=amount)
return packet.send(self.sock, self.addr, self.uid)
def send_button_state(self, map="", button="", amount=0, down=0, axis=0):
"""Send a button event to XBMC
Keyword arguments:
map -- a combination of map_name and button_name refers to a
mapping in the user's Keymap.xml or Lircmap.xml.
map_name can be one of the following:
"KB" => standard keyboard map ( <keyboard> section )
"XG" => xbox gamepad map ( <gamepad> section )
"R1" => xbox remote map ( <remote> section )
"R2" => xbox universal remote map ( <universalremote>
section )
"LI:devicename" => LIRC remote map where 'devicename' is the
actual device's name
button -- a button name defined in the map specified in map, above.
For example, if map is "KB" refering to the <keyboard>
section in Keymap.xml then, valid buttons include
"printscreen", "minus", "x", etc.
"""
if axis:
if amount == 0:
down = 0
else:
down = 1
packet = PacketBUTTON(map_name=str(map), button_name=str(button), amount=amount, down=down, queue=1, axis=axis)
return packet.send(self.sock, self.addr, self.uid)
def send_mouse_position(self, x=0, y=0):
"""Send a mouse event to XBMC
Keywords Arguments:
x -- absolute x position of mouse ranging from 0 to 65535
which maps to the entire screen width
y -- same a 'x' but relates to the screen height
"""
packet = PacketMOUSE(int(x), int(y))
return packet.send(self.sock, self.addr, self.uid)
def send_log(self, loglevel=0, logmessage="", autoprint=True):
"""
Keyword arguments:
loglevel -- the loglevel, follows XBMC standard.
logmessage -- the message to log
autoprint -- if the logmessage should automaticly be printed to stdout
"""
packet = PacketLOG(loglevel, logmessage)
return packet.send(self.sock, self.addr, self.uid)
def send_action(self, actionmessage="", actiontype=ACTION_EXECBUILTIN):
"""
Keyword arguments:
actionmessage -- the ActionString
actiontype -- The ActionType the ActionString should be sent to.
"""
packet = PacketACTION(actionmessage, actiontype)
return packet.send(self.sock, self.addr, self.uid)
def _get_icon_type(self, icon_file):
if icon_file:
if icon_file.lower()[-3:] == "png":
return ICON_PNG
elif icon_file.lower()[-3:] == "gif":
return ICON_GIF
elif icon_file.lower()[-3:] == "jpg":
return ICON_JPEG
return ICON_NONE