diff --git a/netbox/ipam/fields.py b/netbox/ipam/fields.py index da07f68d9..c44385b6d 100644 --- a/netbox/ipam/fields.py +++ b/netbox/ipam/fields.py @@ -6,7 +6,7 @@ from django.db import models from .formfields import IPFormField from .lookups import ( EndsWith, IEndsWith, IRegex, IStartsWith, NetContained, NetContainedOrEqual, NetContains, NetContainsOrEquals, - NetHost, NetMaskLength, Regex, StartsWith, + NetHost, NetHostContained, NetMaskLength, Regex, StartsWith, ) @@ -66,7 +66,6 @@ IPNetworkField.register_lookup(NetContained) IPNetworkField.register_lookup(NetContainedOrEqual) IPNetworkField.register_lookup(NetContains) IPNetworkField.register_lookup(NetContainsOrEquals) -IPNetworkField.register_lookup(NetHost) IPNetworkField.register_lookup(NetMaskLength) @@ -91,4 +90,5 @@ IPAddressField.register_lookup(NetContainedOrEqual) IPAddressField.register_lookup(NetContains) IPAddressField.register_lookup(NetContainsOrEquals) IPAddressField.register_lookup(NetHost) +IPAddressField.register_lookup(NetHostContained) IPAddressField.register_lookup(NetMaskLength) diff --git a/netbox/ipam/filters.py b/netbox/ipam/filters.py index 42809b954..10a18d1b7 100644 --- a/netbox/ipam/filters.py +++ b/netbox/ipam/filters.py @@ -254,7 +254,7 @@ class IPAddressFilter(CustomFieldFilterSet, django_filters.FilterSet): return queryset try: query = str(IPNetwork(value.strip()).cidr) - return queryset.filter(address__net_contained_or_equal=query) + return queryset.filter(address__net_host_contained=query) except AddrFormatError: return queryset.none() diff --git a/netbox/ipam/lookups.py b/netbox/ipam/lookups.py index 8346169ce..05c69dfb2 100644 --- a/netbox/ipam/lookups.py +++ b/netbox/ipam/lookups.py @@ -89,6 +89,20 @@ class NetHost(Lookup): return 'HOST(%s) = %s' % (lhs, rhs), params +class NetHostContained(Lookup): + """ + Check for the host portion of an IP address without regard to its mask. This allows us to find e.g. 192.0.2.1/24 + when specifying a parent prefix of 192.0.2.0/26. + """ + lookup_name = 'net_host_contained' + + def as_sql(self, qn, connection): + lhs, lhs_params = self.process_lhs(qn, connection) + rhs, rhs_params = self.process_rhs(qn, connection) + params = lhs_params + rhs_params + return 'CAST(HOST(%s) AS INET) << %s' % (lhs, rhs), params + + class NetMaskLength(Transform): lookup_name = 'net_mask_length' function = 'MASKLEN' diff --git a/netbox/ipam/views.py b/netbox/ipam/views.py index 71e261dce..6d2025944 100644 --- a/netbox/ipam/views.py +++ b/netbox/ipam/views.py @@ -403,7 +403,7 @@ def prefix(request, pk): aggregate = None # Count child IP addresses - ipaddress_count = IPAddress.objects.filter(vrf=prefix.vrf, address__net_contained_or_equal=str(prefix.prefix))\ + ipaddress_count = IPAddress.objects.filter(vrf=prefix.vrf, address__net_host_contained=str(prefix.prefix))\ .count() # Parent prefixes table @@ -499,7 +499,7 @@ def prefix_ipaddresses(request, pk): prefix = get_object_or_404(Prefix.objects.all(), pk=pk) # Find all IPAddresses belonging to this Prefix - ipaddresses = IPAddress.objects.filter(vrf=prefix.vrf, address__net_contained_or_equal=str(prefix.prefix))\ + ipaddresses = IPAddress.objects.filter(vrf=prefix.vrf, address__net_host_contained=str(prefix.prefix))\ .select_related('vrf', 'interface__device', 'primary_ip4_for', 'primary_ip6_for') ipaddresses = add_available_ipaddresses(prefix.prefix, ipaddresses, prefix.is_pool)