# $Id$
# Copyright (C) 2011  SPARTA, Inc. dba Cobham Analytic Solutions
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND SPARTA DISCLAIMS ALL WARRANTIES WITH
# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS.  IN NO EVENT SHALL SPARTA BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE.
#

default_logfile = '/var/rcynic/data/summary.xml'
default_root = '/var/rcynic/data'

import time, vobject
from rpki.gui.cacheview import models
from rpki.rcynic import rcynic_xml_iterator, label_iterator
from rpki.sundial import datetime
from django.db import transaction
import django.db.models

debug = False
fam_map = { 'roa_prefix_set_ipv6': 6, 'roa_prefix_set_ipv4': 4 }

class rcynic_object(object):

    def __call__(self, vs):
        """
        do initial processing on a rcynic_object instance.

        return value is a tuple: first element is a boolean value indicating whether
        the object is changed/new since the last time we processed it.  second
        element is the db instance.
        """
        if debug:
            print 'processing %s at %s' % (vs.file_class.__name__, vs.uri)

        # rcynic will generation <validation_status/> elements for objects
        # listed in the manifest but not found on disk
        if os.path.exists(vs.filename):
            q = self.model_class.objects.filter(uri=vs.uri)
            if not q:
                if debug:
                    print 'creating new db instance'
                inst = self.model_class(uri=vs.uri)
            else:
                inst = q[0]

            # determine if the object is changed/new
            mtime = os.stat(vs.filename)[8]
            if mtime != inst.mtime:
                inst.mtime = mtime
                obj = vs.obj # causes object to be lazily loaded
                inst.not_before = obj.notBefore.to_sql()
                inst.not_after = obj.notAfter.to_sql()
                if debug:
                    sys.stderr.write('name=%s ski=%s\n' % (obj.subject, obj.ski))
                inst.name = obj.subject
                inst.keyid = obj.ski

                # look up signing cert
                if obj.issuer == obj.subject:
                    # self-signed cert (TA)
                    inst.cert = inst
                else:
                    q = models.Cert.objects.filter(keyid=obj.aki, name=obj.issuer)
                    if q:
                        inst.issuer = q[0]
                    else:
                        sys.stderr.write('warning: unable to find signing cert with ski=%s (%s)\n' % (obj.aki, obj.issuer))
                        return None

                self.callback(obj, inst)
            else:
                if debug:
                    print 'object is unchanged'

            # save required to create new ValidationStatus object refering to it
            inst.save()
            inst.statuses.create(generation=models.generations_dict[vs.generation] if vs.generation else None,
                    timestamp=datetime.fromXMLtime(vs.timestamp).to_sql(),
                    status=models.ValidationLabel.objects.get(label=vs.status))

            return inst
        else:
            if debug:
                print 'ERROR - file is missing: %s' % vs.filename

        return True

class rcynic_cert(rcynic_object):
    model_class = models.Cert

    def callback(self, cert, obj):
        """
        Process a RPKI resource certificate.
        """

        obj.sia = cert.sia_directory_uri
        obj.save()

        # resources can change when a cert is updated
        obj.asns.clear()
        obj.addresses.clear()

        for asr in cert.resources.asn:
            if debug:
                sys.stderr.write('processing %s\n' % asr)

            attrs = { 'min': asr.min, 'max': asr.max }
            q = models.ASRange.objects.filter(**attrs)
            if not q:
                obj.asns.create(**attrs)
            else:
                obj.asns.add(q[0])

        for family, addrset in (4, cert.resources.v4), (6, cert.resources.v6):
            for rng in addrset:
                if debug:
                    sys.stderr.write('processing %s\n' % rng)

                attrs = { 'family': family, 'min': str(rng.min), 'max': str(rng.max) }
                q = models.AddressRange.objects.filter(**attrs)
                if not q:
                    obj.addresses.create(**attrs)
                else:
                    obj.addresses.add(q[0])

        if debug:
            print 'finished processing rescert at %s' % cert.uri

class rcynic_roa(rcynic_object):
    model_class = models.ROA

    def callback(self, roa, obj):
        obj.asid = roa.asID
        obj.save()
        obj.prefixes.clear()
        for pfxset in roa.prefix_sets:
            family = fam_map[pfxset.__class__.__name__]
            for pfx in pfxset:
                attrs = { 'family' : family,
                          'prefix': str(pfx.prefix),
                          'bits' : pfx.prefixlen,
                          'max_length': pfx.max_prefixlen }
                q = models.ROAPrefix.objects.filter(**attrs)
                if not q:
                    obj.prefixes.create(**attrs)
                else:
                    obj.prefixes.add(q[0])

class rcynic_gbr(rcynic_object):
    model_class = models.Ghostbuster

    def callback(self, gbr, obj):
        vcard = vobject.readOne(gbr.vcard)
        if debug:
            vcard.prettyPrint()
        obj.full_name = vcard.fn.value if hasattr(vcard, 'fn') else None
        obj.email_address = vcard.email.value if hasattr(vcard, 'email') else None
        obj.telephone = vcard.tel.value if hasattr(vcard, 'tel') else None
        obj.organization = vcard.org.value[0] if hasattr(vcard, 'org') else None

def process_cache(root, xml_file):
    start = time.time()

    dispatch = {
      'rcynic_certificate': rcynic_cert(),
      'rcynic_roa'        : rcynic_roa(),
      'rcynic_ghostbuster': rcynic_gbr()
    }

    # remove all existing ValidationStatus_* entries
    models.ValidationStatus_Cert.objects.all().delete()
    models.ValidationStatus_ROA.objects.all().delete()
    models.ValidationStatus_Ghostbuster.objects.all().delete()

    # loop over all rcynic objects and dispatch based on the returned
    # rcynic_object subclass
    n = 1
    defer = rcynic_xml_iterator(root, xml_file)
    while defer:
        if debug:
            print 'starting iteration %d for deferred objects' % n
            n = n + 1

        elts = defer
        defer = []
        for vs in elts:
            # need to defer processing this object, most likely because
            # the <validation_status/> element for the signing cert hasn't
            # been seen yet
            if not dispatch[vs.file_class.__name__](vs):
                defer.append(vs)

    # garbage collection
    # remove all objects which have no ValidationStatus references, which
    # means they did not appear in the last XML output
    if debug:
        print 'performing garbage collection'

    # trying to .delete() the querysets directly results in a "too many sql variables" exception
    for qs in (models.Cert.objects.annotate(num_statuses=django.db.models.Count('statuses')).filter(num_statuses=0),
            models.Ghostbuster.objects.annotate(num_statuses=django.db.models.Count('statuses')).filter(num_statuses=0),
            models.ROA.objects.annotate(num_statuses=django.db.models.Count('statuses')).filter(num_statuses=0)):
        for e in qs:
            e.delete()

    if debug:
        stop = time.time()
        sys.stdout.write('elapsed time %d seconds.\n' % (stop - start))

def process_labels(xml_file):
    if debug:
        sys.stderr.write('updating labels...\n')

    for label, kind, desc in label_iterator(xml_file):
        if debug:
            sys.stderr.write('label=%s kind=%s desc=%s\n' % (label, kind, desc))
        if kind:
            q = models.ValidationLabel.objects.filter(label=label)
            if not q:
                obj = models.ValidationLabel(label=label)
            else:
                obj = q[0]

            obj.kind = models.kinds_dict[kind]
            obj.status = desc
            obj.save()

if __name__ == '__main__':
    import optparse

    parser = optparse.OptionParser()
    parser.add_option("-d", "--debug", action="store_true",
            help="enable debugging message")
    parser.add_option("-f", "--file", dest="logfile",
            help="specify the rcynic XML file to parse [default: %default]",
            default=default_logfile)
    parser.add_option("-r", "--root",
            help="specify the chroot directory for the rcynic jail [default: %default]",
            metavar="DIR", default=default_root)
    options, args = parser.parse_args(sys.argv)
    if options.debug:
        debug = True

    with transaction.commit_on_success():
        process_labels(options.logfile)
        process_cache(options.root, options.logfile)

# vim:sw=4 ts=8