Wala shell

Advanced PHP File Manager

Server IP: 127.0.0.1 | PHP Version: 8.3.30 | Domains Hosted: 54

Name	Size	Last Modified	Actions
📁 ..	-	2025-02-24 20:22
📄 linux_test.py # Copyright 2016 Google Inc. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. """Tests for pyu2f.hid.linux.""" import base64 import os import sys import mock from pyu2f.hid import linux # Since the builtins name changed between Python 2 and Python 3, we have to # make sure to mock the corret one. if sys.version_info[0] > 2: import builtins as py_builtins else: import __builtin__ as py_builtins try: from pyfakefs import fake_filesystem # pylint: disable=g-import-not-at-top except ImportError: from fakefs import fake_filesystem # pylint: disable=g-import-not-at-top if sys.version_info[:2] < (2, 7): import unittest2 as unittest # pylint: disable=g-import-not-at-top else: import unittest # pylint: disable=g-import-not-at-top # These are base64 encoded report descriptors of a Yubico token # and a Logitech keyboard. YUBICO_RD = 'BtDxCQGhAQkgFQAm/wB1CJVAgQIJIRUAJv8AdQiVQJECwA==' KEYBOARD_RD = ( 'BQEJAqEBCQGhAAUJGQEpBRUAJQGVBXUBgQKVAXUDgQEFAQkwCTEJOBWBJX91CJUDgQbAwA==') def AddDevice(fs, dev_name, product_name, vendor_id, product_id, report_descriptor_b64): uevent = fs.create_file('/sys/class/hidraw/%s/device/uevent' % dev_name) rd = fs.create_file('/sys/class/hidraw/%s/device/report_descriptor' % dev_name) report_descriptor = base64.b64decode(report_descriptor_b64) rd.set_contents(report_descriptor) buf = 'HID_NAME=%s\n' % product_name.encode('utf8') buf += 'HID_ID=0001:%08X:%08X\n' % (vendor_id, product_id) uevent.set_contents(buf) class FakeDeviceOsModule(object): O_RDWR = os.O_RDWR path = os.path data_written = None data_to_return = None def open(self, unused_path, unused_opts): # pylint: disable=invalid-name return 0 def write(self, unused_dev, data): # pylint: disable=invalid-name self.data_written = data def read(self, unused_dev, unused_length): # pylint: disable=invalid-name return self.data_to_return class LinuxTest(unittest.TestCase): def setUp(self): self.fs = fake_filesystem.FakeFilesystem() self.fs.create_dir('/sys/class/hidraw') def tearDown(self): self.fs.remove_object('/sys/class/hidraw') def testCallEnumerate(self): AddDevice(self.fs, 'hidraw1', 'Logitech USB Keyboard', 0x046d, 0xc31c, KEYBOARD_RD) AddDevice(self.fs, 'hidraw2', 'Yubico U2F', 0x1050, 0x0407, YUBICO_RD) with mock.patch.object(linux, 'os', fake_filesystem.FakeOsModule(self.fs)): fake_open = fake_filesystem.FakeFileOpen(self.fs) with mock.patch.object(py_builtins, 'open', fake_open): devs = list(linux.LinuxHidDevice.Enumerate()) devs = sorted(devs, key=lambda k: (k['vendor_id'])) self.assertEqual(len(devs), 2) self.assertEqual(devs[0]['vendor_id'], 0x046d) self.assertEqual(devs[0]['product_id'], 0x0c31c) self.assertEqual(devs[1]['vendor_id'], 0x1050) self.assertEqual(devs[1]['product_id'], 0x0407) self.assertEqual(devs[1]['usage_page'], 0xf1d0) self.assertEqual(devs[1]['usage'], 1) def testCallOpen(self): AddDevice(self.fs, 'hidraw1', 'Yubico U2F', 0x1050, 0x0407, YUBICO_RD) fake_open = fake_filesystem.FakeFileOpen(self.fs) # The open() builtin is used to read from the fake sysfs with mock.patch.object(py_builtins, 'open', fake_open): # The os.open function is used to interact with the low # level device. We don't want to use fakefs for that. fake_dev_os = FakeDeviceOsModule() with mock.patch.object(linux, 'os', fake_dev_os): dev = linux.LinuxHidDevice('/dev/hidraw1') self.assertEqual(dev.GetInReportDataLength(), 64) self.assertEqual(dev.GetOutReportDataLength(), 64) dev.Write(list(range(0, 64))) # The HidDevice implementation prepends a zero-byte representing the # report ID self.assertEqual(list(fake_dev_os.data_written), [0] + list(range(0, 64))) fake_dev_os.data_to_return = b'x' * 64 self.assertEqual(dev.Read(), [120] * 64) # chr(120) = 'x' if __name__ == '__main__': unittest.main()	4.51 KB	2025-02-24 20:22
📄 macos_test.py # Copyright 2016 Google Inc. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. """Tests for pyu2f.hid.macos.""" import ctypes import sys import mock from pyu2f import errors from pyu2f.hid import macos if sys.version_info[:2] < (2, 7): import unittest2 as unittest # pylint: disable=g-import-not-at-top else: import unittest # pylint: disable=g-import-not-at-top def init_mock_iokit(mock_iokit): # Device open should always return 0 (success) mock_iokit.IOHIDDeviceOpen = mock.Mock(return_value=0) mock_iokit.IOHIDDeviceSetReport = mock.Mock(return_value=0) mock_iokit.IOHIDDeviceCreate = mock.Mock(return_value='handle') def init_mock_cf(mock_cf): mock_cf.CFGetTypeID = mock.Mock(return_value=123) mock_cf.CFNumberGetTypeID = mock.Mock(return_value=123) mock_cf.CFStringGetTypeID = mock.Mock(return_value=123) def init_mock_get_int_property(mock_get_int_property): mock_get_int_property.return_value = 64 class MacOsTest(unittest.TestCase): @mock.patch.object(macos.threading, 'Thread') @mock.patch.multiple(macos, iokit=mock.DEFAULT, cf=mock.DEFAULT, GetDeviceIntProperty=mock.DEFAULT) def testInitHidDevice(self, thread, iokit, cf, GetDeviceIntProperty): # pylint: disable=invalid-name init_mock_iokit(iokit) init_mock_cf(cf) init_mock_get_int_property(GetDeviceIntProperty) device = macos.MacOsHidDevice('fakepath') self.assertEqual(64, device.GetInReportDataLength()) self.assertEqual(64, device.GetOutReportDataLength()) @mock.patch.object(macos.threading, 'Thread') @mock.patch.multiple(macos, iokit=mock.DEFAULT, cf=mock.DEFAULT, GetDeviceIntProperty=mock.DEFAULT) def testCallWriteSuccess(self, thread, iokit, cf, GetDeviceIntProperty): # pylint: disable=invalid-name init_mock_iokit(iokit) init_mock_cf(cf) init_mock_get_int_property(GetDeviceIntProperty) device = macos.MacOsHidDevice('fakepath') # Write 64 bytes to device data = bytearray(range(64)) # Write to device device.Write(data) # Validate that write calls into IOKit set_report_call_args = iokit.IOHIDDeviceSetReport.call_args self.assertIsNotNone(set_report_call_args) set_report_call_pos_args = iokit.IOHIDDeviceSetReport.call_args[0] self.assertEqual(len(set_report_call_pos_args), 5) self.assertEqual(set_report_call_pos_args[0], 'handle') self.assertEqual(set_report_call_pos_args[1], 1) self.assertEqual(set_report_call_pos_args[2], 0) self.assertEqual(set_report_call_pos_args[4], 64) report_buffer = set_report_call_pos_args[3] self.assertEqual(len(report_buffer), 64) self.assertEqual(bytearray(report_buffer), data, 'Data sent to ' 'IOHIDDeviceSetReport should match data sent to the ' 'device') @mock.patch.object(macos.threading, 'Thread') @mock.patch.multiple(macos, iokit=mock.DEFAULT, cf=mock.DEFAULT, GetDeviceIntProperty=mock.DEFAULT) def testCallWriteFailure(self, thread, iokit, cf, GetDeviceIntProperty): # pylint: disable=invalid-name init_mock_iokit(iokit) init_mock_cf(cf) init_mock_get_int_property(GetDeviceIntProperty) # Make set report call return failure (non-zero) iokit.IOHIDDeviceSetReport.return_value = -1 device = macos.MacOsHidDevice('fakepath') # Write 64 bytes to device data = bytearray(range(64)) # Write should throw an OsHidError exception with self.assertRaises(errors.OsHidError): device.Write(data) @mock.patch.object(macos.threading, 'Thread') @mock.patch.multiple(macos, iokit=mock.DEFAULT, cf=mock.DEFAULT, GetDeviceIntProperty=mock.DEFAULT) def testCallReadSuccess(self, thread, iokit, cf, GetDeviceIntProperty): # pylint: disable=invalid-name init_mock_iokit(iokit) init_mock_cf(cf) init_mock_get_int_property(GetDeviceIntProperty) device = macos.MacOsHidDevice('fakepath') # Call callback for IN report report = (ctypes.c_uint8 * 64)() report[:] = range(64)[:] q = device.read_queue macos.HidReadCallback(q, None, None, None, 0, report, 64) # Device read should return the callback data read_result = device.Read() self.assertEqual(read_result, list(range(64)), 'Read data should match ' 'data passed into the callback') if __name__ == '__main__': unittest.main()	4.83 KB	2025-02-24 20:22

Terminal