RepoMirrors/setools
1
0
Fork 0
mirror of https://github.com/SELinuxProject/setools synced 2025-04-11 03:51:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

PortconProtocol: add support for specifiying by protocol string

Browse Source 
Leverage Python lib socket.getprotobyname()
This commit is contained in:
Chris PeBenito 2016-04-01 10:42:00 -04:00
parent 57d446de1b
commit 6c147f8c7b
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
setools/policyrep/netcontext.py
View File

@ -16,7 +16,7 @@
# License along with SETools. If not, see # License along with SETools. If not, see
# <http://www.gnu.org/licenses/>. # <http://www.gnu.org/licenses/>.
# #
import socket from socket import IPPROTO_TCP, IPPROTO_UDP, getprotobyname
from collections import namedtuple from collections import namedtuple
from . import qpol from . import qpol
@ -146,16 +146,22 @@ class PortconProtocol(int):
corresponding protocol string (udp, tcp). corresponding protocol string (udp, tcp).
""" """
_proto_to_text = {socket.IPPROTO_TCP: 'tcp', _proto_to_text = {IPPROTO_TCP: 'tcp',
socket.IPPROTO_UDP: 'udp'} IPPROTO_UDP: 'udp'}
def __new__(cls, value): def __new__(cls, value):
if value not in cls._proto_to_text: try:
raise ValueError("{0} is not a supported IP protocol number. " # convert string representation
"Values such as {1} (TCP) or {2} (UDP) should be used.". num = getprotobyname(value)
format(value, socket.IPPROTO_TCP, socket.IPPROTO_UDP)) except TypeError:
num = value
return super(PortconProtocol, cls).__new__(cls, value) if num not in cls._proto_to_text:
raise ValueError("{0} is not a supported IP protocol. "
"Values such as {1} (TCP) or {2} (UDP) should be used.".
format(value, IPPROTO_TCP, IPPROTO_UDP))
return super(PortconProtocol, cls).__new__(cls, num)
def __str__(self): def __str__(self):
return self._proto_to_text[self] return self._proto_to_text[self]