pysnmp-sky/pysnmp/carrier/asyncore/dgram/unix.py

62 lines
1.6 KiB
Python

#
# This file is part of pysnmp software.
#
# Copyright (c) 2005-2017, Ilya Etingof <etingof@gmail.com>
# License: http://pysnmp.sf.net/license.html
#
import os
import random
try:
from socket import AF_UNIX
except ImportError:
AF_UNIX = None
from pysnmp.carrier.base import AbstractTransportAddress
from pysnmp.carrier.asyncore.dgram.base import DgramSocketTransport
domainName = snmpLocalDomain = (1, 3, 6, 1, 2, 1, 100, 1, 13)
random.seed()
class UnixTransportAddress(str, AbstractTransportAddress):
pass
class UnixSocketTransport(DgramSocketTransport):
sockFamily = AF_UNIX
addressType = UnixTransportAddress
_iface = ''
def openClientMode(self, iface=None):
if iface is None:
# UNIX domain sockets must be explicitly bound
iface = ''
while len(iface) < 8:
iface += chr(random.randrange(65, 91))
iface += chr(random.randrange(97, 123))
iface = os.path.sep + 'tmp' + os.path.sep + 'pysnmp' + iface
if os.path.exists(iface):
os.remove(iface)
DgramSocketTransport.openClientMode(self, iface)
self._iface = iface
return self
def openServerMode(self, iface):
DgramSocketTransport.openServerMode(self, iface)
self._iface = iface
return self
def closeTransport(self):
DgramSocketTransport.closeTransport(self)
try:
os.remove(self._iface)
except OSError:
pass
UnixTransport = UnixSocketTransport
# Compatibility stub
UnixDgramSocketTransport = UnixSocketTransport