| 1 | |
| 2 | |
| 3 | |
| 4 | |
| 5 | |
| 6 |
|
| 7 | |
| 8 | from __future__ import print_function
|
| 9 |
|
| 10 | import argparse
|
| 11 | import re
|
| 12 | import sys, os
|
| 13 |
|
| 14 | class NoHelperFound(BaseException):
|
| 15 | pass
|
| 16 |
|
| 17 | class NoSyscallCommandFound(BaseException):
|
| 18 | pass
|
| 19 |
|
| 20 | class ParsingError(BaseException):
|
| 21 | def __init__(self, line='<line not provided>', reader=None):
|
| 22 | if reader:
|
| 23 | BaseException.__init__(self,
|
| 24 | 'Error at file offset %d, parsing line: %s' %
|
| 25 | (reader.tell(), line))
|
| 26 | else:
|
| 27 | BaseException.__init__(self, 'Error parsing line: %s' % line)
|
| 28 |
|
| 29 |
|
| 30 | class APIElement(object):
|
| 31 | |
| 32 | An object representing the description of an aspect of the eBPF API.
|
| 33 | @proto: prototype of the API symbol
|
| 34 | @desc: textual description of the symbol
|
| 35 | @ret: (optional) description of any associated return value
|
| 36 |
|
| 37 | def __init__(self, proto='', desc='', ret=''):
|
| 38 | self.proto = proto
|
| 39 | self.desc = desc
|
| 40 | self.ret = ret
|
| 41 |
|
| 42 |
|
| 43 | class Helper(APIElement):
|
| 44 | |
| 45 | An object representing the description of an eBPF helper function.
|
| 46 | @proto: function prototype of the helper function
|
| 47 | @desc: textual description of the helper function
|
| 48 | @ret: description of the return value of the helper function
|
| 49 |
|
| 50 | def proto_break_down(self):
|
| 51 | |
| 52 | Break down helper function protocol into smaller chunks: return type,
|
| 53 | name, distincts arguments.
|
| 54 |
|
| 55 | arg_re = re.compile('((\w+ )*?(\w+|...))( (\**)(\w+))?$')
|
| 56 | res = {}
|
| 57 | proto_re = re.compile('(.+) (\**)(\w+)\(((([^,]+)(, )?){1,5})\)$')
|
| 58 |
|
| 59 | capture = proto_re.match(self.proto)
|
| 60 | res['ret_type'] = capture.group(1)
|
| 61 | res['ret_star'] = capture.group(2)
|
| 62 | res['name'] = capture.group(3)
|
| 63 | res['args'] = []
|
| 64 |
|
| 65 | args = capture.group(4).split(', ')
|
| 66 | for a in args:
|
| 67 | capture = arg_re.match(a)
|
| 68 | res['args'].append({
|
| 69 | 'type' : capture.group(1),
|
| 70 | 'star' : capture.group(5),
|
| 71 | 'name' : capture.group(6)
|
| 72 | })
|
| 73 |
|
| 74 | return res
|
| 75 |
|
| 76 |
|
| 77 | class HeaderParser(object):
|
| 78 | |
| 79 | An object used to parse a file in order to extract the documentation of a
|
| 80 | list of eBPF helper functions. All the helpers that can be retrieved are
|
| 81 | stored as Helper object, in the self.helpers() array.
|
| 82 | @filename: name of file to parse, usually include/uapi/linux/bpf.h in the
|
| 83 | kernel tree
|
| 84 |
|
| 85 | def __init__(self, filename):
|
| 86 | self.reader = open(filename, 'r')
|
| 87 | self.line = ''
|
| 88 | self.helpers = []
|
| 89 | self.commands = []
|
| 90 |
|
| 91 | def parse_element(self):
|
| 92 | proto = self.parse_symbol()
|
| 93 | desc = self.parse_desc()
|
| 94 | ret = self.parse_ret()
|
| 95 | return APIElement(proto=proto, desc=desc, ret=ret)
|
| 96 |
|
| 97 | def parse_helper(self):
|
| 98 | proto = self.parse_proto()
|
| 99 | desc = self.parse_desc()
|
| 100 | ret = self.parse_ret()
|
| 101 | return Helper(proto=proto, desc=desc, ret=ret)
|
| 102 |
|
| 103 | def parse_symbol(self):
|
| 104 | p = re.compile(' \* ?(.+)$')
|
| 105 | capture = p.match(self.line)
|
| 106 | if not capture:
|
| 107 | raise NoSyscallCommandFound
|
| 108 | end_re = re.compile(' \* ?NOTES$')
|
| 109 | end = end_re.match(self.line)
|
| 110 | if end:
|
| 111 | raise NoSyscallCommandFound
|
| 112 | self.line = self.reader.readline()
|
| 113 | return capture.group(1)
|
| 114 |
|
| 115 | def parse_proto(self):
|
| 116 | |
| 117 | |
| 118 | |
| 119 | |
| 120 | |
| 121 | |
| 122 | |
| 123 | p = re.compile(' \* ?((.+) \**\w+\((((const )?(struct )?(\w+|\.\.\.)( \**\w+)?)(, )?){1,5}\))$')
|
| 124 | capture = p.match(self.line)
|
| 125 | if not capture:
|
| 126 | raise NoHelperFound
|
| 127 | self.line = self.reader.readline()
|
| 128 | return capture.group(1)
|
| 129 |
|
| 130 | def parse_desc(self):
|
| 131 | p = re.compile(' \* ?(?:\t| {5,8})Description$')
|
| 132 | capture = p.match(self.line)
|
| 133 | if not capture:
|
| 134 | |
| 135 | |
| 136 | return ''
|
| 137 | |
| 138 | |
| 139 | desc = ''
|
| 140 | while True:
|
| 141 | self.line = self.reader.readline()
|
| 142 | if self.line == ' *\n':
|
| 143 | desc += '\n'
|
| 144 | else:
|
| 145 | p = re.compile(' \* ?(?:\t| {5,8})(?:\t| {8})(.*)')
|
| 146 | capture = p.match(self.line)
|
| 147 | if capture:
|
| 148 | desc += capture.group(1) + '\n'
|
| 149 | else:
|
| 150 | break
|
| 151 | return desc
|
| 152 |
|
| 153 | def parse_ret(self):
|
| 154 | p = re.compile(' \* ?(?:\t| {5,8})Return$')
|
| 155 | capture = p.match(self.line)
|
| 156 | if not capture:
|
| 157 | |
| 158 | |
| 159 | return ''
|
| 160 | |
| 161 | |
| 162 | ret = ''
|
| 163 | while True:
|
| 164 | self.line = self.reader.readline()
|
| 165 | if self.line == ' *\n':
|
| 166 | ret += '\n'
|
| 167 | else:
|
| 168 | p = re.compile(' \* ?(?:\t| {5,8})(?:\t| {8})(.*)')
|
| 169 | capture = p.match(self.line)
|
| 170 | if capture:
|
| 171 | ret += capture.group(1) + '\n'
|
| 172 | else:
|
| 173 | break
|
| 174 | return ret
|
| 175 |
|
| 176 | def seek_to(self, target, help_message):
|
| 177 | self.reader.seek(0)
|
| 178 | offset = self.reader.read().find(target)
|
| 179 | if offset == -1:
|
| 180 | raise Exception(help_message)
|
| 181 | self.reader.seek(offset)
|
| 182 | self.reader.readline()
|
| 183 | self.reader.readline()
|
| 184 | self.line = self.reader.readline()
|
| 185 |
|
| 186 | def parse_syscall(self):
|
| 187 | self.seek_to('* DOC: eBPF Syscall Commands',
|
| 188 | 'Could not find start of eBPF syscall descriptions list')
|
| 189 | while True:
|
| 190 | try:
|
| 191 | command = self.parse_element()
|
| 192 | self.commands.append(command)
|
| 193 | except NoSyscallCommandFound:
|
| 194 | break
|
| 195 |
|
| 196 | def parse_helpers(self):
|
| 197 | self.seek_to('* Start of BPF helper function descriptions:',
|
| 198 | 'Could not find start of eBPF helper descriptions list')
|
| 199 | while True:
|
| 200 | try:
|
| 201 | helper = self.parse_helper()
|
| 202 | self.helpers.append(helper)
|
| 203 | except NoHelperFound:
|
| 204 | break
|
| 205 |
|
| 206 | def run(self):
|
| 207 | self.parse_syscall()
|
| 208 | self.parse_helpers()
|
| 209 | self.reader.close()
|
| 210 |
|
| 211 | |
| 212 |
|
| 213 | class Printer(object):
|
| 214 | |
| 215 | A generic class for printers. Printers should be created with an array of
|
| 216 | Helper objects, and implement a way to print them in the desired fashion.
|
| 217 | @parser: A HeaderParser with objects to print to standard output
|
| 218 |
|
| 219 | def __init__(self, parser):
|
| 220 | self.parser = parser
|
| 221 | self.elements = []
|
| 222 |
|
| 223 | def print_header(self):
|
| 224 | pass
|
| 225 |
|
| 226 | def print_footer(self):
|
| 227 | pass
|
| 228 |
|
| 229 | def print_one(self, helper):
|
| 230 | pass
|
| 231 |
|
| 232 | def print_all(self):
|
| 233 | self.print_header()
|
| 234 | for elem in self.elements:
|
| 235 | self.print_one(elem)
|
| 236 | self.print_footer()
|
| 237 |
|
| 238 |
|
| 239 | class PrinterRST(Printer):
|
| 240 | |
| 241 | A generic class for printers that print ReStructured Text. Printers should
|
| 242 | be created with a HeaderParser object, and implement a way to print API
|
| 243 | elements in the desired fashion.
|
| 244 | @parser: A HeaderParser with objects to print to standard output
|
| 245 |
|
| 246 | def __init__(self, parser):
|
| 247 | self.parser = parser
|
| 248 |
|
| 249 | def print_license(self):
|
| 250 | license = '''\
|
| 251 | .. Copyright (C) All BPF authors and contributors from 2014 to present.
|
| 252 | .. See git log include/uapi/linux/bpf.h in kernel tree for details.
|
| 253 | ..
|
| 254 | .. %%%LICENSE_START(VERBATIM)
|
| 255 | .. Permission is granted to make and distribute verbatim copies of this
|
| 256 | .. manual provided the copyright notice and this permission notice are
|
| 257 | .. preserved on all copies.
|
| 258 | ..
|
| 259 | .. Permission is granted to copy and distribute modified versions of this
|
| 260 | .. manual under the conditions for verbatim copying, provided that the
|
| 261 | .. entire resulting derived work is distributed under the terms of a
|
| 262 | .. permission notice identical to this one.
|
| 263 | ..
|
| 264 | .. Since the Linux kernel and libraries are constantly changing, this
|
| 265 | .. manual page may be incorrect or out-of-date. The author(s) assume no
|
| 266 | .. responsibility for errors or omissions, or for damages resulting from
|
| 267 | .. the use of the information contained herein. The author(s) may not
|
| 268 | .. have taken the same level of care in the production of this manual,
|
| 269 | .. which is licensed free of charge, as they might when working
|
| 270 | .. professionally.
|
| 271 | ..
|
| 272 | .. Formatted or processed versions of this manual, if unaccompanied by
|
| 273 | .. the source, must acknowledge the copyright and authors of this work.
|
| 274 | .. %%%LICENSE_END
|
| 275 | ..
|
| 276 | .. Please do not edit this file. It was generated from the documentation
|
| 277 | .. located in file include/uapi/linux/bpf.h of the Linux kernel sources
|
| 278 | .. (helpers description), and from scripts/bpf_doc.py in the same
|
| 279 | .. repository (header and footer).
|
| 280 | '''
|
| 281 | print(license)
|
| 282 |
|
| 283 | def print_elem(self, elem):
|
| 284 | if (elem.desc):
|
| 285 | print('\tDescription')
|
| 286 | |
| 287 | |
| 288 | for line in re.sub('\n$', '', elem.desc, count=1).split('\n'):
|
| 289 | print('{}{}'.format('\t\t' if line else '', line))
|
| 290 |
|
| 291 | if (elem.ret):
|
| 292 | print('\tReturn')
|
| 293 | for line in elem.ret.rstrip().split('\n'):
|
| 294 | print('{}{}'.format('\t\t' if line else '', line))
|
| 295 |
|
| 296 | print('')
|
| 297 |
|
| 298 |
|
| 299 | class PrinterHelpersRST(PrinterRST):
|
| 300 | |
| 301 | A printer for dumping collected information about helpers as a ReStructured
|
| 302 | Text page compatible with the rst2man program, which can be used to
|
| 303 | generate a manual page for the helpers.
|
| 304 | @parser: A HeaderParser with Helper objects to print to standard output
|
| 305 |
|
| 306 | def __init__(self, parser):
|
| 307 | self.elements = parser.helpers
|
| 308 |
|
| 309 | def print_header(self):
|
| 310 | header = '''\
|
| 311 | ===========
|
| 312 | BPF-HELPERS
|
| 313 | ===========
|
| 314 | -------------------------------------------------------------------------------
|
| 315 | list of eBPF helper functions
|
| 316 | -------------------------------------------------------------------------------
|
| 317 |
|
| 318 | :Manual section: 7
|
| 319 |
|
| 320 | DESCRIPTION
|
| 321 | ===========
|
| 322 |
|
| 323 | The extended Berkeley Packet Filter (eBPF) subsystem consists in programs
|
| 324 | written in a pseudo-assembly language, then attached to one of the several
|
| 325 | kernel hooks and run in reaction of specific events. This framework differs
|
| 326 | from the older, "classic" BPF (or "cBPF") in several aspects, one of them being
|
| 327 | the ability to call special functions (or "helpers") from within a program.
|
| 328 | These functions are restricted to a white-list of helpers defined in the
|
| 329 | kernel.
|
| 330 |
|
| 331 | These helpers are used by eBPF programs to interact with the system, or with
|
| 332 | the context in which they work. For instance, they can be used to print
|
| 333 | debugging messages, to get the time since the system was booted, to interact
|
| 334 | with eBPF maps, or to manipulate network packets. Since there are several eBPF
|
| 335 | program types, and that they do not run in the same context, each program type
|
| 336 | can only call a subset of those helpers.
|
| 337 |
|
| 338 | Due to eBPF conventions, a helper can not have more than five arguments.
|
| 339 |
|
| 340 | Internally, eBPF programs call directly into the compiled helper functions
|
| 341 | without requiring any foreign-function interface. As a result, calling helpers
|
| 342 | introduces no overhead, thus offering excellent performance.
|
| 343 |
|
| 344 | This document is an attempt to list and document the helpers available to eBPF
|
| 345 | developers. They are sorted by chronological order (the oldest helpers in the
|
| 346 | kernel at the top).
|
| 347 |
|
| 348 | HELPERS
|
| 349 | =======
|
| 350 | '''
|
| 351 | PrinterRST.print_license(self)
|
| 352 | print(header)
|
| 353 |
|
| 354 | def print_footer(self):
|
| 355 | footer = '''
|
| 356 | EXAMPLES
|
| 357 | ========
|
| 358 |
|
| 359 | Example usage for most of the eBPF helpers listed in this manual page are
|
| 360 | available within the Linux kernel sources, at the following locations:
|
| 361 |
|
| 362 | * *samples/bpf/*
|
| 363 | * *tools/testing/selftests/bpf/*
|
| 364 |
|
| 365 | LICENSE
|
| 366 | =======
|
| 367 |
|
| 368 | eBPF programs can have an associated license, passed along with the bytecode
|
| 369 | instructions to the kernel when the programs are loaded. The format for that
|
| 370 | string is identical to the one in use for kernel modules (Dual licenses, such
|
| 371 | as "Dual BSD/GPL", may be used). Some helper functions are only accessible to
|
| 372 | programs that are compatible with the GNU Privacy License (GPL).
|
| 373 |
|
| 374 | In order to use such helpers, the eBPF program must be loaded with the correct
|
| 375 | license string passed (via **attr**) to the **bpf**\ () system call, and this
|
| 376 | generally translates into the C source code of the program containing a line
|
| 377 | similar to the following:
|
| 378 |
|
| 379 | ::
|
| 380 |
|
| 381 | char ____license[] __attribute__((section("license"), used)) = "GPL";
|
| 382 |
|
| 383 | IMPLEMENTATION
|
| 384 | ==============
|
| 385 |
|
| 386 | This manual page is an effort to document the existing eBPF helper functions.
|
| 387 | But as of this writing, the BPF sub-system is under heavy development. New eBPF
|
| 388 | program or map types are added, along with new helper functions. Some helpers
|
| 389 | are occasionally made available for additional program types. So in spite of
|
| 390 | the efforts of the community, this page might not be up-to-date. If you want to
|
| 391 | check by yourself what helper functions exist in your kernel, or what types of
|
| 392 | programs they can support, here are some files among the kernel tree that you
|
| 393 | may be interested in:
|
| 394 |
|
| 395 | * *include/uapi/linux/bpf.h* is the main BPF header. It contains the full list
|
| 396 | of all helper functions, as well as many other BPF definitions including most
|
| 397 | of the flags, structs or constants used by the helpers.
|
| 398 | * *net/core/filter.c* contains the definition of most network-related helper
|
| 399 | functions, and the list of program types from which they can be used.
|
| 400 | * *kernel/trace/bpf_trace.c* is the equivalent for most tracing program-related
|
| 401 | helpers.
|
| 402 | * *kernel/bpf/verifier.c* contains the functions used to check that valid types
|
| 403 | of eBPF maps are used with a given helper function.
|
| 404 | * *kernel/bpf/* directory contains other files in which additional helpers are
|
| 405 | defined (for cgroups, sockmaps, etc.).
|
| 406 | * The bpftool utility can be used to probe the availability of helper functions
|
| 407 | on the system (as well as supported program and map types, and a number of
|
| 408 | other parameters). To do so, run **bpftool feature probe** (see
|
| 409 | **bpftool-feature**\ (8) for details). Add the **unprivileged** keyword to
|
| 410 | list features available to unprivileged users.
|
| 411 |
|
| 412 | Compatibility between helper functions and program types can generally be found
|
| 413 | in the files where helper functions are defined. Look for the **struct
|
| 414 | bpf_func_proto** objects and for functions returning them: these functions
|
| 415 | contain a list of helpers that a given program type can call. Note that the
|
| 416 | **default:** label of the **switch ... case** used to filter helpers can call
|
| 417 | other functions, themselves allowing access to additional helpers. The
|
| 418 | requirement for GPL license is also in those **struct bpf_func_proto**.
|
| 419 |
|
| 420 | Compatibility between helper functions and map types can be found in the
|
| 421 | **check_map_func_compatibility**\ () function in file *kernel/bpf/verifier.c*.
|
| 422 |
|
| 423 | Helper functions that invalidate the checks on **data** and **data_end**
|
| 424 | pointers for network processing are listed in function
|
| 425 | **bpf_helper_changes_pkt_data**\ () in file *net/core/filter.c*.
|
| 426 |
|
| 427 | SEE ALSO
|
| 428 | ========
|
| 429 |
|
| 430 | **bpf**\ (2),
|
| 431 | **bpftool**\ (8),
|
| 432 | **cgroups**\ (7),
|
| 433 | **ip**\ (8),
|
| 434 | **perf_event_open**\ (2),
|
| 435 | **sendmsg**\ (2),
|
| 436 | **socket**\ (7),
|
| 437 | **tc-bpf**\ (8)'''
|
| 438 | print(footer)
|
| 439 |
|
| 440 | def print_proto(self, helper):
|
| 441 | |
| 442 | Format function protocol with bold and italics markers. This makes RST
|
| 443 | file less readable, but gives nice results in the manual page.
|
| 444 |
|
| 445 | proto = helper.proto_break_down()
|
| 446 |
|
| 447 | print('**%s %s%s(' % (proto['ret_type'],
|
| 448 | proto['ret_star'].replace('*', '\\*'),
|
| 449 | proto['name']),
|
| 450 | end='')
|
| 451 |
|
| 452 | comma = ''
|
| 453 | for a in proto['args']:
|
| 454 | one_arg = '{}{}'.format(comma, a['type'])
|
| 455 | if a['name']:
|
| 456 | if a['star']:
|
| 457 | one_arg += ' {}**\ '.format(a['star'].replace('*', '\\*'))
|
| 458 | else:
|
| 459 | one_arg += '** '
|
| 460 | one_arg += '*{}*\\ **'.format(a['name'])
|
| 461 | comma = ', '
|
| 462 | print(one_arg, end='')
|
| 463 |
|
| 464 | print(')**')
|
| 465 |
|
| 466 | def print_one(self, helper):
|
| 467 | self.print_proto(helper)
|
| 468 | self.print_elem(helper)
|
| 469 |
|
| 470 |
|
| 471 | class PrinterSyscallRST(PrinterRST):
|
| 472 | |
| 473 | A printer for dumping collected information about the syscall API as a
|
| 474 | ReStructured Text page compatible with the rst2man program, which can be
|
| 475 | used to generate a manual page for the syscall.
|
| 476 | @parser: A HeaderParser with APIElement objects to print to standard
|
| 477 | output
|
| 478 |
|
| 479 | def __init__(self, parser):
|
| 480 | self.elements = parser.commands
|
| 481 |
|
| 482 | def print_header(self):
|
| 483 | header = '''\
|
| 484 | ===
|
| 485 | bpf
|
| 486 | ===
|
| 487 | -------------------------------------------------------------------------------
|
| 488 | Perform a command on an extended BPF object
|
| 489 | -------------------------------------------------------------------------------
|
| 490 |
|
| 491 | :Manual section: 2
|
| 492 |
|
| 493 | COMMANDS
|
| 494 | ========
|
| 495 | '''
|
| 496 | PrinterRST.print_license(self)
|
| 497 | print(header)
|
| 498 |
|
| 499 | def print_one(self, command):
|
| 500 | print('**%s**' % (command.proto))
|
| 501 | self.print_elem(command)
|
| 502 |
|
| 503 |
|
| 504 | class PrinterHelpers(Printer):
|
| 505 | |
| 506 | A printer for dumping collected information about helpers as C header to
|
| 507 | be included from BPF program.
|
| 508 | @parser: A HeaderParser with Helper objects to print to standard output
|
| 509 |
|
| 510 | def __init__(self, parser):
|
| 511 | self.elements = parser.helpers
|
| 512 |
|
| 513 | type_fwds = [
|
| 514 | 'struct bpf_fib_lookup',
|
| 515 | 'struct bpf_sk_lookup',
|
| 516 | 'struct bpf_perf_event_data',
|
| 517 | 'struct bpf_perf_event_value',
|
| 518 | 'struct bpf_pidns_info',
|
| 519 | 'struct bpf_redir_neigh',
|
| 520 | 'struct bpf_sock',
|
| 521 | 'struct bpf_sock_addr',
|
| 522 | 'struct bpf_sock_ops',
|
| 523 | 'struct bpf_sock_tuple',
|
| 524 | 'struct bpf_spin_lock',
|
| 525 | 'struct bpf_sysctl',
|
| 526 | 'struct bpf_tcp_sock',
|
| 527 | 'struct bpf_tunnel_key',
|
| 528 | 'struct bpf_xfrm_state',
|
| 529 | 'struct linux_binprm',
|
| 530 | 'struct pt_regs',
|
| 531 | 'struct sk_reuseport_md',
|
| 532 | 'struct sockaddr',
|
| 533 | 'struct tcphdr',
|
| 534 | 'struct seq_file',
|
| 535 | 'struct tcp6_sock',
|
| 536 | 'struct tcp_sock',
|
| 537 | 'struct tcp_timewait_sock',
|
| 538 | 'struct tcp_request_sock',
|
| 539 | 'struct udp6_sock',
|
| 540 | 'struct unix_sock',
|
| 541 | 'struct task_struct',
|
| 542 |
|
| 543 | 'struct __sk_buff',
|
| 544 | 'struct sk_msg_md',
|
| 545 | 'struct xdp_md',
|
| 546 | 'struct path',
|
| 547 | 'struct btf_ptr',
|
| 548 | 'struct inode',
|
| 549 | 'struct socket',
|
| 550 | 'struct file',
|
| 551 | 'struct bpf_timer',
|
| 552 | ]
|
| 553 | known_types = {
|
| 554 | '...',
|
| 555 | 'void',
|
| 556 | 'const void',
|
| 557 | 'char',
|
| 558 | 'const char',
|
| 559 | 'int',
|
| 560 | 'long',
|
| 561 | 'unsigned long',
|
| 562 |
|
| 563 | '__be16',
|
| 564 | '__be32',
|
| 565 | '__wsum',
|
| 566 |
|
| 567 | 'struct bpf_fib_lookup',
|
| 568 | 'struct bpf_perf_event_data',
|
| 569 | 'struct bpf_perf_event_value',
|
| 570 | 'struct bpf_pidns_info',
|
| 571 | 'struct bpf_redir_neigh',
|
| 572 | 'struct bpf_sk_lookup',
|
| 573 | 'struct bpf_sock',
|
| 574 | 'struct bpf_sock_addr',
|
| 575 | 'struct bpf_sock_ops',
|
| 576 | 'struct bpf_sock_tuple',
|
| 577 | 'struct bpf_spin_lock',
|
| 578 | 'struct bpf_sysctl',
|
| 579 | 'struct bpf_tcp_sock',
|
| 580 | 'struct bpf_tunnel_key',
|
| 581 | 'struct bpf_xfrm_state',
|
| 582 | 'struct linux_binprm',
|
| 583 | 'struct pt_regs',
|
| 584 | 'struct sk_reuseport_md',
|
| 585 | 'struct sockaddr',
|
| 586 | 'struct tcphdr',
|
| 587 | 'struct seq_file',
|
| 588 | 'struct tcp6_sock',
|
| 589 | 'struct tcp_sock',
|
| 590 | 'struct tcp_timewait_sock',
|
| 591 | 'struct tcp_request_sock',
|
| 592 | 'struct udp6_sock',
|
| 593 | 'struct unix_sock',
|
| 594 | 'struct task_struct',
|
| 595 | 'struct path',
|
| 596 | 'struct btf_ptr',
|
| 597 | 'struct inode',
|
| 598 | 'struct socket',
|
| 599 | 'struct file',
|
| 600 | 'struct bpf_timer',
|
| 601 | }
|
| 602 | mapped_types = {
|
| 603 | 'u8': '__u8',
|
| 604 | 'u16': '__u16',
|
| 605 | 'u32': '__u32',
|
| 606 | 'u64': '__u64',
|
| 607 | 's8': '__s8',
|
| 608 | 's16': '__s16',
|
| 609 | 's32': '__s32',
|
| 610 | 's64': '__s64',
|
| 611 | 'size_t': 'unsigned long',
|
| 612 | 'struct bpf_map': 'void',
|
| 613 | 'struct sk_buff': 'struct __sk_buff',
|
| 614 | 'const struct sk_buff': 'const struct __sk_buff',
|
| 615 | 'struct sk_msg_buff': 'struct sk_msg_md',
|
| 616 | 'struct xdp_buff': 'struct xdp_md',
|
| 617 | }
|
| 618 | |
| 619 | overloaded_helpers = [
|
| 620 | 'bpf_get_socket_cookie',
|
| 621 | 'bpf_sk_assign',
|
| 622 | ]
|
| 623 |
|
| 624 | def print_header(self):
|
| 625 | header = '''\
|
| 626 | /* This is auto-generated file. See bpf_doc.py for details. */
|
| 627 |
|
| 628 | /* Forward declarations of BPF structs */'''
|
| 629 |
|
| 630 | print(header)
|
| 631 | for fwd in self.type_fwds:
|
| 632 | print('%s;' % fwd)
|
| 633 | print('')
|
| 634 |
|
| 635 | def print_footer(self):
|
| 636 | footer = ''
|
| 637 | print(footer)
|
| 638 |
|
| 639 | def map_type(self, t):
|
| 640 | if t in self.known_types:
|
| 641 | return t
|
| 642 | if t in self.mapped_types:
|
| 643 | return self.mapped_types[t]
|
| 644 | print("Unrecognized type '%s', please add it to known types!" % t,
|
| 645 | file=sys.stderr)
|
| 646 | sys.exit(1)
|
| 647 |
|
| 648 | seen_helpers = set()
|
| 649 |
|
| 650 | def print_one(self, helper):
|
| 651 | proto = helper.proto_break_down()
|
| 652 |
|
| 653 | if proto['name'] in self.seen_helpers:
|
| 654 | return
|
| 655 | self.seen_helpers.add(proto['name'])
|
| 656 |
|
| 657 | print('/*')
|
| 658 | print(" * %s" % proto['name'])
|
| 659 | print(" *")
|
| 660 | if (helper.desc):
|
| 661 | |
| 662 | |
| 663 | for line in re.sub('\n$', '', helper.desc, count=1).split('\n'):
|
| 664 | print(' *{}{}'.format(' \t' if line else '', line))
|
| 665 |
|
| 666 | if (helper.ret):
|
| 667 | print(' *')
|
| 668 | print(' * Returns')
|
| 669 | for line in helper.ret.rstrip().split('\n'):
|
| 670 | print(' *{}{}'.format(' \t' if line else '', line))
|
| 671 |
|
| 672 | print(' */')
|
| 673 | print('static %s %s(*%s)(' % (self.map_type(proto['ret_type']),
|
| 674 | proto['ret_star'], proto['name']), end='')
|
| 675 | comma = ''
|
| 676 | for i, a in enumerate(proto['args']):
|
| 677 | t = a['type']
|
| 678 | n = a['name']
|
| 679 | if proto['name'] in self.overloaded_helpers and i == 0:
|
| 680 | t = 'void'
|
| 681 | n = 'ctx'
|
| 682 | one_arg = '{}{}'.format(comma, self.map_type(t))
|
| 683 | if n:
|
| 684 | if a['star']:
|
| 685 | one_arg += ' {}'.format(a['star'])
|
| 686 | else:
|
| 687 | one_arg += ' '
|
| 688 | one_arg += '{}'.format(n)
|
| 689 | comma = ', '
|
| 690 | print(one_arg, end='')
|
| 691 |
|
| 692 | print(') = (void *) %d;' % len(self.seen_helpers))
|
| 693 | print('')
|
| 694 |
|
| 695 | |
| 696 |
|
| 697 | |
| 698 | |
| 699 | |
| 700 | script = os.path.abspath(sys.argv[0])
|
| 701 | linuxRoot = os.path.dirname(os.path.dirname(script))
|
| 702 | bpfh = os.path.join(linuxRoot, 'include/uapi/linux/bpf.h')
|
| 703 |
|
| 704 | printers = {
|
| 705 | 'helpers': PrinterHelpersRST,
|
| 706 | 'syscall': PrinterSyscallRST,
|
| 707 | }
|
| 708 |
|
| 709 | argParser = argparse.ArgumentParser(description="""
|
| 710 | Parse eBPF header file and generate documentation for the eBPF API.
|
| 711 | The RST-formatted output produced can be turned into a manual page with the
|
| 712 | rst2man utility.
|
| 713 | """)
|
| 714 | argParser.add_argument('--header', action='store_true',
|
| 715 | help='generate C header file')
|
| 716 | if (os.path.isfile(bpfh)):
|
| 717 | argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h',
|
| 718 | default=bpfh)
|
| 719 | else:
|
| 720 | argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h')
|
| 721 | argParser.add_argument('target', nargs='?', default='helpers',
|
| 722 | choices=printers.keys(), help='eBPF API target')
|
| 723 | args = argParser.parse_args()
|
| 724 |
|
| 725 | |
| 726 | headerParser = HeaderParser(args.filename)
|
| 727 | headerParser.run()
|
| 728 |
|
| 729 | |
| 730 | if args.header:
|
| 731 | if args.target != 'helpers':
|
| 732 | raise NotImplementedError('Only helpers header generation is supported')
|
| 733 | printer = PrinterHelpers(headerParser)
|
| 734 | else:
|
| 735 | printer = printers[args.target](headerParser)
|
| 736 | printer.print_all()
|