From b42fc6fb1f0997b4375f57a3f6229cc5d8dc075d Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <stick@gk2.sk>
Date: Wed, 13 Dec 2017 02:37:59 +0100
Subject: [PATCH] trezorctl: set homescreen for T2

---
 trezorctl | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/trezorctl b/trezorctl
index 7dc830b10..4ae4af461 100755
--- a/trezorctl
+++ b/trezorctl
@@ -209,7 +209,11 @@ def set_flags(connect, flags):
 @click.option('-f', '--filename', default=None)
 @click.pass_obj
 def set_homescreen(connect, filename):
-    if filename is not None:
+    if filename and filename.endswith('.toif'):
+        img = open(filename, 'rb').read()
+        if img[:8] != b'TOIf\x90\x00\x90\x00':
+            raise ValueError('File is not a TOIF file with size of 144x144')
+    elif filename is not None:
         from PIL import Image
         im = Image.open(filename)
         if im.size != (128, 64):