p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Add get_local_network_interfaces()

This commit is contained in:
Mike Salvatore 2021-12-16 13:15:46 -05:00
parent 332649d5d1
commit da3c6a4245
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/infection_monkey/network/info.py
View File

@ -1,7 +1,9 @@
import itertools import itertools
import socket import socket
import struct import struct
from ipaddress import IPv4Network
from random import randint # noqa: DUO102 from random import randint # noqa: DUO102
from typing import List
import netifaces import netifaces
import psutil import psutil
@ -9,6 +11,8 @@ import psutil
from common.network.network_range import CidrRange from common.network.network_range import CidrRange
from infection_monkey.utils.environment import is_windows_os from infection_monkey.utils.environment import is_windows_os
from . import NetworkInterface
# Timeout for monkey connections # Timeout for monkey connections
TIMEOUT = 15 TIMEOUT = 15
LOOPBACK_NAME = b"lo" LOOPBACK_NAME = b"lo"
@ -18,6 +22,14 @@ RTF_UP = 0x0001 # Route usable
RTF_REJECT = 0x0200 RTF_REJECT = 0x0200
def get_local_network_interfaces() -> List[NetworkInterface]:
for i in get_host_subnets():
netmask_bits = IPv4Network(f"{i['addr']}/{i['netmask']}", strict=False).prefixlen
cidr_netmask = f"/{netmask_bits}"
return [NetworkInterface(i["addr"], cidr_netmask) for i in get_host_subnets()]
def get_host_subnets(): def get_host_subnets():
""" """
Returns a list of subnets visible to host (omitting loopback and auto conf networks) Returns a list of subnets visible to host (omitting loopback and auto conf networks)