ddp.py

# -*- coding: utf-8 -*-
from __future__ import print_function

import re
import socket
import logging

UDP_IP = ''
UDP_PORT = 0

DDP_PORT = 987
DDP_VERSION = '00020020'

_LOGGER = logging.getLogger(__name__)

def get_ddp_message(msg_type, data=None):
    """Get DDP message."""
    msg = u'{} * HTTP/1.1\n'.format(msg_type)
    if data:
        for key, value in data.items():
            msg += '{}:{}\n'.format(key, value)
    msg += 'device-discovery-protocol-version:{}\n'.format(DDP_VERSION)
    return msg


def parse_ddp_response(rsp):
    """Parse the response."""
    data = {}
    for line in rsp.splitlines():
        re_status = re.compile(r'HTTP/1.1 (?P<code>\d+) (?P<status>.*)')
        line = line.strip()
        # skip empty lines
        if not line:
            continue
        elif re_status.match(line):
            data[u'status_code'] = int(re_status.match(line).group('code'))
            data[u'status'] = re_status.match(line).group('status')
        else:
            values = line.split(':')
            data[values[0]] = values[1]
    return data


def get_ddp_search_message():
    """Get DDP search message."""
    return get_ddp_message('SRCH')


def get_ddp_wake_message(credential):
    """Get DDP wake message."""
    data = {
        'user-credential': credential,
        'client-type': 'a',
        'auth-type': 'C',
    }
    return get_ddp_message('WAKEUP', data)


def get_ddp_launch_message(credential):
    """Get DDP launch message."""
    data = {
        'user-credential': credential,
        'client-type': 'a',
        'auth-type': 'C',
    }
    return get_ddp_message('LAUNCH', data)


def _send_recv_msg(host, broadcast, msg, receive=True):
    """Send a ddp message and receive the response."""

    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    except OSError as error:
        _LOGGER.error('failed to create socket, %s', error)
        return [ None, host ]
    try:
        sock.bind((UDP_IP, UDP_PORT))
        sock.settimeout(3.0)
    except OSError as error:
        _LOGGER.error('failed to bind socket %s:%s, %s', UDP_IP, UDP_PORT, error)
        sock.close()
        return [ None, host ]

    try:
        if broadcast:
            sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
            host = '255.255.255.255'

        _LOGGER.debug('send_recv_msg %s [%s]', host, msg)
        sock.sendto(msg.encode('utf-8'), (host, DDP_PORT))

        if receive:
            return sock.recvfrom(1024)

        sock.close()
    except OSError as error:
        _LOGGER.error('failed to send data using socket, %s', error)
        sock.close()
        return [ None, host ]


def _send_msg(host, broadcast, msg):
    """Send a ddp message."""
    _send_recv_msg(host, broadcast, msg, receive=False)


def search(host=None, broadcast=True):
    """Discover PS4s."""
    ps_list = []
    msg = get_ddp_search_message()

    data, addr = _send_recv_msg(host, broadcast, msg)
    if data is not None:
        data = parse_ddp_response(data.decode('utf-8'))
        data[u'host-ip'] = addr[0]
        ps_list.append(data)
    return ps_list


def get_status(host):
    """Get status."""
    for ps_list in search(host=host):
        return ps_list
    return {}


def wakeup(host, credential, broadcast=None):
    """Wakeup PS4s."""
    msg = get_ddp_wake_message(credential)
    _send_msg(host, broadcast, msg)


def launch(host, credential, broadcast=None):
    """Launch."""
    msg = get_ddp_launch_message(credential)
    _send_msg(host, broadcast, msg)