#!/usr/bin/python3

#  Copyright (c) 2016-2016 David Anderson.

#  All rights reserved.

#

#  Redistribution and use in source and binary forms, with or without

#  modification, are permitted provided that the following conditions are met:

#  * Redistributions of source code must retain the above copyright

#    notice, this list of conditions and the following disclaimer.

#  * Redistributions in binary form must reproduce the above copyright

#    notice, this list of conditions and the following disclaimer in the

#    documentation and/or other materials provided with the distribution.

#  * Neither the name of the example nor the

#    names of its contributors may be used to endorse or promote products

#    derived from this software without specific prior written permission.

#

#  THIS SOFTWARE IS PROVIDED BY David Anderson ''AS IS'' AND ANY

#  EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

#  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

#  DISCLAIMED. IN NO EVENT SHALL David Anderson BE LIABLE FOR ANY

#  DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

#  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

#  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

#  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

#  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 

#  OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY 

#  OF SUCH DAMAGE.

import sys

# Use only 
 or 
 all by itself in data.xml.

# No other data on either of such lines.

# All the lines between these two markers should be

# shown in individual lines.

def xmlize(linea,inhtml,inpre):

  outi =  []

  l = linea

  if l.find("
") != -1:

     if inhtml == 'y':

       s2 = '
' +l + '\n'

     else:

       s2 = l + '\n'

     inpre = 'y'

     return s2,inpre

  if l.find("") != -1:

     if inhtml == 'y':

       s2 = l + '\n' + "
"

     else:

       s2 = l + '\n' 

     inpre = 'n'

     return s2, inpre

  if inpre == 'y' and inhtml == 'n':

    outi += ["<preline>"]

  for c in l:

    if c == '<':

       outi += ["<"]

    elif c == '>':

       outi += [">"]

    elif c == "&":

       outi += ["&"]

    #elif c == "'":

    #   outi += ["'"]

    elif c == '"':

       outi += ["""]

    else:

       outi += [c]

  if inpre == 'y' and inhtml == 'n':

    outi += ["</preline>"]

  outi += ["\n"]

  s2 = ''.join(outi)

  return s2,inpre

def paraline(name,linea):

  inpre = 'n'

  out = ''

  if len(linea) <1:

    out = "
" + name + ":"+ "
"

    return out

  out = "
" + name + ": "

  out +=linea 

  out += "
"

  return out;

def paralines(name,lines):

  inpre = 'n'

  if len(lines) <1:

    out = "
" + name + ":"+ "
"

    return out

  out = "
" + name + ": "

  for lin in lines:

     f,inpre = xmlize(lin,'y',inpre)

     out += f

  out += "
"

  return out;

def para(name,str):

  if str ==  None:

      out = "
" + name + ":"+ "
"

  elif len(str) > 0:

      out = "
" + name + ":&nbsp" + str + "
"

  else:

      out = "
" + name + ":"+ "
"

  return out

class bugrecord:

  def __init__(self,dwid):

    self._id= dwid.strip()

    self._cve  = ''

    self._datereported  = ''

    self._reportedby  = ''

    self._vulnerability  = []

    self._product  = ''

    self._description  = []

    self._datefixed  = ''

    self._references  = []

    self._gitfixid  = ''

    self._tarrelease  = ''

  def setcve(self,pubid):

    if self._cve != '':

      print("Duplicate cve ",self._cve,pubid)

      sys.exit(1)

    self._cve  = pubid.strip()

  def setdatereported(self,rep):

    if self._datereported != '':

      print("Duplicate datereported ",self._datereported,rep)

      sys.exit(1)

    self._datereported  = rep.strip()

  def setreportedby(self,rep):

    if self._reportedby != '':

      print("Duplicate reportedby ",self._reportedby,rep)

      sys.exit(1)

    self._reportedby  = rep.strip()

  def setvulnerability(self,vuln):

    if len(self._vulnerability) != 0:

      print("Duplicate vulnerability ",self._vulnerability,vuln)

      sys.exit(1)

    self._vulnerability  = vuln

  def setproduct(self,p):

    if len(self._product) != 0:

      print("Duplicate product ",self._product,p)

      sys.exit(1)

    self._product  = p.strip()

  def setdescription(self,d):

    if len(self._description) != 0:

      print("Duplicate description ",self._description,d)

      sys.exit(1)

    self._description  = d

  def setdatefixed(self,d):

    if len(self._datefixed) != 0:

      print("Duplicate datefixed ",self._datefixed,d)

      sys.exit(1)

    self._datefixed  = d.strip()

  def setreferences(self,r):

    if len(self._references) != 0:

      print("Duplicate references ",self._references,r)

      sys.exit(1)

    self._references  = r

  def setgitfixid(self,g):

    if len(self._gitfixid) != 0:

      print("Duplicate gitfixid ",self._gitfixid,g)

      sys.exit(1)

    self._gitfixid  = g.strip()

  def settarrelease(self,g):

    if len(self._tarrelease) != 0:

      print("Duplicate tarrelease ",self._tarrelease,g)

      sys.exit(1)

    self._tarrelease  = g.strip()

  def plist(self,title,lines):

    if lines == None:

      print(title)

      return

    if len(lines) == 1:

      print(title,lines[0])

      return

    print(title)

    for l in lines:

        print(l)

  def printbug(self):

    print("")

    print("id:",self._id)

    print("cve:",self._cve)

    print("datereported:",self._datereported)

    print("reportedby:",self._reportedby)

    self.plist("vulnerability:",self._vulnerability)

    print("product:",self._product)

    self.plist("description:",self._description)

    print("datefixed:",self._datefixed)

    self.plist("references:",self._references)

    print("gitfixid:",self._gitfixid)

    print("tarrelease:",self._tarrelease)

  def generate_html(self):

    s5= ''.join(self._id)

    t = ''.join(['
',self._id,'
'])

    txt = [t]

    inpre = 'n'

    s,inp= xmlize(self._id,'y',inpre)

    t = paraline("id",s)

    txt += [t]

    s,inp= xmlize(self._cve,'y',inpre)

    t = paraline("cve",s)

    txt += [t]

    s,inp= xmlize(self._datereported,'y',inpre)

    t = paraline("datereported",s)

    txt += [t]

    s,inp= xmlize(self._reportedby,'y',inpre)

    t = paraline("reportedby",s)

    txt += [t]

    #MULTI

    t = paralines("vulnerability",self._vulnerability)

    txt += [t]

    s,inp= xmlize(self._product,'y',inpre)

    t = paraline("product",s)

    txt += [t]

    #MULTI

    t = paralines("description",self._description)

    txt += [t]

    s,inp= xmlize(self._datefixed,'y',inpre)

    t = paraline("datefixed",s)

    txt += [t]

    #MULTI

    t = paralines("references",self._references)

    txt += [t]

    s,inp= xmlize(self._gitfixid,'y',inpre)

    t = paraline("gitfixid",s)

    txt += [t]

    s,inp= xmlize(self._tarrelease,'y',inpre)

    t = paraline("tarrelease",s)

    txt += [t]

    t = '
 [top] 
'

    txt += [t]

    return txt

  def paraxml(self,start,main,term):

    # For single line xml remove the newline from the main text line.

    out = start

    l=main.strip()

    if len(l) > 0:

      out += l 

    out += term + "\n" 

    return out

  def paraxmlN(self,start,main,term):

    # For multi line xml leave newlines present.

    out = start 

    inpre = 'n'

    for x in main:

      l=x.rstrip()

      t,inpre = xmlize(l,'n',inpre);

      if len(t) > 0:

        out += t

    out += term + "\n"

    return out

  def generate_xml(self):

    txt=[]

    t = '<dwbug>'

    txt += [t]

    inpre = 'n'

    s,inpre= xmlize(self._id,'n',inpre)

    s = self.paraxml('<dwid>',s,'</dwid>')

    s,inpre= xmlize(self._cve,'n',inpre)

    t = self.paraxml('<cve>',s,'</cve>')

    txt += [t]

    s,inpre= xmlize(self._datereported,'n',inpre)

    t = self.paraxml('<datereported>',s,'</datereported>')

    txt += [t];

    s,inpre= xmlize(self._reportedby,'n',inpre)

    t = self.paraxml('<reportedby>',s,'</reportedby>')

    txt += [t];

    s,inpre= xmlize(self._product,'n',inpre)

    t = self.paraxml('<product>',s,'</product>')

    txt += [t];

    #MULTI

    p = self._vulnerability

    t = self.paraxmlN("<vulnerability>",p,"</vulnerability>")

    txt += [t]

    #MULTI

    p = self._description

    t = self.paraxmlN("<description>",p,"</description>")

    txt += [t]

    s,inpre= xmlize(self._datefixed,'n',inpre)

    t = self.paraxml('<datefixed>',s,'</datefixed>')

    txt += [t];

    #MULTI

    p = self._references    

    t = self.paraxmlN("<references>",p,"</references>")

    txt += [t]

    s,inpre= xmlize(self._gitfixid,'n',inpre)

    t = self.paraxml('<gitfixid>',s,'</gitfixid>')

    txt += [t];

    s,inpre= xmlize(self._tarrelease,'n',inpre)

    t = self.paraxml('<tarrelease>',s,'</tarrelease>')

    txt += [t];

    t = '</dwbug>'

    txt += [t];

    return txt