140 lines
4.1 KiB
Python
140 lines
4.1 KiB
Python
#
|
|
# Copyright (c) 2016-2023 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
from __future__ import print_function
|
|
|
|
import time
|
|
import socket
|
|
|
|
database_name = "/var/lib/sm/sm.db"
|
|
database_running_name = "/var/run/sm/sm.db"
|
|
|
|
SM_API_SERVER_ADDR = "/tmp/.sm_server_api"
|
|
|
|
SM_API_MSG_VERSION = "1"
|
|
SM_API_MSG_REVISION = "1"
|
|
|
|
SM_API_MSG_TYPE_RESTART_SERVICE = "RESTART_SERVICE"
|
|
SM_API_MSG_SKIP_DEP_CHECK = "skip-dep"
|
|
|
|
SM_API_MSG_TYPE_PROVISION_SERVICE = "PROVISION_SERVICE"
|
|
SM_API_MSG_TYPE_DEPROVISION_SERVICE = "DEPROVISION_SERVICE"
|
|
|
|
SM_API_MSG_TYPE_PROVISION_SERVICE_DOMAIN_INTERFACE = "PROVISION_SERVICE_DOMAIN_INTERFACE"
|
|
SM_API_MSG_TYPE_DEPROVISION_SERVICE_DOMAIN_INTERFACE = "DEPROVISION_SERVICE_DOMAIN_INTERFACE"
|
|
|
|
SM_API_MSG_TYPE_RELOAD_DATA = "RELOAD_DATA"
|
|
|
|
SM_API_MSG_TYPE_SDI_SET_STATE = "SERVICE_DOMAIN_INTERFACE_SET_STATE"
|
|
|
|
# offsets
|
|
SM_API_MSG_VERSION_FIELD = 0
|
|
SM_API_MSG_REVISION_FIELD = 1
|
|
SM_API_MSG_SEQNO_FIELD = 2
|
|
SM_API_MSG_TYPE_FIELD = 3
|
|
SM_API_MSG_ORIGIN_FIELD = 4
|
|
SM_API_MSG_SERVICE_NAME_FIELD = 5
|
|
# For provisioning a service domain interface, the SM API
|
|
# message expects the service domain name at the same offset
|
|
# as the service name field (used for provisioning a service)
|
|
SM_API_MSG_SERVICE_DOMAIN_NAME_FIELD = 5
|
|
SM_API_MSG_PARAM = 6
|
|
|
|
|
|
def _send_msg_to_sm(sm_api_msg):
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
|
|
try:
|
|
s.setblocking(True)
|
|
s.sendto(sm_api_msg.encode("ascii", "ignore"), SM_API_SERVER_ADDR)
|
|
time.sleep(1)
|
|
|
|
except socket.error as e:
|
|
print("sm-api socket error: %s on %s" % (e, sm_api_msg))
|
|
|
|
|
|
def restart_service(service_name):
|
|
"""
|
|
Message SM to restart a service
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_RESTART_SERVICE, "sm-action",
|
|
service_name))
|
|
|
|
_send_msg_to_sm(sm_api_msg)
|
|
|
|
|
|
def restart_service_safe(service_name):
|
|
"""
|
|
Message SM to restart a service w/o checking dependency
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_RESTART_SERVICE, "sm-action",
|
|
service_name, SM_API_MSG_SKIP_DEP_CHECK))
|
|
|
|
_send_msg_to_sm(sm_api_msg)
|
|
|
|
|
|
def provision_service(service_name, service_group_name):
|
|
"""
|
|
|
|
:param service_name:
|
|
:param service_group_name:
|
|
:return:
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_PROVISION_SERVICE, "sm-action",
|
|
service_name, service_group_name))
|
|
_send_msg_to_sm(sm_api_msg)
|
|
|
|
|
|
def deprovision_service(service_name, service_group_name):
|
|
"""
|
|
|
|
:param service_name:
|
|
:param service_group_name:
|
|
:return:
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_DEPROVISION_SERVICE, "sm-action",
|
|
service_name, service_group_name))
|
|
_send_msg_to_sm(sm_api_msg)
|
|
|
|
|
|
def provision_service_domain_interface(service_domain,
|
|
service_domain_interface_name):
|
|
"""
|
|
|
|
:param service_domain:
|
|
:param service_domain_interface_name:
|
|
:return:
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_PROVISION_SERVICE_DOMAIN_INTERFACE,
|
|
"sm-action", service_domain,
|
|
service_domain_interface_name))
|
|
_send_msg_to_sm(sm_api_msg)
|
|
|
|
|
|
def deprovision_service_domain_interface(service_domain,
|
|
service_domain_interface_name):
|
|
"""
|
|
|
|
:param service_domain:
|
|
:param service_domain_interface_name:
|
|
:return:
|
|
"""
|
|
sm_api_msg = ("%s,%s,%i,%s,%s,%s,%s"
|
|
% (SM_API_MSG_VERSION, SM_API_MSG_REVISION, 1,
|
|
SM_API_MSG_TYPE_DEPROVISION_SERVICE_DOMAIN_INTERFACE,
|
|
"sm-action", service_domain,
|
|
service_domain_interface_name))
|
|
_send_msg_to_sm(sm_api_msg)
|