58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
# Copyright (C) 2025 James Brotosky, Brandon Wickline
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
import logging
|
|
from services.agenthandler import selectAgents
|
|
from utils.utils import colorText
|
|
from utils.selector import Selector
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
from services.API import AirlockAPIWrapper
|
|
|
|
|
|
def generate(api: AirlockAPIWrapper):
|
|
agents = selectAgents(api)
|
|
purpose = input(colorText(" Please enter the purpose for the OTP: ", "white"))
|
|
possible_durations = [15, 60, 360, 1440, 10080]
|
|
|
|
print(colorText("Please select a duration:", "white"))
|
|
duration_selected = Selector.select_int(possible_durations)
|
|
otp_dict = {}
|
|
if duration_selected:
|
|
for agent in agents:
|
|
otp_code = api.otp_generate(agent.agentid, duration_selected, purpose)
|
|
logger.info(f"Generated OTP for {agent.hostname}: {otp_code}")
|
|
otp_dict[agent.hostname] = otp_code
|
|
|
|
return otp_dict
|
|
|
|
def otp_activities_by_agent(api: AirlockAPIWrapper):
|
|
agents = selectAgents(api)
|
|
otp_dict = {}
|
|
for agent in agents:
|
|
otp_info = api.otp_find_by_agent(agent.agentid)
|
|
otp_dict[agent.hostname] = otp_info
|
|
|
|
return otp_dict
|
|
|
|
def revoke(api: AirlockAPIWrapper):
|
|
otp_dict = otp_activities_by_agent(api)
|
|
list_to_revoke = [entry["otpid"] for entry in otp_dict]
|
|
for revokee in list_to_revoke:
|
|
api.otp_revoke(revokee) |