# Django Conntrackt is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

# details.

#

# You should have received a copy of the GNU General Public License along with

# Django Conntrackt.  If not, see <http://www.gnu.org/licenses/>.

#

# Django imports.

from django.contrib import admin

# Application imports.

from .models import Project, Location, Entity, Interface, Communication

class InterfaceInline(admin.StackedInline):

    """

    This class implements the inline admin view of the Interface instances. This

    is used when adding the entities (since it's easier to specify interface for

    an entity right away).

    Properties:

# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

# details.

#

# You should have received a copy of the GNU General Public License along with

# Django Conntrackt.  If not, see <http://www.gnu.org/licenses/>.

#

# Django imports.

from django.contrib.admin.utils import NestedObjects

from django.core.exceptions import ValidationError

from django.db import models

from django.db.models.query_utils import Q

# Application imports.

from .utils import list_formatter_callback, get_distinct_colors

class SearchManager(models.Manager):

    """

    Custom model manager that implements search for model instances that contain

    a specific string (search term) in fields "name" or "description".

    """

    Fields:

      name - String denoting the entity name.

      description - Free-form description of an entity.

      project - Foreign key pointing to the project to which the entity

      belongs.

      location - Foreign key pointing to the location at which the entity is

      located.

    """

    name = models.CharField(max_length=100)

    description = models.TextField(blank=True)

    objects = SearchManager()

    class Meta:

        # Fix the plural form used by Django.

        verbose_name_plural = 'entities'

        # Enforce uniqueness of entity name in a project.

        unique_together = ("name", "project")

    def __unicode__(self):

        """

        Returns:

          String representation of an entity. This identifier contains name of

      description - Free-form description of an interface.

      entity - Foreign key pointing to the entity to which the interface

      belongs.

      address - IP address of an interface. It's possible to store network

      address in it as well.

      netmask - Netmask of the interface. By default this is /32

      (255.255.255.255), but in case of subnet entities this can be used for

      denoting the network netmask.

    """

    name = models.CharField(max_length=100, default='eth0')

    description = models.TextField(blank=True, default='Main network interface.')

    address = models.GenericIPAddressField()

    netmask = models.GenericIPAddressField(default='255.255.255.255')

    class Meta:

        # Enforce uniqueness of interface name in an entity. Enforce uniqueness

        # of IP address in a subnet for an entity.

        unique_together = (("name", "entity"),

                           ("entity", "address", "netmask"),)

    def __unicode__(self):

        """

        Returns:

      type (in numeric form).

      description - Free-form text that can be used to describe the

      communication. This is also used when generating the iptables rules for

      documenting the rules.

    """

    PROTOCOL_CHOICES = (

        ('TCP', 'TCP'),

        ('UDP', 'UDP'),

        ('ICMP', 'ICMP'),

        )

    protocol = models.CharField(max_length=10, choices=PROTOCOL_CHOICES)

    port = models.IntegerField(default=0)

    description = models.TextField(blank=True)

    class Meta:

        # Enforce uniqueness of communication.

        unique_together = ("source", "destination", "protocol", "port")

    def __unicode__(self):

        """

        Returns:

          String representation of an interface. This involves showing the

{% extends "admin/change_list.html" %}

{% load i18n admin_static admin_list %}

{% load admin_urls %}

{% block object-tools-items %}

            <li>

              {# Add the GET parameters from the admin's filter to 'Add entity' button #}

              {# so we can perform some filtering on source/destination interfaces. #}

              <a href="{% url cl.opts|admin_urlname:'add' %}?{% if is_popup %}_popup=1{% endif %}{% if 'location' in cl.params %}&amp;location={{ cl.params.location }}{% endif %}{% if 'project' in cl.params %}&amp;project={{ cl.params.project }}{% endif %}" class="addlink">

                {% blocktrans with cl.opts.verbose_name as name %}Add {{ name }}{% endblocktrans %}

              </a>

            </li>

{% endblock %}

# Django Conntrackt is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

# details.

#

# You should have received a copy of the GNU General Public License along with

# Django Conntrackt.  If not, see <http://www.gnu.org/licenses/>.

#

# Django imports.

from django import template

from django.utils.html import format_html

# Get an instance of Django's template library.

register = template.Library()

@register.simple_tag(takes_context=False)

def html_link(text, view, *args, **kwargs):

    """

    A small wrapper for showing HTML links.

        get - Additional GET parameter that should be appended to the URL.

    """

    # Verify the passed-in keyword arguments first.

    for key in kwargs.keys():

        if key not in ("get", "class", "title", "id"):

            raise template.TemplateSyntaxError("Unknown argument for 'html_link' tag: %r" % key)

    # Generate the URL by using the supplied view name and arguments that should

    # be passed to the view.

    # Set-up the base pattern (url, parameters, text).

    if 'get' in kwargs:

        pattern = '<a href="{url}?{get}"'

    else:

        pattern = '<a href="{url}"'

    if 'class' in kwargs:

        pattern += ' class="{class}"'

    if 'title' in kwargs:

        pattern += ' title="{title}"'

    Arguments:

      - context - Context of the view being rendered.

      - url_name - Name of the URL against which the context request path is

      being checked.

    """

    # Assume that we have not been able to resolve the request path to an URL.

    resolved = False

    try:

        # Use the request path, and resolve it to a URL name.

        # This means we haven't been able to resolve the path from request.

        pass

    # If the request was resolved and URL names match, verify that the kwargs

    # match as well.

    matches = resolved and resolved.url_name == url_name

    if matches and kwargs:

        for key in kwargs:

            kwarg = kwargs.get(key)

            resolved_kwarg = resolved.kwargs.get(key)

            if not resolved_kwarg or kwarg != resolved_kwarg:

                return False

# Python third-party library imports.

import mock

# Django imports.

from django.template import Context, Template, TemplateSyntaxError

from django.test import TestCase

# Application imports

from conntrackt.templatetags.conntrackt_tags import html_link, active_link, current_url_equals

class HtmlLinkTest(TestCase):

    def test_url_reverse_called_with_passed_in_args(self, mock_reverse):

        """

        Tests if URL reversing is performed using the correct set of

        passed-in arguments.

        """

        html_link("My link", "my_view", 'arg1', 'arg2', 'arg3')

        mock_reverse.assert_called_once_with("my_view", args=('arg1', 'arg2', 'arg3'))

# Standard library imports.

import json

from StringIO import StringIO

from zipfile import ZipFile, ZIP_DEFLATED

# Python third-party library imports.

import mock

# Django imports.

from django.core.exceptions import ValidationError

from django.http import Http404

from django.test import RequestFactory

from django.test import TestCase

from django.utils.http import urlquote

# Application imports

from conntrackt.models import Project, Location, Entity, Interface, Communication

from conntrackt.views import IndexView, SearchView, APISearchView

from conntrackt.views import entity_iptables, project_iptables, project_diagram

from conntrackt.views import ProjectView, ProjectCreateView, ProjectUpdateView, ProjectDeleteView

#

# Standard library imports.

from StringIO import StringIO

from zipfile import ZipFile, ZIP_DEFLATED

import json

# Django imports.

from django.contrib.auth.decorators import permission_required

from django.contrib import messages

from django.core.exceptions import ValidationError

from django.db.models import Q

from django.db.models.deletion import Collector

from django.http import HttpResponse

from django.shortcuts import render_to_response, get_object_or_404

from django.views.generic import TemplateView, DetailView, CreateView, UpdateView, DeleteView, View

# Third-party application imports.

from braces.views import MultiplePermissionsRequiredMixin, SetHeadlineMixin

# Application imports.

from .forms import ProjectForm, LocationForm, EntityForm, InterfaceForm, CommunicationForm

from .models import Project, Entity, Location, Interface, Communication

from .utils import generate_entity_iptables, generate_project_diagram

class RedirectToNextMixin(object):

# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

# details.

#

# You should have received a copy of the GNU General Public License along with

# Django Conntrackt.  If not, see <http://www.gnu.org/licenses/>.

#

# Convenience mixins for Django (for authentication etc).

django-braces~=1.12.0

# Convenience tools for controlling rendering of forms while embracing DRY.

# Web framework used by application.

# Library for programatic calculation of colours (contrasts,

# inversions etc).

palette~=0.2.0

# Interaface towards Graphviz for chart generation.

pydot~=1.2.0

#

# This file is autogenerated by pip-compile

# To update, run:

#

#    pip-compile --output-file requirements/development.txt requirements/development.in

#

alabaster==0.7.10         # via sphinx

babel==2.5.1              # via sphinx

certifi==2017.11.5        # via requests

chardet==3.0.4            # via requests

coverage==4.4.2

django-braces==1.12.0

docutils==0.14            # via sphinx

factory-boy==2.1.2

funcsigs==1.0.2           # via mock

idna==2.6                 # via requests

imagesize==0.7.1          # via sphinx

jinja2==2.10              # via sphinx

markupsafe==1.0           # via jinja2

mock==1.3.0

palette==0.2

pbr==3.1.1                # via mock

pydot==1.2.3

pygments==2.2.0           # via sphinx

#

# This file is autogenerated by pip-compile

# To update, run:

#

#    pip-compile --output-file requirements/test.txt requirements/test.in

#

alabaster==0.7.10         # via sphinx

babel==2.5.1              # via sphinx

certifi==2017.11.5        # via requests

chardet==3.0.4            # via requests

coverage==4.4.2

django-braces==1.12.0

docutils==0.14            # via sphinx

factory-boy==2.1.2

funcsigs==1.0.2           # via mock

idna==2.6                 # via requests

imagesize==0.7.1          # via sphinx

jinja2==2.10              # via sphinx

markupsafe==1.0           # via jinja2

mock==1.3.0

palette==0.2

pbr==3.1.1                # via mock

pydot==1.2.3

pygments==2.2.0           # via sphinx

#

# You should have received a copy of the GNU General Public License along with

# Django Conntrackt.  If not, see <http://www.gnu.org/licenses/>.

#

import os

from setuptools import setup, find_packages

README = open(os.path.join(os.path.dirname(__file__), 'README.rst')).read()

INSTALL_REQUIREMENTS = [

    "django-braces~=1.12.0",

    "palette~=0.2.0",

    "pydot~=1.2.0",

]

# allow setup.py to be run from any path

os.chdir(os.path.normpath(os.path.join(os.path.abspath(__file__), os.pardir)))

setup(

    name='django-conntrackt',

    version='dev',

    packages=find_packages(exclude=["testproject", "testproject.*"]),

    include_package_data=True,