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