summaryrefslogtreecommitdiff
path: root/site_scons/site_tools/xml-external-entity-scanner.py
blob: 8a44ca155d157f009f586f79b62385c2dafb717b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# Copyright (C) 2010 Daniel Burrows
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program 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 this program; see the file COPYING.  If not, write to the
# Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
# MA 02111-1307, USA.

import errno
import xml.parsers.expat
from SCons.Script import File, Scanner

def exists():
    return True

def generate(env):
    env.Append(SCANNERS = XMLExternalEntitiesScanner)

def XMLExternalEntities(node, env, path):

    result = []

    class Finished:
        pass

    def entity_decl(entityName,
                    is_parameter_entity,
                    value,
                    base,
                    systemId,
                    publicId,
                    notationName):
        if value is None:
            result.append(systemId)

    def start_element(name, attrs):
        raise Finished()

    try:
        infile = file(node.path)
    except EnvironmentError, e:
        if e.errno == errno.ENOENT:
            return []
        else:
            raise

    try:
        parser = xml.parsers.expat.ParserCreate()
        parser.StartElementHandler = start_element
        parser.EntityDeclHandler = entity_decl

        parser.ParseFile(infile)
    except Finished:
        pass
    except xml.parsers.expat.ExpatError, e:
        print e
        # Probably an XML syntax error -- don't blow up the scanner
        # for that!
        pass
    finally:
        infile.close()

    return result

XMLExternalEntitiesScanner = Scanner(function = XMLExternalEntities,
                                     skeys = [ '.xml' ],
                                     recursive = True,
                                     )