From 5d4f0feebf2fd548308301a9463a844dcc123a1e Mon Sep 17 00:00:00 2001
From: Piv <18462828+Piv200@users.noreply.github.com>
Date: Sun, 17 Nov 2019 17:17:41 +1030
Subject: [PATCH] Add credentials and secure port to motor server.

---
 MotorControl/MotorServer.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/MotorControl/MotorServer.py b/MotorControl/MotorServer.py
index c8d8686..c82240a 100755
--- a/MotorControl/MotorServer.py
+++ b/MotorControl/MotorServer.py
@@ -57,11 +57,20 @@ class MotorServicer(motorService_pb2_grpc.CarControlServicer):
     def start_server(self):
         server = grpc.server(futures.ThreadPoolExecutor(max_workers=8))
         motorService_pb2_grpc.add_CarControlServicer_to_server(self, server)
-        server.add_insecure_port('[::]:50051')
+        server.add_secure_port('[::]:50051', self.create_credentials())
         server.start()
         while True:
             time.sleep(60*60)
 
+    def create_credentials(self):
+        pvtKeyPath = '/home/pi/tls/device.key'
+        pvtCertPath = '/home/pi/tls/device.crt'
+
+        pvtKeyBytes = open(pvtKeyPath).read()
+        pvtCertBytes = open(pvtCertPath).read()
+
+        return grpc.ssl_channel_credentials(pvtCertBytes, pvtKeyBytes)
+
 motor = Motor()
 servo = Servo(servo_pin)
 servicer = MotorServicer(motor, servo)