Loading...
1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0-only
3#
4# Copyright (C) 2018-2019 Netronome Systems, Inc.
5# Copyright (C) 2021 Isovalent, Inc.
6
7# In case user attempts to run with Python 2.
8from __future__ import print_function
9
10import argparse
11import re
12import sys, os
13import subprocess
14
15helpersDocStart = 'Start of BPF helper function descriptions:'
16
17class NoHelperFound(BaseException):
18 pass
19
20class NoSyscallCommandFound(BaseException):
21 pass
22
23class ParsingError(BaseException):
24 def __init__(self, line='<line not provided>', reader=None):
25 if reader:
26 BaseException.__init__(self,
27 'Error at file offset %d, parsing line: %s' %
28 (reader.tell(), line))
29 else:
30 BaseException.__init__(self, 'Error parsing line: %s' % line)
31
32
33class APIElement(object):
34 """
35 An object representing the description of an aspect of the eBPF API.
36 @proto: prototype of the API symbol
37 @desc: textual description of the symbol
38 @ret: (optional) description of any associated return value
39 """
40 def __init__(self, proto='', desc='', ret=''):
41 self.proto = proto
42 self.desc = desc
43 self.ret = ret
44
45
46class Helper(APIElement):
47 """
48 An object representing the description of an eBPF helper function.
49 @proto: function prototype of the helper function
50 @desc: textual description of the helper function
51 @ret: description of the return value of the helper function
52 """
53 def __init__(self, *args, **kwargs):
54 super().__init__(*args, **kwargs)
55 self.enum_val = None
56
57 def proto_break_down(self):
58 """
59 Break down helper function protocol into smaller chunks: return type,
60 name, distincts arguments.
61 """
62 arg_re = re.compile(r'((\w+ )*?(\w+|...))( (\**)(\w+))?$')
63 res = {}
64 proto_re = re.compile(r'(.+) (\**)(\w+)\(((([^,]+)(, )?){1,5})\)$')
65
66 capture = proto_re.match(self.proto)
67 res['ret_type'] = capture.group(1)
68 res['ret_star'] = capture.group(2)
69 res['name'] = capture.group(3)
70 res['args'] = []
71
72 args = capture.group(4).split(', ')
73 for a in args:
74 capture = arg_re.match(a)
75 res['args'].append({
76 'type' : capture.group(1),
77 'star' : capture.group(5),
78 'name' : capture.group(6)
79 })
80
81 return res
82
83
84class HeaderParser(object):
85 """
86 An object used to parse a file in order to extract the documentation of a
87 list of eBPF helper functions. All the helpers that can be retrieved are
88 stored as Helper object, in the self.helpers() array.
89 @filename: name of file to parse, usually include/uapi/linux/bpf.h in the
90 kernel tree
91 """
92 def __init__(self, filename):
93 self.reader = open(filename, 'r')
94 self.line = ''
95 self.helpers = []
96 self.commands = []
97 self.desc_unique_helpers = set()
98 self.define_unique_helpers = []
99 self.helper_enum_vals = {}
100 self.helper_enum_pos = {}
101 self.desc_syscalls = []
102 self.enum_syscalls = []
103
104 def parse_element(self):
105 proto = self.parse_symbol()
106 desc = self.parse_desc(proto)
107 ret = self.parse_ret(proto)
108 return APIElement(proto=proto, desc=desc, ret=ret)
109
110 def parse_helper(self):
111 proto = self.parse_proto()
112 desc = self.parse_desc(proto)
113 ret = self.parse_ret(proto)
114 return Helper(proto=proto, desc=desc, ret=ret)
115
116 def parse_symbol(self):
117 p = re.compile(r' \* ?(BPF\w+)$')
118 capture = p.match(self.line)
119 if not capture:
120 raise NoSyscallCommandFound
121 end_re = re.compile(r' \* ?NOTES$')
122 end = end_re.match(self.line)
123 if end:
124 raise NoSyscallCommandFound
125 self.line = self.reader.readline()
126 return capture.group(1)
127
128 def parse_proto(self):
129 # Argument can be of shape:
130 # - "void"
131 # - "type name"
132 # - "type *name"
133 # - Same as above, with "const" and/or "struct" in front of type
134 # - "..." (undefined number of arguments, for bpf_trace_printk())
135 # There is at least one term ("void"), and at most five arguments.
136 p = re.compile(r' \* ?((.+) \**\w+\((((const )?(struct )?(\w+|\.\.\.)( \**\w+)?)(, )?){1,5}\))$')
137 capture = p.match(self.line)
138 if not capture:
139 raise NoHelperFound
140 self.line = self.reader.readline()
141 return capture.group(1)
142
143 def parse_desc(self, proto):
144 p = re.compile(r' \* ?(?:\t| {5,8})Description$')
145 capture = p.match(self.line)
146 if not capture:
147 raise Exception("No description section found for " + proto)
148 # Description can be several lines, some of them possibly empty, and it
149 # stops when another subsection title is met.
150 desc = ''
151 desc_present = False
152 while True:
153 self.line = self.reader.readline()
154 if self.line == ' *\n':
155 desc += '\n'
156 else:
157 p = re.compile(r' \* ?(?:\t| {5,8})(?:\t| {8})(.*)')
158 capture = p.match(self.line)
159 if capture:
160 desc_present = True
161 desc += capture.group(1) + '\n'
162 else:
163 break
164
165 if not desc_present:
166 raise Exception("No description found for " + proto)
167 return desc
168
169 def parse_ret(self, proto):
170 p = re.compile(r' \* ?(?:\t| {5,8})Return$')
171 capture = p.match(self.line)
172 if not capture:
173 raise Exception("No return section found for " + proto)
174 # Return value description can be several lines, some of them possibly
175 # empty, and it stops when another subsection title is met.
176 ret = ''
177 ret_present = False
178 while True:
179 self.line = self.reader.readline()
180 if self.line == ' *\n':
181 ret += '\n'
182 else:
183 p = re.compile(r' \* ?(?:\t| {5,8})(?:\t| {8})(.*)')
184 capture = p.match(self.line)
185 if capture:
186 ret_present = True
187 ret += capture.group(1) + '\n'
188 else:
189 break
190
191 if not ret_present:
192 raise Exception("No return found for " + proto)
193 return ret
194
195 def seek_to(self, target, help_message, discard_lines = 1):
196 self.reader.seek(0)
197 offset = self.reader.read().find(target)
198 if offset == -1:
199 raise Exception(help_message)
200 self.reader.seek(offset)
201 self.reader.readline()
202 for _ in range(discard_lines):
203 self.reader.readline()
204 self.line = self.reader.readline()
205
206 def parse_desc_syscall(self):
207 self.seek_to('* DOC: eBPF Syscall Commands',
208 'Could not find start of eBPF syscall descriptions list')
209 while True:
210 try:
211 command = self.parse_element()
212 self.commands.append(command)
213 self.desc_syscalls.append(command.proto)
214
215 except NoSyscallCommandFound:
216 break
217
218 def parse_enum_syscall(self):
219 self.seek_to('enum bpf_cmd {',
220 'Could not find start of bpf_cmd enum', 0)
221 # Searches for either one or more BPF\w+ enums
222 bpf_p = re.compile(r'\s*(BPF\w+)+')
223 # Searches for an enum entry assigned to another entry,
224 # for e.g. BPF_PROG_RUN = BPF_PROG_TEST_RUN, which is
225 # not documented hence should be skipped in check to
226 # determine if the right number of syscalls are documented
227 assign_p = re.compile(r'\s*(BPF\w+)\s*=\s*(BPF\w+)')
228 bpf_cmd_str = ''
229 while True:
230 capture = assign_p.match(self.line)
231 if capture:
232 # Skip line if an enum entry is assigned to another entry
233 self.line = self.reader.readline()
234 continue
235 capture = bpf_p.match(self.line)
236 if capture:
237 bpf_cmd_str += self.line
238 else:
239 break
240 self.line = self.reader.readline()
241 # Find the number of occurences of BPF\w+
242 self.enum_syscalls = re.findall(r'(BPF\w+)+', bpf_cmd_str)
243
244 def parse_desc_helpers(self):
245 self.seek_to(helpersDocStart,
246 'Could not find start of eBPF helper descriptions list')
247 while True:
248 try:
249 helper = self.parse_helper()
250 self.helpers.append(helper)
251 proto = helper.proto_break_down()
252 self.desc_unique_helpers.add(proto['name'])
253 except NoHelperFound:
254 break
255
256 def parse_define_helpers(self):
257 # Parse FN(...) in #define ___BPF_FUNC_MAPPER to compare later with the
258 # number of unique function names present in description and use the
259 # correct enumeration value.
260 # Note: seek_to(..) discards the first line below the target search text,
261 # resulting in FN(unspec, 0, ##ctx) being skipped and not added to
262 # self.define_unique_helpers.
263 self.seek_to('#define ___BPF_FUNC_MAPPER(FN, ctx...)',
264 'Could not find start of eBPF helper definition list')
265 # Searches for one FN(\w+) define or a backslash for newline
266 p = re.compile(r'\s*FN\((\w+), (\d+), ##ctx\)|\\\\')
267 fn_defines_str = ''
268 i = 0
269 while True:
270 capture = p.match(self.line)
271 if capture:
272 fn_defines_str += self.line
273 helper_name = capture.expand(r'bpf_\1')
274 self.helper_enum_vals[helper_name] = int(capture.group(2))
275 self.helper_enum_pos[helper_name] = i
276 i += 1
277 else:
278 break
279 self.line = self.reader.readline()
280 # Find the number of occurences of FN(\w+)
281 self.define_unique_helpers = re.findall(r'FN\(\w+, \d+, ##ctx\)', fn_defines_str)
282
283 def validate_helpers(self):
284 last_helper = ''
285 seen_helpers = set()
286 seen_enum_vals = set()
287 i = 0
288 for helper in self.helpers:
289 proto = helper.proto_break_down()
290 name = proto['name']
291 try:
292 enum_val = self.helper_enum_vals[name]
293 enum_pos = self.helper_enum_pos[name]
294 except KeyError:
295 raise Exception("Helper %s is missing from enum bpf_func_id" % name)
296
297 if name in seen_helpers:
298 if last_helper != name:
299 raise Exception("Helper %s has multiple descriptions which are not grouped together" % name)
300 continue
301
302 # Enforce current practice of having the descriptions ordered
303 # by enum value.
304 if enum_pos != i:
305 raise Exception("Helper %s (ID %d) comment order (#%d) must be aligned with its position (#%d) in enum bpf_func_id" % (name, enum_val, i + 1, enum_pos + 1))
306 if enum_val in seen_enum_vals:
307 raise Exception("Helper %s has duplicated value %d" % (name, enum_val))
308
309 seen_helpers.add(name)
310 last_helper = name
311 seen_enum_vals.add(enum_val)
312
313 helper.enum_val = enum_val
314 i += 1
315
316 def run(self):
317 self.parse_desc_syscall()
318 self.parse_enum_syscall()
319 self.parse_desc_helpers()
320 self.parse_define_helpers()
321 self.validate_helpers()
322 self.reader.close()
323
324###############################################################################
325
326class Printer(object):
327 """
328 A generic class for printers. Printers should be created with an array of
329 Helper objects, and implement a way to print them in the desired fashion.
330 @parser: A HeaderParser with objects to print to standard output
331 """
332 def __init__(self, parser):
333 self.parser = parser
334 self.elements = []
335
336 def print_header(self):
337 pass
338
339 def print_footer(self):
340 pass
341
342 def print_one(self, helper):
343 pass
344
345 def print_all(self):
346 self.print_header()
347 for elem in self.elements:
348 self.print_one(elem)
349 self.print_footer()
350
351 def elem_number_check(self, desc_unique_elem, define_unique_elem, type, instance):
352 """
353 Checks the number of helpers/syscalls documented within the header file
354 description with those defined as part of enum/macro and raise an
355 Exception if they don't match.
356 """
357 nr_desc_unique_elem = len(desc_unique_elem)
358 nr_define_unique_elem = len(define_unique_elem)
359 if nr_desc_unique_elem != nr_define_unique_elem:
360 exception_msg = '''
361The number of unique %s in description (%d) doesn\'t match the number of unique %s defined in %s (%d)
362''' % (type, nr_desc_unique_elem, type, instance, nr_define_unique_elem)
363 if nr_desc_unique_elem < nr_define_unique_elem:
364 # Function description is parsed until no helper is found (which can be due to
365 # misformatting). Hence, only print the first missing/misformatted helper/enum.
366 exception_msg += '''
367The description for %s is not present or formatted correctly.
368''' % (define_unique_elem[nr_desc_unique_elem])
369 raise Exception(exception_msg)
370
371class PrinterRST(Printer):
372 """
373 A generic class for printers that print ReStructured Text. Printers should
374 be created with a HeaderParser object, and implement a way to print API
375 elements in the desired fashion.
376 @parser: A HeaderParser with objects to print to standard output
377 """
378 def __init__(self, parser):
379 self.parser = parser
380
381 def print_license(self):
382 license = '''\
383.. Copyright (C) All BPF authors and contributors from 2014 to present.
384.. See git log include/uapi/linux/bpf.h in kernel tree for details.
385..
386.. SPDX-License-Identifier: Linux-man-pages-copyleft
387..
388.. Please do not edit this file. It was generated from the documentation
389.. located in file include/uapi/linux/bpf.h of the Linux kernel sources
390.. (helpers description), and from scripts/bpf_doc.py in the same
391.. repository (header and footer).
392'''
393 print(license)
394
395 def print_elem(self, elem):
396 if (elem.desc):
397 print('\tDescription')
398 # Do not strip all newline characters: formatted code at the end of
399 # a section must be followed by a blank line.
400 for line in re.sub('\n$', '', elem.desc, count=1).split('\n'):
401 print('{}{}'.format('\t\t' if line else '', line))
402
403 if (elem.ret):
404 print('\tReturn')
405 for line in elem.ret.rstrip().split('\n'):
406 print('{}{}'.format('\t\t' if line else '', line))
407
408 print('')
409
410 def get_kernel_version(self):
411 try:
412 version = subprocess.run(['git', 'describe'], cwd=linuxRoot,
413 capture_output=True, check=True)
414 version = version.stdout.decode().rstrip()
415 except:
416 try:
417 version = subprocess.run(['make', 'kernelversion'], cwd=linuxRoot,
418 capture_output=True, check=True)
419 version = version.stdout.decode().rstrip()
420 except:
421 return 'Linux'
422 return 'Linux {version}'.format(version=version)
423
424 def get_last_doc_update(self, delimiter):
425 try:
426 cmd = ['git', 'log', '-1', '--pretty=format:%cs', '--no-patch',
427 '-L',
428 '/{}/,/\\*\\//:include/uapi/linux/bpf.h'.format(delimiter)]
429 date = subprocess.run(cmd, cwd=linuxRoot,
430 capture_output=True, check=True)
431 return date.stdout.decode().rstrip()
432 except:
433 return ''
434
435class PrinterHelpersRST(PrinterRST):
436 """
437 A printer for dumping collected information about helpers as a ReStructured
438 Text page compatible with the rst2man program, which can be used to
439 generate a manual page for the helpers.
440 @parser: A HeaderParser with Helper objects to print to standard output
441 """
442 def __init__(self, parser):
443 self.elements = parser.helpers
444 self.elem_number_check(parser.desc_unique_helpers, parser.define_unique_helpers, 'helper', '___BPF_FUNC_MAPPER')
445
446 def print_header(self):
447 header = '''\
448===========
449BPF-HELPERS
450===========
451-------------------------------------------------------------------------------
452list of eBPF helper functions
453-------------------------------------------------------------------------------
454
455:Manual section: 7
456:Version: {version}
457{date_field}{date}
458
459DESCRIPTION
460===========
461
462The extended Berkeley Packet Filter (eBPF) subsystem consists in programs
463written in a pseudo-assembly language, then attached to one of the several
464kernel hooks and run in reaction of specific events. This framework differs
465from the older, "classic" BPF (or "cBPF") in several aspects, one of them being
466the ability to call special functions (or "helpers") from within a program.
467These functions are restricted to a white-list of helpers defined in the
468kernel.
469
470These helpers are used by eBPF programs to interact with the system, or with
471the context in which they work. For instance, they can be used to print
472debugging messages, to get the time since the system was booted, to interact
473with eBPF maps, or to manipulate network packets. Since there are several eBPF
474program types, and that they do not run in the same context, each program type
475can only call a subset of those helpers.
476
477Due to eBPF conventions, a helper can not have more than five arguments.
478
479Internally, eBPF programs call directly into the compiled helper functions
480without requiring any foreign-function interface. As a result, calling helpers
481introduces no overhead, thus offering excellent performance.
482
483This document is an attempt to list and document the helpers available to eBPF
484developers. They are sorted by chronological order (the oldest helpers in the
485kernel at the top).
486
487HELPERS
488=======
489'''
490 kernelVersion = self.get_kernel_version()
491 lastUpdate = self.get_last_doc_update(helpersDocStart)
492
493 PrinterRST.print_license(self)
494 print(header.format(version=kernelVersion,
495 date_field = ':Date: ' if lastUpdate else '',
496 date=lastUpdate))
497
498 def print_footer(self):
499 footer = '''
500EXAMPLES
501========
502
503Example usage for most of the eBPF helpers listed in this manual page are
504available within the Linux kernel sources, at the following locations:
505
506* *samples/bpf/*
507* *tools/testing/selftests/bpf/*
508
509LICENSE
510=======
511
512eBPF programs can have an associated license, passed along with the bytecode
513instructions to the kernel when the programs are loaded. The format for that
514string is identical to the one in use for kernel modules (Dual licenses, such
515as "Dual BSD/GPL", may be used). Some helper functions are only accessible to
516programs that are compatible with the GNU General Public License (GNU GPL).
517
518In order to use such helpers, the eBPF program must be loaded with the correct
519license string passed (via **attr**) to the **bpf**\\ () system call, and this
520generally translates into the C source code of the program containing a line
521similar to the following:
522
523::
524
525 char ____license[] __attribute__((section("license"), used)) = "GPL";
526
527IMPLEMENTATION
528==============
529
530This manual page is an effort to document the existing eBPF helper functions.
531But as of this writing, the BPF sub-system is under heavy development. New eBPF
532program or map types are added, along with new helper functions. Some helpers
533are occasionally made available for additional program types. So in spite of
534the efforts of the community, this page might not be up-to-date. If you want to
535check by yourself what helper functions exist in your kernel, or what types of
536programs they can support, here are some files among the kernel tree that you
537may be interested in:
538
539* *include/uapi/linux/bpf.h* is the main BPF header. It contains the full list
540 of all helper functions, as well as many other BPF definitions including most
541 of the flags, structs or constants used by the helpers.
542* *net/core/filter.c* contains the definition of most network-related helper
543 functions, and the list of program types from which they can be used.
544* *kernel/trace/bpf_trace.c* is the equivalent for most tracing program-related
545 helpers.
546* *kernel/bpf/verifier.c* contains the functions used to check that valid types
547 of eBPF maps are used with a given helper function.
548* *kernel/bpf/* directory contains other files in which additional helpers are
549 defined (for cgroups, sockmaps, etc.).
550* The bpftool utility can be used to probe the availability of helper functions
551 on the system (as well as supported program and map types, and a number of
552 other parameters). To do so, run **bpftool feature probe** (see
553 **bpftool-feature**\\ (8) for details). Add the **unprivileged** keyword to
554 list features available to unprivileged users.
555
556Compatibility between helper functions and program types can generally be found
557in the files where helper functions are defined. Look for the **struct
558bpf_func_proto** objects and for functions returning them: these functions
559contain a list of helpers that a given program type can call. Note that the
560**default:** label of the **switch ... case** used to filter helpers can call
561other functions, themselves allowing access to additional helpers. The
562requirement for GPL license is also in those **struct bpf_func_proto**.
563
564Compatibility between helper functions and map types can be found in the
565**check_map_func_compatibility**\\ () function in file *kernel/bpf/verifier.c*.
566
567Helper functions that invalidate the checks on **data** and **data_end**
568pointers for network processing are listed in function
569**bpf_helper_changes_pkt_data**\\ () in file *net/core/filter.c*.
570
571SEE ALSO
572========
573
574**bpf**\\ (2),
575**bpftool**\\ (8),
576**cgroups**\\ (7),
577**ip**\\ (8),
578**perf_event_open**\\ (2),
579**sendmsg**\\ (2),
580**socket**\\ (7),
581**tc-bpf**\\ (8)'''
582 print(footer)
583
584 def print_proto(self, helper):
585 """
586 Format function protocol with bold and italics markers. This makes RST
587 file less readable, but gives nice results in the manual page.
588 """
589 proto = helper.proto_break_down()
590
591 print('**%s %s%s(' % (proto['ret_type'],
592 proto['ret_star'].replace('*', '\\*'),
593 proto['name']),
594 end='')
595
596 comma = ''
597 for a in proto['args']:
598 one_arg = '{}{}'.format(comma, a['type'])
599 if a['name']:
600 if a['star']:
601 one_arg += ' {}**\\ '.format(a['star'].replace('*', '\\*'))
602 else:
603 one_arg += '** '
604 one_arg += '*{}*\\ **'.format(a['name'])
605 comma = ', '
606 print(one_arg, end='')
607
608 print(')**')
609
610 def print_one(self, helper):
611 self.print_proto(helper)
612 self.print_elem(helper)
613
614
615class PrinterSyscallRST(PrinterRST):
616 """
617 A printer for dumping collected information about the syscall API as a
618 ReStructured Text page compatible with the rst2man program, which can be
619 used to generate a manual page for the syscall.
620 @parser: A HeaderParser with APIElement objects to print to standard
621 output
622 """
623 def __init__(self, parser):
624 self.elements = parser.commands
625 self.elem_number_check(parser.desc_syscalls, parser.enum_syscalls, 'syscall', 'bpf_cmd')
626
627 def print_header(self):
628 header = '''\
629===
630bpf
631===
632-------------------------------------------------------------------------------
633Perform a command on an extended BPF object
634-------------------------------------------------------------------------------
635
636:Manual section: 2
637
638COMMANDS
639========
640'''
641 PrinterRST.print_license(self)
642 print(header)
643
644 def print_one(self, command):
645 print('**%s**' % (command.proto))
646 self.print_elem(command)
647
648
649class PrinterHelpers(Printer):
650 """
651 A printer for dumping collected information about helpers as C header to
652 be included from BPF program.
653 @parser: A HeaderParser with Helper objects to print to standard output
654 """
655 def __init__(self, parser):
656 self.elements = parser.helpers
657 self.elem_number_check(parser.desc_unique_helpers, parser.define_unique_helpers, 'helper', '___BPF_FUNC_MAPPER')
658
659 type_fwds = [
660 'struct bpf_fib_lookup',
661 'struct bpf_sk_lookup',
662 'struct bpf_perf_event_data',
663 'struct bpf_perf_event_value',
664 'struct bpf_pidns_info',
665 'struct bpf_redir_neigh',
666 'struct bpf_sock',
667 'struct bpf_sock_addr',
668 'struct bpf_sock_ops',
669 'struct bpf_sock_tuple',
670 'struct bpf_spin_lock',
671 'struct bpf_sysctl',
672 'struct bpf_tcp_sock',
673 'struct bpf_tunnel_key',
674 'struct bpf_xfrm_state',
675 'struct linux_binprm',
676 'struct pt_regs',
677 'struct sk_reuseport_md',
678 'struct sockaddr',
679 'struct tcphdr',
680 'struct seq_file',
681 'struct tcp6_sock',
682 'struct tcp_sock',
683 'struct tcp_timewait_sock',
684 'struct tcp_request_sock',
685 'struct udp6_sock',
686 'struct unix_sock',
687 'struct task_struct',
688 'struct cgroup',
689
690 'struct __sk_buff',
691 'struct sk_msg_md',
692 'struct xdp_md',
693 'struct path',
694 'struct btf_ptr',
695 'struct inode',
696 'struct socket',
697 'struct file',
698 'struct bpf_timer',
699 'struct mptcp_sock',
700 'struct bpf_dynptr',
701 'struct iphdr',
702 'struct ipv6hdr',
703 ]
704 known_types = {
705 '...',
706 'void',
707 'const void',
708 'char',
709 'const char',
710 'int',
711 'long',
712 'unsigned long',
713
714 '__be16',
715 '__be32',
716 '__wsum',
717
718 'struct bpf_fib_lookup',
719 'struct bpf_perf_event_data',
720 'struct bpf_perf_event_value',
721 'struct bpf_pidns_info',
722 'struct bpf_redir_neigh',
723 'struct bpf_sk_lookup',
724 'struct bpf_sock',
725 'struct bpf_sock_addr',
726 'struct bpf_sock_ops',
727 'struct bpf_sock_tuple',
728 'struct bpf_spin_lock',
729 'struct bpf_sysctl',
730 'struct bpf_tcp_sock',
731 'struct bpf_tunnel_key',
732 'struct bpf_xfrm_state',
733 'struct linux_binprm',
734 'struct pt_regs',
735 'struct sk_reuseport_md',
736 'struct sockaddr',
737 'struct tcphdr',
738 'struct seq_file',
739 'struct tcp6_sock',
740 'struct tcp_sock',
741 'struct tcp_timewait_sock',
742 'struct tcp_request_sock',
743 'struct udp6_sock',
744 'struct unix_sock',
745 'struct task_struct',
746 'struct cgroup',
747 'struct path',
748 'struct btf_ptr',
749 'struct inode',
750 'struct socket',
751 'struct file',
752 'struct bpf_timer',
753 'struct mptcp_sock',
754 'struct bpf_dynptr',
755 'const struct bpf_dynptr',
756 'struct iphdr',
757 'struct ipv6hdr',
758 }
759 mapped_types = {
760 'u8': '__u8',
761 'u16': '__u16',
762 'u32': '__u32',
763 'u64': '__u64',
764 's8': '__s8',
765 's16': '__s16',
766 's32': '__s32',
767 's64': '__s64',
768 'size_t': 'unsigned long',
769 'struct bpf_map': 'void',
770 'struct sk_buff': 'struct __sk_buff',
771 'const struct sk_buff': 'const struct __sk_buff',
772 'struct sk_msg_buff': 'struct sk_msg_md',
773 'struct xdp_buff': 'struct xdp_md',
774 }
775 # Helpers overloaded for different context types.
776 overloaded_helpers = [
777 'bpf_get_socket_cookie',
778 'bpf_sk_assign',
779 ]
780
781 def print_header(self):
782 header = '''\
783/* This is auto-generated file. See bpf_doc.py for details. */
784
785/* Forward declarations of BPF structs */'''
786
787 print(header)
788 for fwd in self.type_fwds:
789 print('%s;' % fwd)
790 print('')
791
792 def print_footer(self):
793 footer = ''
794 print(footer)
795
796 def map_type(self, t):
797 if t in self.known_types:
798 return t
799 if t in self.mapped_types:
800 return self.mapped_types[t]
801 print("Unrecognized type '%s', please add it to known types!" % t,
802 file=sys.stderr)
803 sys.exit(1)
804
805 seen_helpers = set()
806
807 def print_one(self, helper):
808 proto = helper.proto_break_down()
809
810 if proto['name'] in self.seen_helpers:
811 return
812 self.seen_helpers.add(proto['name'])
813
814 print('/*')
815 print(" * %s" % proto['name'])
816 print(" *")
817 if (helper.desc):
818 # Do not strip all newline characters: formatted code at the end of
819 # a section must be followed by a blank line.
820 for line in re.sub('\n$', '', helper.desc, count=1).split('\n'):
821 print(' *{}{}'.format(' \t' if line else '', line))
822
823 if (helper.ret):
824 print(' *')
825 print(' * Returns')
826 for line in helper.ret.rstrip().split('\n'):
827 print(' *{}{}'.format(' \t' if line else '', line))
828
829 print(' */')
830 print('static %s %s(*%s)(' % (self.map_type(proto['ret_type']),
831 proto['ret_star'], proto['name']), end='')
832 comma = ''
833 for i, a in enumerate(proto['args']):
834 t = a['type']
835 n = a['name']
836 if proto['name'] in self.overloaded_helpers and i == 0:
837 t = 'void'
838 n = 'ctx'
839 one_arg = '{}{}'.format(comma, self.map_type(t))
840 if n:
841 if a['star']:
842 one_arg += ' {}'.format(a['star'])
843 else:
844 one_arg += ' '
845 one_arg += '{}'.format(n)
846 comma = ', '
847 print(one_arg, end='')
848
849 print(') = (void *) %d;' % helper.enum_val)
850 print('')
851
852###############################################################################
853
854# If script is launched from scripts/ from kernel tree and can access
855# ../include/uapi/linux/bpf.h, use it as a default name for the file to parse,
856# otherwise the --filename argument will be required from the command line.
857script = os.path.abspath(sys.argv[0])
858linuxRoot = os.path.dirname(os.path.dirname(script))
859bpfh = os.path.join(linuxRoot, 'include/uapi/linux/bpf.h')
860
861printers = {
862 'helpers': PrinterHelpersRST,
863 'syscall': PrinterSyscallRST,
864}
865
866argParser = argparse.ArgumentParser(description="""
867Parse eBPF header file and generate documentation for the eBPF API.
868The RST-formatted output produced can be turned into a manual page with the
869rst2man utility.
870""")
871argParser.add_argument('--header', action='store_true',
872 help='generate C header file')
873if (os.path.isfile(bpfh)):
874 argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h',
875 default=bpfh)
876else:
877 argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h')
878argParser.add_argument('target', nargs='?', default='helpers',
879 choices=printers.keys(), help='eBPF API target')
880args = argParser.parse_args()
881
882# Parse file.
883headerParser = HeaderParser(args.filename)
884headerParser.run()
885
886# Print formatted output to standard output.
887if args.header:
888 if args.target != 'helpers':
889 raise NotImplementedError('Only helpers header generation is supported')
890 printer = PrinterHelpers(headerParser)
891else:
892 printer = printers[args.target](headerParser)
893printer.print_all()
1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0-only
3#
4# Copyright (C) 2018-2019 Netronome Systems, Inc.
5# Copyright (C) 2021 Isovalent, Inc.
6
7# In case user attempts to run with Python 2.
8from __future__ import print_function
9
10import argparse
11import re
12import sys, os
13
14class NoHelperFound(BaseException):
15 pass
16
17class NoSyscallCommandFound(BaseException):
18 pass
19
20class 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
30class 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
43class 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
77class 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 # Argument can be of shape:
117 # - "void"
118 # - "type name"
119 # - "type *name"
120 # - Same as above, with "const" and/or "struct" in front of type
121 # - "..." (undefined number of arguments, for bpf_trace_printk())
122 # There is at least one term ("void"), and at most five arguments.
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 # Helper can have empty description and we might be parsing another
135 # attribute: return but do not consume.
136 return ''
137 # Description can be several lines, some of them possibly empty, and it
138 # stops when another subsection title is met.
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 # Helper can have empty retval and we might be parsing another
158 # attribute: return but do not consume.
159 return ''
160 # Return value description can be several lines, some of them possibly
161 # empty, and it stops when another subsection title is met.
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
213class 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
239class 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 # Do not strip all newline characters: formatted code at the end of
287 # a section must be followed by a blank line.
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
299class 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===========
312BPF-HELPERS
313===========
314-------------------------------------------------------------------------------
315list of eBPF helper functions
316-------------------------------------------------------------------------------
317
318:Manual section: 7
319
320DESCRIPTION
321===========
322
323The extended Berkeley Packet Filter (eBPF) subsystem consists in programs
324written in a pseudo-assembly language, then attached to one of the several
325kernel hooks and run in reaction of specific events. This framework differs
326from the older, "classic" BPF (or "cBPF") in several aspects, one of them being
327the ability to call special functions (or "helpers") from within a program.
328These functions are restricted to a white-list of helpers defined in the
329kernel.
330
331These helpers are used by eBPF programs to interact with the system, or with
332the context in which they work. For instance, they can be used to print
333debugging messages, to get the time since the system was booted, to interact
334with eBPF maps, or to manipulate network packets. Since there are several eBPF
335program types, and that they do not run in the same context, each program type
336can only call a subset of those helpers.
337
338Due to eBPF conventions, a helper can not have more than five arguments.
339
340Internally, eBPF programs call directly into the compiled helper functions
341without requiring any foreign-function interface. As a result, calling helpers
342introduces no overhead, thus offering excellent performance.
343
344This document is an attempt to list and document the helpers available to eBPF
345developers. They are sorted by chronological order (the oldest helpers in the
346kernel at the top).
347
348HELPERS
349=======
350'''
351 PrinterRST.print_license(self)
352 print(header)
353
354 def print_footer(self):
355 footer = '''
356EXAMPLES
357========
358
359Example usage for most of the eBPF helpers listed in this manual page are
360available within the Linux kernel sources, at the following locations:
361
362* *samples/bpf/*
363* *tools/testing/selftests/bpf/*
364
365LICENSE
366=======
367
368eBPF programs can have an associated license, passed along with the bytecode
369instructions to the kernel when the programs are loaded. The format for that
370string is identical to the one in use for kernel modules (Dual licenses, such
371as "Dual BSD/GPL", may be used). Some helper functions are only accessible to
372programs that are compatible with the GNU Privacy License (GPL).
373
374In order to use such helpers, the eBPF program must be loaded with the correct
375license string passed (via **attr**) to the **bpf**\ () system call, and this
376generally translates into the C source code of the program containing a line
377similar to the following:
378
379::
380
381 char ____license[] __attribute__((section("license"), used)) = "GPL";
382
383IMPLEMENTATION
384==============
385
386This manual page is an effort to document the existing eBPF helper functions.
387But as of this writing, the BPF sub-system is under heavy development. New eBPF
388program or map types are added, along with new helper functions. Some helpers
389are occasionally made available for additional program types. So in spite of
390the efforts of the community, this page might not be up-to-date. If you want to
391check by yourself what helper functions exist in your kernel, or what types of
392programs they can support, here are some files among the kernel tree that you
393may 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
412Compatibility between helper functions and program types can generally be found
413in the files where helper functions are defined. Look for the **struct
414bpf_func_proto** objects and for functions returning them: these functions
415contain 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
417other functions, themselves allowing access to additional helpers. The
418requirement for GPL license is also in those **struct bpf_func_proto**.
419
420Compatibility between helper functions and map types can be found in the
421**check_map_func_compatibility**\ () function in file *kernel/bpf/verifier.c*.
422
423Helper functions that invalidate the checks on **data** and **data_end**
424pointers for network processing are listed in function
425**bpf_helper_changes_pkt_data**\ () in file *net/core/filter.c*.
426
427SEE 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
471class 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===
485bpf
486===
487-------------------------------------------------------------------------------
488Perform a command on an extended BPF object
489-------------------------------------------------------------------------------
490
491:Manual section: 2
492
493COMMANDS
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
504class 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 task_struct',
541
542 'struct __sk_buff',
543 'struct sk_msg_md',
544 'struct xdp_md',
545 'struct path',
546 'struct btf_ptr',
547 'struct inode',
548 'struct socket',
549 'struct file',
550 ]
551 known_types = {
552 '...',
553 'void',
554 'const void',
555 'char',
556 'const char',
557 'int',
558 'long',
559 'unsigned long',
560
561 '__be16',
562 '__be32',
563 '__wsum',
564
565 'struct bpf_fib_lookup',
566 'struct bpf_perf_event_data',
567 'struct bpf_perf_event_value',
568 'struct bpf_pidns_info',
569 'struct bpf_redir_neigh',
570 'struct bpf_sk_lookup',
571 'struct bpf_sock',
572 'struct bpf_sock_addr',
573 'struct bpf_sock_ops',
574 'struct bpf_sock_tuple',
575 'struct bpf_spin_lock',
576 'struct bpf_sysctl',
577 'struct bpf_tcp_sock',
578 'struct bpf_tunnel_key',
579 'struct bpf_xfrm_state',
580 'struct linux_binprm',
581 'struct pt_regs',
582 'struct sk_reuseport_md',
583 'struct sockaddr',
584 'struct tcphdr',
585 'struct seq_file',
586 'struct tcp6_sock',
587 'struct tcp_sock',
588 'struct tcp_timewait_sock',
589 'struct tcp_request_sock',
590 'struct udp6_sock',
591 'struct task_struct',
592 'struct path',
593 'struct btf_ptr',
594 'struct inode',
595 'struct socket',
596 'struct file',
597 }
598 mapped_types = {
599 'u8': '__u8',
600 'u16': '__u16',
601 'u32': '__u32',
602 'u64': '__u64',
603 's8': '__s8',
604 's16': '__s16',
605 's32': '__s32',
606 's64': '__s64',
607 'size_t': 'unsigned long',
608 'struct bpf_map': 'void',
609 'struct sk_buff': 'struct __sk_buff',
610 'const struct sk_buff': 'const struct __sk_buff',
611 'struct sk_msg_buff': 'struct sk_msg_md',
612 'struct xdp_buff': 'struct xdp_md',
613 }
614 # Helpers overloaded for different context types.
615 overloaded_helpers = [
616 'bpf_get_socket_cookie',
617 'bpf_sk_assign',
618 ]
619
620 def print_header(self):
621 header = '''\
622/* This is auto-generated file. See bpf_doc.py for details. */
623
624/* Forward declarations of BPF structs */'''
625
626 print(header)
627 for fwd in self.type_fwds:
628 print('%s;' % fwd)
629 print('')
630
631 def print_footer(self):
632 footer = ''
633 print(footer)
634
635 def map_type(self, t):
636 if t in self.known_types:
637 return t
638 if t in self.mapped_types:
639 return self.mapped_types[t]
640 print("Unrecognized type '%s', please add it to known types!" % t,
641 file=sys.stderr)
642 sys.exit(1)
643
644 seen_helpers = set()
645
646 def print_one(self, helper):
647 proto = helper.proto_break_down()
648
649 if proto['name'] in self.seen_helpers:
650 return
651 self.seen_helpers.add(proto['name'])
652
653 print('/*')
654 print(" * %s" % proto['name'])
655 print(" *")
656 if (helper.desc):
657 # Do not strip all newline characters: formatted code at the end of
658 # a section must be followed by a blank line.
659 for line in re.sub('\n$', '', helper.desc, count=1).split('\n'):
660 print(' *{}{}'.format(' \t' if line else '', line))
661
662 if (helper.ret):
663 print(' *')
664 print(' * Returns')
665 for line in helper.ret.rstrip().split('\n'):
666 print(' *{}{}'.format(' \t' if line else '', line))
667
668 print(' */')
669 print('static %s %s(*%s)(' % (self.map_type(proto['ret_type']),
670 proto['ret_star'], proto['name']), end='')
671 comma = ''
672 for i, a in enumerate(proto['args']):
673 t = a['type']
674 n = a['name']
675 if proto['name'] in self.overloaded_helpers and i == 0:
676 t = 'void'
677 n = 'ctx'
678 one_arg = '{}{}'.format(comma, self.map_type(t))
679 if n:
680 if a['star']:
681 one_arg += ' {}'.format(a['star'])
682 else:
683 one_arg += ' '
684 one_arg += '{}'.format(n)
685 comma = ', '
686 print(one_arg, end='')
687
688 print(') = (void *) %d;' % len(self.seen_helpers))
689 print('')
690
691###############################################################################
692
693# If script is launched from scripts/ from kernel tree and can access
694# ../include/uapi/linux/bpf.h, use it as a default name for the file to parse,
695# otherwise the --filename argument will be required from the command line.
696script = os.path.abspath(sys.argv[0])
697linuxRoot = os.path.dirname(os.path.dirname(script))
698bpfh = os.path.join(linuxRoot, 'include/uapi/linux/bpf.h')
699
700printers = {
701 'helpers': PrinterHelpersRST,
702 'syscall': PrinterSyscallRST,
703}
704
705argParser = argparse.ArgumentParser(description="""
706Parse eBPF header file and generate documentation for the eBPF API.
707The RST-formatted output produced can be turned into a manual page with the
708rst2man utility.
709""")
710argParser.add_argument('--header', action='store_true',
711 help='generate C header file')
712if (os.path.isfile(bpfh)):
713 argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h',
714 default=bpfh)
715else:
716 argParser.add_argument('--filename', help='path to include/uapi/linux/bpf.h')
717argParser.add_argument('target', nargs='?', default='helpers',
718 choices=printers.keys(), help='eBPF API target')
719args = argParser.parse_args()
720
721# Parse file.
722headerParser = HeaderParser(args.filename)
723headerParser.run()
724
725# Print formatted output to standard output.
726if args.header:
727 if args.target != 'helpers':
728 raise NotImplementedError('Only helpers header generation is supported')
729 printer = PrinterHelpers(headerParser)
730else:
731 printer = printers[args.target](headerParser)
732printer.print_all()