mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-13 00:26:35 +00:00
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
# Copyright (c) 2018 Atmosphere-NX
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms and conditions of the GNU General Public License,
|
|
# version 2, as published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope 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, see <http://www.gnu.org/licenses/>.
|
|
import usb, zlib
|
|
from struct import unpack as up, pack as pk
|
|
|
|
def crc32(s):
|
|
return zlib.crc32(s) & 0xFFFFFFFF
|
|
|
|
class UsbInterface():
|
|
def __init__(self):
|
|
self.dev = usb.core.find(idVendor=0x057e, idProduct=0x3000)
|
|
if self.dev is None:
|
|
raise ValueError('Device not found')
|
|
|
|
self.dev.set_configuration()
|
|
self.cfg = self.dev.get_active_configuration()
|
|
self.intf = usb.util.find_descriptor(self.cfg, bInterfaceClass=0xff, bInterfaceSubClass=0xff, bInterfaceProtocol=0xfc)
|
|
assert self.intf is not None
|
|
|
|
self.ep_in = usb.util.find_descriptor(
|
|
self.intf,
|
|
custom_match = \
|
|
lambda e: \
|
|
usb.util.endpoint_direction(e.bEndpointAddress) == \
|
|
usb.util.ENDPOINT_IN)
|
|
assert self.ep_in is not None
|
|
|
|
self.ep_out = usb.util.find_descriptor(
|
|
self.intf,
|
|
custom_match = \
|
|
lambda e: \
|
|
usb.util.endpoint_direction(e.bEndpointAddress) == \
|
|
usb.util.ENDPOINT_OUT)
|
|
assert self.ep_out is not None
|
|
def close(self):
|
|
usb.util.dispose_resources(self.dev)
|
|
def blocking_read(self, size):
|
|
return ''.join(chr(x) for x in self.ep_in.read(size, 0xFFFFFFFFFFFFFFFF))
|
|
def blocking_write(self, data):
|
|
self.ep_out.write(data, 0xFFFFFFFFFFFFFFFF)
|
|
def read_packet(self):
|
|
hdr = self.blocking_read(0x28)
|
|
_, _, _, body_size, _, _, _, _, body_chk, hdr_chk = up('<IIIIIIIIII', hdr)
|
|
if crc32(hdr[:-4]) != hdr_chk:
|
|
raise ValueError('Invalid header checksum in received packet!')
|
|
body = self.blocking_read(body_size)
|
|
if len(body) != body_size:
|
|
raise ValueError('Failed to receive packet body!')
|
|
elif crc32(body) != body_chk:
|
|
raise ValueError('Invalid body checksum in received packet!')
|
|
return (hdr, body)
|
|
def send_packet(self, body):
|
|
hdr = pk('<IIIIIIIII', 0, 0, 0, len(body), 0, 0, 0, 0, crc32(body))
|
|
hdr += pk('<I', crc32(hdr))
|
|
self.blocking_write(hdr)
|
|
self.blocking_write(body)
|
|
|
|
|