2 * ============LICENSE_START========================================================================
\r
3 * ONAP : tr-069-adapter
\r
4 * =================================================================================================
\r
5 * Copyright (C) 2020 CommScope Inc Intellectual Property.
\r
6 * =================================================================================================
\r
7 * This tr-069-adapter software file is distributed by CommScope Inc under the Apache License,
\r
8 * Version 2.0 (the "License"); you may not use this file except in compliance with the License. You
\r
9 * may obtain a copy of the License at
\r
11 * http://www.apache.org/licenses/LICENSE-2.0
\r
13 * This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
\r
14 * either express or implied. See the License for the specific language governing permissions and
\r
15 * limitations under the License.
\r
16 * ===============LICENSE_END=======================================================================
\r
19 package org.commscope.tr069adapter.netconf.server.helper;
\r
21 import java.io.IOException;
\r
22 import java.net.Socket;
\r
23 import java.util.HashMap;
\r
24 import java.util.Map;
\r
25 import java.util.PriorityQueue;
\r
26 import java.util.concurrent.Semaphore;
\r
28 import javax.annotation.PostConstruct;
\r
30 import org.commscope.tr069adapter.netconf.config.NetConfServerProperties;
\r
31 import org.commscope.tr069adapter.netconf.error.ServerPortAllocationException;
\r
32 import org.slf4j.Logger;
\r
33 import org.slf4j.LoggerFactory;
\r
34 import org.springframework.beans.factory.annotation.Autowired;
\r
35 import org.springframework.stereotype.Component;
\r
38 public class ServerPortAllocationHelper {
\r
40 private static final Logger LOG = LoggerFactory.getLogger(ServerPortAllocationHelper.class);
\r
42 private static Map<String, Semaphore> semaphoreMap = new HashMap<>();
\r
44 private PriorityQueue<String> availablePorts = new PriorityQueue<>();
\r
47 NetConfServerProperties config;
\r
50 public void init() {
\r
51 // read the port range and it the available ports.
\r
53 Integer startPort = config.getDefaultNetconfStartPort();
\r
54 Integer maxServers = config.getDefaultMaxServers();
\r
57 startPort = Integer.parseInt(config.getNetconfServersStartPort());
\r
58 } catch (Exception e) {
\r
60 "Failed to initialize the starting port from the environment {}. Hence using the default port range.",
\r
61 config.getNetconfServersStartPort());
\r
65 maxServers = Integer.parseInt(config.getMaxNumOfNetconfServers());
\r
66 } catch (Exception e) {
\r
68 "Failed to initialize the max netconf server from the environment {} Hence using the default max servers.",
\r
69 config.getMaxNumOfNetconfServers());
\r
72 for (int i = startPort + maxServers - 1; i >= startPort; i--) {
\r
73 semaphoreMap.put(String.valueOf(i), new Semaphore(1));
\r
74 availablePorts.add(String.valueOf(i));
\r
76 LOG.debug("Successfully populated available ports list.");
\r
79 public String reserveServerPort() throws ServerPortAllocationException {
\r
81 if (availablePorts.isEmpty()) {
\r
83 "All ports are exhausted. Hence cannot allocate a port to start new netconf server.");
\r
86 String port = availablePorts.peek();
\r
88 LOG.debug("Trying to reserve port : {}", port);
\r
89 if (isServerPortInUse(port)) {
\r
90 LOG.debug("Port {} is already in use.", port);
\r
91 return reserveServerPort(); // retry if current port is not available
\r
94 Semaphore semaphore = semaphoreMap.get(port);
\r
95 boolean isAcquired = semaphore.tryAcquire();
\r
97 LOG.debug("Failed to acquire a lock for port :{}. Hence retrying...", port);
\r
98 return reserveServerPort();
\r
101 availablePorts.poll();
\r
102 semaphore.release();
\r
103 LOG.debug("Rserved port is {}", port);
\r
107 public boolean checkAndReserveServerPort(String port) {
\r
110 Semaphore semaphore = semaphoreMap.get(port);
\r
111 semaphore.acquire();
\r
112 if (isServerPortInUse(port)) {
\r
113 LOG.error("Port {} already in use.", port);
\r
114 semaphore.release();
\r
117 availablePorts.remove(port);
\r
118 semaphore.release();
\r
119 LOG.error("Successfully reserved the port {} to start netconf server", port);
\r
120 } catch (InterruptedException e) {
\r
121 Thread.currentThread().interrupt();
\r
122 LOG.error("Failed to lock the port {} : Exception :{}", port, e.toString());
\r
123 return checkAndReserveServerPort(port); // retry acquiring the lock.
\r
129 public boolean isServerPortInUse(String port) {
\r
130 return checkIfPortAvailable(port);
\r
133 private static boolean checkIfPortAvailable(String portStr) {
\r
134 Integer port = Integer.parseInt(portStr);
\r
135 try (Socket ignored = new Socket("localhost", port)) {
\r
137 } catch (IOException e) {
\r
138 LOG.error("while checkIfPortAvailable {}", e.toString());
\r