From c388ec92c36ff6c5f5ddc4a10f6c220a9b1b04c0 Mon Sep 17 00:00:00 2001
From: Yelin Jeong <yelini.jeong@samsung.com>
Date: Mon, 22 Apr 2024 20:56:21 +0900
Subject: [PATCH] App/Service: Find available port using serverSocket

This patch adds initPort function to find availble port for tensor_query_server

Signed-off-by: Yelin Jeong <yelini.jeong@samsung.com>
---
 .../src/main/AndroidManifest.xml               |  1 +
 .../ml/inference/offloading/MainService.kt     | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/ml_inference_offloading/src/main/AndroidManifest.xml b/ml_inference_offloading/src/main/AndroidManifest.xml
index c572016..fa91404 100644
--- a/ml_inference_offloading/src/main/AndroidManifest.xml
+++ b/ml_inference_offloading/src/main/AndroidManifest.xml
@@ -4,6 +4,7 @@
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE"/>
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE_SPECIAL_USE"/>
     <uses-permission android:name="android.permission.POST_NOTIFICATIONS" />
+    <uses-permission android:name="android.permission.INTERNET" />
 
     <application
         android:icon="@mipmap/ic_launcher"
diff --git a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
index c5732ed..4484891 100644
--- a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
+++ b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
@@ -19,6 +19,8 @@ import androidx.core.app.NotificationCompat
 import androidx.core.app.ServiceCompat
 import androidx.core.content.ContextCompat
 import org.nnsuite.nnstreamer.NNStreamer
+import java.net.ServerSocket
+import kotlin.concurrent.thread
 
 
 class MainService : Service() {
@@ -40,6 +42,7 @@ class MainService : Service() {
     private lateinit var serviceLooper : Looper
     private lateinit var handlerThread: HandlerThread
     private var initialized = false
+    private var serverPort = 0
 
     private fun startForeground() {
         // Get NotificationManager
@@ -78,6 +81,7 @@ class MainService : Service() {
 
     override fun onCreate() {
         initNNStreamer()
+        initPort()
         handlerThread = HandlerThread("ServiceStartArguments", Process.THREAD_PRIORITY_BACKGROUND).apply {
             start()
         }
@@ -125,4 +129,18 @@ class MainService : Service() {
             }
         }
     }
+
+    private fun initPort() {
+        val portFinder = thread() {
+            try {
+                val serverSocket = ServerSocket(0)
+                Log.i(TAG, "listening on port: " + serverSocket.localPort)
+                serverPort = serverSocket.localPort
+                serverSocket.close()
+            } catch (e: Exception) {
+                Log.e(TAG, e.toString())
+            }
+        }
+        portFinder.join()
+    }
 }