From 5a935d99fce3116d4dc59ea7a799f6720e1017a5 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Tue, 13 Sep 2022 13:29:42 -0400
Subject: [PATCH] Have SMM devices register cleanly

In remote discovery registration, the
SMM path was not fully implemented
---
 confluent_server/confluent/selfservice.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/confluent_server/confluent/selfservice.py b/confluent_server/confluent/selfservice.py
index 3cbf58aa..3a48db75 100644
--- a/confluent_server/confluent/selfservice.py
+++ b/confluent_server/confluent/selfservice.py
@@ -186,7 +186,14 @@ def handle_request(env, start_response):
             rb['addresses'] = [(newhost, newport)]
             rb['forwarder_url'] = targurl
             rb['forwarder_server'] = nodename
-            ssdp.check_fish(('/DeviceDescription.json', rb), newport, verify_cert)
+            if rb['type'] == 'lenovo-xcc':
+                ssdp.check_fish(('/DeviceDescription.json', rb), newport, verify_cert)
+            elif rb['type'] == 'lenovo-smm2':
+                rb['services'] = ['service:lenovo-smm2']
+            else:
+                start_response('400 Unsupported Device', [])
+                yield 'Unsupported device for remote discovery registration'
+                return
         disco.detected(rb)
         start_response('200 OK', [])
         yield 'Registered'