near-rt-ric-simulator/src/common/maincommon.py

   1 #  ============LICENSE_START===============================================
   2 #  Copyright (C) 2020 Nordix Foundation. All rights reserved.
   3 #  ========================================================================
   4 #  Licensed under the Apache License, Version 2.0 (the "License");
   5 #  you may not use this file except in compliance with the License.
   6 #  You may obtain a copy of the License at
   7 #
   8 #       http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 #  Unless required by applicable law or agreed to in writing, software
  11 #  distributed under the License is distributed on an "AS IS" BASIS,
  12 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 #  See the License for the specific language governing permissions and
  14 #  limitations under the License.
  15 #  ============LICENSE_END=================================================
  16 #
  17
  18 import os
  19 import sys
  20 from pathlib import Path
  21 from flask import Response
  22 import socket
  23 import ssl
  24
  25 #Must exist
  26 apipath=os.environ['APIPATH']
  27 #May exist
  28 remote_hosts_logging=os.getenv('REMOTE_HOSTS_LOGGING')
  29
  30 # Make sure the api path for the interface yaml file is set, otherwise exit
  31 def check_apipath():
  32     if (apipath is None):
  33         print("Env APIPATH not set. Exiting....")
  34         sys.exit(1)
  35
  36 # Return a Response of currently supported interfaces
  37 def get_supported_interfaces_response():
  38     p=Path(os.getcwd())
  39     pp=p.parts
  40     arr = os.listdir("../")
  41     del arr[arr.index('common')] # Remove the common lib
  42     del arr[arr.index('start.sh')] # Remove the start script
  43     arr.sort()
  44     return Response("Current interface: " + str(pp[len(pp)-1]) + "  All supported A1 interface yamls in this container: "+str(arr), 200, mimetype='text/plain')
  45
  46 # Remote host lookup and store host name in a set
  47 def extract_host_name(hosts_set, request):
  48     if (remote_hosts_logging is not None):
  49         host_ip=str(request.environ['REMOTE_ADDR'])
  50         prefix='::ffff:'
  51         if (host_ip.startswith('::ffff:')):
  52             host_ip=host_ip[len(prefix):]
  53         try:
  54             name, alias, addresslist = socket.gethostbyaddr(host_ip)
  55             hosts_set.add(name)
  56         except Exception as e:
  57             hosts_set.add(host_ip)
  58     else:
  59         hosts_set.add("logging_of_remote_host_names_not_enabled")
  60
  61 # Check if cert is available and return a sec context, if not return 'None'
  62 def get_security_context():
  63
  64     try:
  65         path="/usr/src/app/cert"
  66         if (os.path.isdir(path)):
  67             certpath=path+"/cert.crt"
  68             keypath=path+"/key.crt"
  69             if (os.path.isfile(certpath) and os.path.isfile(keypath)):
  70                 context = ssl.SSLContext(ssl.PROTOCOL_TLS)
  71                 context.load_cert_chain(certpath, keypath, password="test")
  72                 return context
  73             else:
  74                 print("Cert and/or key does not exists in dir "+str(path))
  75
  76         else:
  77             print("Path "+str(path)+" to certificate and key does not exists")
  78     except Exception as e:
  79         print("Problem when loading cert and key: "+str(e))
  80     return None