Check for IP environment variable in so-allow and so-deny

This commit is contained in:
William Wernert
2021-10-21 13:54:06 -04:00
parent 15fe7512b7
commit f374dcbb58
2 changed files with 36 additions and 28 deletions

View File

@@ -183,8 +183,12 @@ def main():
return_code = e.errno
sys.exit(return_code)
elif args.roles is not None and args.ip is None:
if os.environ.get('IP') is None:
main_parser.print_help()
sys.exit(1)
else:
args.ip = os.environ['IP']
if validate_ip_cidr(args.ip):
try:
for role in args.roles:

View File

@@ -184,8 +184,12 @@ def main():
return_code = e.errno
sys.exit(return_code)
elif args.roles is not None and args.ip is None:
if os.environ.get('IP') is None:
main_parser.print_help()
sys.exit(1)
else:
args.ip = os.environ['IP']
if validate_ip_cidr(args.ip):
try:
for role in args.roles: