Python Network / CIDR Расчеты - PullRequest
Новая
       25

Python Network / CIDR Расчеты

3 голосов
/ 06 февраля 2011

Я работаю над созданием встроенного сетевого устройства (на основе linux) и столкнулся с необходимостью динамически создавать файлы conf демона.Таким образом, я должен иметь возможность выполнять некоторые вычисления сетевых адресов в коде Python, который будет создавать файлы conf.Я не программист, поэтому я боюсь, что написал модуль, который не сможет функционировать, как я надеялся, когда устройство начнет поставляться.Я мог бы найти на этом сайте и Google.

Есть ли лучший способ найти сетевой адрес и cidr для сетевого интерфейса?Преобразование сетевой маски в строку bin и подсчет единиц выглядят довольно не элегантно. 

import socket
import fcntl
import struct

SIOCGIFNETMASK = 0x891b
SIOCGIFADDR = 0x8915

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

def _GetIfaceMask(iface):
    return struct.unpack('L', fcntl.ioctl(s, SIOCGIFNETMASK, struct.pack('256s', iface))[20:24])[0]

def _GetIfaceAddr(iface):
    return struct.unpack('L', fcntl.ioctl(s, SIOCGIFADDR, struct.pack('256s', iface[:15]))[20:24])[0]

def GetIfaceNet(iface):
    net_addr = _GetIfaceAddr(iface) & _GetIfaceMask(iface)
    return socket.inet_ntoa(struct.pack('L', net_addr))

def GetIfaceCidr(iface):
    bin_str = bin(_GetIfaceMask(iface))[2:]
    cidr = 0
    for c in bin_str:
        if c == '1':  cidr += 1
    return cidr

Спасибо за любой вклад, я действительно несколько растерялся в этом.Если это не место для такого типа обратной связи, пожалуйста, дайте мне знать.

Ответы [ 2 ]

4 голосов
/ 06 февраля 2011

Это можно решить с помощью алгоритма веса Хэмминга.Похищен из Как посчитать количество установленных бит в 32-разрядном целом числе? и переведен на Python: 

def number_of_set_bits(x):
    x -= (x >> 1) & 0x55555555
    x = ((x >> 2) & 0x33333333) + (x & 0x33333333)
    x = ((x >> 4) + x) & 0x0f0f0f0f
    x += x >> 8
    x += x >> 16
    return x & 0x0000003f

Другое, более читаемое решение (но работает в O(log x)): 

def number_of_set_bits(x):
    n = 0
    while x:
        n += x & 1
        x = x >> 1
    return n
2 голосов
/ 06 февраля 2011

Вы можете проверить модуль iptools Python http://code.google.com/p/python-iptools/ он может конвертировать из длинного в точечный формат ip и наоборот.

...