blob: f523aa68a36b397b3c6f12998ec597c28e7ad69a [file] [log] [blame]
Mario Six78a88f72018-07-10 08:40:17 +02001#!/usr/bin/env python3
2# -*- coding: utf-8; mode: python -*-
3# pylint: disable=R0903, C0330, R0914, R0912, E0401
4
5u"""
6 kernel-include
7 ~~~~~~~~~~~~~~
8
9 Implementation of the ``kernel-include`` reST-directive.
10
11 :copyright: Copyright (C) 2016 Markus Heiser
12 :license: GPL Version 2, June 1991 see linux/COPYING for details.
13
14 The ``kernel-include`` reST-directive is a replacement for the ``include``
15 directive. The ``kernel-include`` directive expand environment variables in
16 the path name and allows to include files from arbitrary locations.
17
18 .. hint::
19
20 Including files from arbitrary locations (e.g. from ``/etc``) is a
21 security risk for builders. This is why the ``include`` directive from
22 docutils *prohibit* pathnames pointing to locations *above* the filesystem
23 tree where the reST document with the include directive is placed.
24
25 Substrings of the form $name or ${name} are replaced by the value of
26 environment variable name. Malformed variable names and references to
27 non-existing variables are left unchanged.
28"""
29
30# ==============================================================================
31# imports
32# ==============================================================================
33
34import os.path
35
36from docutils import io, nodes, statemachine
37from docutils.utils.error_reporting import SafeString, ErrorString
38from docutils.parsers.rst import directives
39from docutils.parsers.rst.directives.body import CodeBlock, NumberLines
40from docutils.parsers.rst.directives.misc import Include
41
42__version__ = '1.0'
43
44# ==============================================================================
45def setup(app):
46# ==============================================================================
47
48 app.add_directive("kernel-include", KernelInclude)
49 return dict(
50 version = __version__,
51 parallel_read_safe = True,
52 parallel_write_safe = True
53 )
54
55# ==============================================================================
56class KernelInclude(Include):
57# ==============================================================================
58
59 u"""KernelInclude (``kernel-include``) directive"""
60
61 def run(self):
62 path = os.path.realpath(
63 os.path.expandvars(self.arguments[0]))
64
65 # to get a bit security back, prohibit /etc:
66 if path.startswith(os.sep + "etc"):
67 raise self.severe(
68 'Problems with "%s" directive, prohibited path: %s'
69 % (self.name, path))
70
71 self.arguments[0] = path
72
73 #return super(KernelInclude, self).run() # won't work, see HINTs in _run()
74 return self._run()
75
76 def _run(self):
77 """Include a file as part of the content of this reST file."""
78
79 # HINT: I had to copy&paste the whole Include.run method. I'am not happy
80 # with this, but due to security reasons, the Include.run method does
81 # not allow absolute or relative pathnames pointing to locations *above*
82 # the filesystem tree where the reST document is placed.
83
84 if not self.state.document.settings.file_insertion_enabled:
85 raise self.warning('"%s" directive disabled.' % self.name)
86 source = self.state_machine.input_lines.source(
87 self.lineno - self.state_machine.input_offset - 1)
88 source_dir = os.path.dirname(os.path.abspath(source))
89 path = directives.path(self.arguments[0])
90 if path.startswith('<') and path.endswith('>'):
91 path = os.path.join(self.standard_include_path, path[1:-1])
92 path = os.path.normpath(os.path.join(source_dir, path))
93
94 # HINT: this is the only line I had to change / commented out:
95 #path = utils.relative_path(None, path)
96
97 path = nodes.reprunicode(path)
98 encoding = self.options.get(
99 'encoding', self.state.document.settings.input_encoding)
100 e_handler=self.state.document.settings.input_encoding_error_handler
101 tab_width = self.options.get(
102 'tab-width', self.state.document.settings.tab_width)
103 try:
104 self.state.document.settings.record_dependencies.add(path)
105 include_file = io.FileInput(source_path=path,
106 encoding=encoding,
107 error_handler=e_handler)
108 except UnicodeEncodeError as error:
109 raise self.severe('Problems with "%s" directive path:\n'
110 'Cannot encode input file path "%s" '
111 '(wrong locale?).' %
112 (self.name, SafeString(path)))
113 except IOError as error:
114 raise self.severe('Problems with "%s" directive path:\n%s.' %
115 (self.name, ErrorString(error)))
116 startline = self.options.get('start-line', None)
117 endline = self.options.get('end-line', None)
118 try:
119 if startline or (endline is not None):
120 lines = include_file.readlines()
121 rawtext = ''.join(lines[startline:endline])
122 else:
123 rawtext = include_file.read()
124 except UnicodeError as error:
125 raise self.severe('Problem with "%s" directive:\n%s' %
126 (self.name, ErrorString(error)))
127 # start-after/end-before: no restrictions on newlines in match-text,
128 # and no restrictions on matching inside lines vs. line boundaries
129 after_text = self.options.get('start-after', None)
130 if after_text:
131 # skip content in rawtext before *and incl.* a matching text
132 after_index = rawtext.find(after_text)
133 if after_index < 0:
134 raise self.severe('Problem with "start-after" option of "%s" '
135 'directive:\nText not found.' % self.name)
136 rawtext = rawtext[after_index + len(after_text):]
137 before_text = self.options.get('end-before', None)
138 if before_text:
139 # skip content in rawtext after *and incl.* a matching text
140 before_index = rawtext.find(before_text)
141 if before_index < 0:
142 raise self.severe('Problem with "end-before" option of "%s" '
143 'directive:\nText not found.' % self.name)
144 rawtext = rawtext[:before_index]
145
146 include_lines = statemachine.string2lines(rawtext, tab_width,
147 convert_whitespace=True)
148 if 'literal' in self.options:
149 # Convert tabs to spaces, if `tab_width` is positive.
150 if tab_width >= 0:
151 text = rawtext.expandtabs(tab_width)
152 else:
153 text = rawtext
154 literal_block = nodes.literal_block(rawtext, source=path,
155 classes=self.options.get('class', []))
156 literal_block.line = 1
157 self.add_name(literal_block)
158 if 'number-lines' in self.options:
159 try:
160 startline = int(self.options['number-lines'] or 1)
161 except ValueError:
162 raise self.error(':number-lines: with non-integer '
163 'start value')
164 endline = startline + len(include_lines)
165 if text.endswith('\n'):
166 text = text[:-1]
167 tokens = NumberLines([([], text)], startline, endline)
168 for classes, value in tokens:
169 if classes:
170 literal_block += nodes.inline(value, value,
171 classes=classes)
172 else:
173 literal_block += nodes.Text(value, value)
174 else:
175 literal_block += nodes.Text(text, text)
176 return [literal_block]
177 if 'code' in self.options:
178 self.options['source'] = path
179 codeblock = CodeBlock(self.name,
180 [self.options.pop('code')], # arguments
181 self.options,
182 include_lines, # content
183 self.lineno,
184 self.content_offset,
185 self.block_text,
186 self.state,
187 self.state_machine)
188 return codeblock.run()
189 self.state_machine.insert_input(include_lines, path)
190 return []