From 8fe9460ed6b714ebbd5b3ca60779b47996c6b0a5 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Wed, 20 Dec 2017 15:05:07 +0200
Subject: [PATCH] trezor: allow UDP connection (for emulator)

---
 libagent/device/trezor.py      | 4 +++-
 libagent/device/trezor_defs.py | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/libagent/device/trezor.py b/libagent/device/trezor.py
index 3206242..7c88194 100644
--- a/libagent/device/trezor.py
+++ b/libagent/device/trezor.py
@@ -40,7 +40,9 @@ class Trezor(interface.Device):
         # Allow using TREZOR bridge transport (instead of the HID default)
         trezor_defs.Transport = {
             'bridge': trezor_defs.BridgeTransport,
-        }.get(os.environ.get('TREZOR_TRANSPORT'), trezor_defs.HidTransport)
+            'udp': trezor_defs.UdpTransport,
+            'hid': trezor_defs.HidTransport,
+        }[os.environ.get('TREZOR_TRANSPORT', 'hid')]
         return trezor_defs
 
     required_version = '>=1.4.0'
diff --git a/libagent/device/trezor_defs.py b/libagent/device/trezor_defs.py
index 044496c..62501ef 100644
--- a/libagent/device/trezor_defs.py
+++ b/libagent/device/trezor_defs.py
@@ -7,3 +7,4 @@ from trezorlib.client import TrezorClient as Client
 from trezorlib.messages import IdentityType, PassphraseAck, PinMatrixAck
 from trezorlib.transport_bridge import BridgeTransport
 from trezorlib.transport_hid import HidTransport
+from trezorlib.transport_udp import UdpTransport