port-range
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

port-range
View Options

	#!/usr/bin/python3

	import sys
	import re

	from vyos.util import read_file

	services_file = '/etc/services'

	def get_services():
	names = []
	service_data = read_file(services_file, "")
	for line in service_data.split("\n"):
	if not line or line[0] == '#':
	continue
	names.append(line.split(None, 1)[0])
	return names

	def error(port_range):
	print(f'Error: {port_range} is not a valid port or port range')
	sys.exit(1)

	if __name__ == '__main__':
	if len(sys.argv)>1:
	port_range = sys.argv[1]
	if re.match('^[0-9]{1,5}-[0-9]{1,5}$', port_range):
	port_1, port_2 = port_range.split('-')
	if int(port_1) not in range(1, 65536) or int(port_2) not in range(1, 65536):
	error(port_range)
	if int(port_1) > int(port_2):
	error(port_range)
	elif port_range.isnumeric() and int(port_range) not in range(1, 65536):
	error(port_range)
	elif not port_range.isnumeric() and port_range not in get_services():
	print(f'Error: {port_range} is not a valid service name')
	sys.exit(1)
	else:
	sys.exit(2)

	sys.exit(0)