Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //= OSLog.h - Analysis of calls to os_log builtins --*- C++ -*-===============// | 
| 2 | // | ||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| 4 | // See https://llvm.org/LICENSE.txt for license information. | ||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| 6 | // | ||
| 7 | //===----------------------------------------------------------------------===// | ||
| 8 | // | ||
| 9 | // This file defines APIs for determining the layout of the data buffer for | ||
| 10 | // os_log() and os_trace(). | ||
| 11 | // | ||
| 12 | //===----------------------------------------------------------------------===// | ||
| 13 | |||
| 14 | #ifndef LLVM_CLANG_AST_OSLOG_H | ||
| 15 | #define LLVM_CLANG_AST_OSLOG_H | ||
| 16 | |||
| 17 | #include "clang/AST/ASTContext.h" | ||
| 18 | #include "clang/AST/Expr.h" | ||
| 19 | |||
| 20 | namespace clang { | ||
| 21 | namespace analyze_os_log { | ||
| 22 | |||
| 23 | /// An OSLogBufferItem represents a single item in the data written by a call | ||
| 24 | /// to os_log() or os_trace(). | ||
| 25 | class OSLogBufferItem { | ||
| 26 | public: | ||
| 27 | enum Kind { | ||
| 28 |     // The item is a scalar (int, float, raw pointer, etc.). No further copying | ||
| 29 |     // is required. This is the only kind allowed by os_trace(). | ||
| 30 | ScalarKind = 0, | ||
| 31 | |||
| 32 |     // The item is a count, which describes the length of the following item to | ||
| 33 |     // be copied. A count may only be followed by an item of kind StringKind, | ||
| 34 |     // WideStringKind, or PointerKind. | ||
| 35 | CountKind, | ||
| 36 | |||
| 37 |     // The item is a pointer to a C string. If preceded by a count 'n', | ||
| 38 |     // os_log() will copy at most 'n' bytes from the pointer. | ||
| 39 | StringKind, | ||
| 40 | |||
| 41 |     // The item is a pointer to a block of raw data. This item must be preceded | ||
| 42 |     // by a count 'n'. os_log() will copy exactly 'n' bytes from the pointer. | ||
| 43 | PointerKind, | ||
| 44 | |||
| 45 |     // The item is a pointer to an Objective-C object. os_log() may retain the | ||
| 46 |     // object for later processing. | ||
| 47 | ObjCObjKind, | ||
| 48 | |||
| 49 |     // The item is a pointer to wide-char string. | ||
| 50 | WideStringKind, | ||
| 51 | |||
| 52 |     // The item is corresponding to the '%m' format specifier, no value is | ||
| 53 |     // populated in the buffer and the runtime is loading the errno value. | ||
| 54 | ErrnoKind, | ||
| 55 | |||
| 56 |     // The item is a mask type. | ||
| 57 | MaskKind | ||
| 58 | }; | ||
| 59 | |||
| 60 | enum { | ||
| 61 |     // The item is marked "private" in the format string. | ||
| 62 | IsPrivate = 0x1, | ||
| 63 | |||
| 64 |     // The item is marked "public" in the format string. | ||
| 65 | IsPublic = 0x2, | ||
| 66 | |||
| 67 |     // The item is marked "sensitive" in the format string. | ||
| 68 | IsSensitive = 0x4 | IsPrivate | ||
| 69 | }; | ||
| 70 | |||
| 71 | private: | ||
| 72 | Kind TheKind = ScalarKind; | ||
| 73 | const Expr *TheExpr = nullptr; | ||
| 74 |   CharUnits ConstValue; | ||
| 75 | CharUnits Size; // size of the data, not including the header bytes | ||
| 76 | unsigned Flags = 0; | ||
| 77 |   StringRef MaskType; | ||
| 78 | |||
| 79 | public: | ||
| 80 | OSLogBufferItem(Kind kind, const Expr *expr, CharUnits size, unsigned flags, | ||
| 81 | StringRef maskType = StringRef()) | ||
| 82 | : TheKind(kind), TheExpr(expr), Size(size), Flags(flags), | ||
| 83 | MaskType(maskType) { | ||
| 84 | assert(((Flags == 0) || (Flags == IsPrivate) || (Flags == IsPublic) || | ||
| 85 | (Flags == IsSensitive)) && | ||
| 86 | "unexpected privacy flag"); | ||
| 87 |   } | ||
| 88 | |||
| 89 | OSLogBufferItem(ASTContext &Ctx, CharUnits value, unsigned flags) | ||
| 90 | : TheKind(CountKind), ConstValue(value), | ||
| 91 | Size(Ctx.getTypeSizeInChars(Ctx.IntTy)), Flags(flags) {} | ||
| 92 | |||
| 93 | unsigned char getDescriptorByte() const { | ||
| 94 | unsigned char result = Flags; | ||
| 95 | result |= ((unsigned)getKind()) << 4; | ||
| 96 | return result; | ||
| 97 |   } | ||
| 98 | |||
| 99 | unsigned char getSizeByte() const { return size().getQuantity(); } | ||
| 100 | |||
| 101 | Kind getKind() const { return TheKind; } | ||
| 102 | bool getIsPrivate() const { return (Flags & IsPrivate) != 0; } | ||
| 103 | |||
| 104 | const Expr *getExpr() const { return TheExpr; } | ||
| 105 | CharUnits getConstValue() const { return ConstValue; } | ||
| 106 | CharUnits size() const { return Size; } | ||
| 107 | |||
| 108 | StringRef getMaskType() const { return MaskType; } | ||
| 109 | }; | ||
| 110 | |||
| 111 | class OSLogBufferLayout { | ||
| 112 | public: | ||
| 113 | SmallVector<OSLogBufferItem, 4> Items; | ||
| 114 | |||
| 115 | enum Flags { HasPrivateItems = 1, HasNonScalarItems = 1 << 1 }; | ||
| 116 | |||
| 117 | CharUnits size() const { | ||
| 118 |     CharUnits result; | ||
| 119 | result += CharUnits::fromQuantity(2); // summary byte, num-args byte | ||
| 120 | for (auto &item : Items) { | ||
| 121 |       // descriptor byte, size byte | ||
| 122 | result += item.size() + CharUnits::fromQuantity(2); | ||
| 123 |     } | ||
| 124 | return result; | ||
| 125 |   } | ||
| 126 | |||
| 127 | bool hasPrivateItems() const { | ||
| 128 | return llvm::any_of( | ||
| 129 | Items, [](const OSLogBufferItem &Item) { return Item.getIsPrivate(); }); | ||
| 130 |   } | ||
| 131 | |||
| 132 | bool hasNonScalarOrMask() const { | ||
| 133 | return llvm::any_of(Items, [](const OSLogBufferItem &Item) { | ||
| 134 | return Item.getKind() != OSLogBufferItem::ScalarKind || | ||
| 135 | !Item.getMaskType().empty(); | ||
| 136 | }); | ||
| 137 |   } | ||
| 138 | |||
| 139 | unsigned char getSummaryByte() const { | ||
| 140 | unsigned char result = 0; | ||
| 141 | if (hasPrivateItems()) | ||
| 142 | result |= HasPrivateItems; | ||
| 143 | if (hasNonScalarOrMask()) | ||
| 144 | result |= HasNonScalarItems; | ||
| 145 | return result; | ||
| 146 |   } | ||
| 147 | |||
| 148 | unsigned char getNumArgsByte() const { return Items.size(); } | ||
| 149 | }; | ||
| 150 | |||
| 151 | // Given a call 'E' to one of the builtins __builtin_os_log_format() or | ||
| 152 | // __builtin_os_log_format_buffer_size(), compute the layout of the buffer that | ||
| 153 | // the call will write into and store it in 'layout'. Returns 'false' if there | ||
| 154 | // was some error encountered while computing the layout, and 'true' otherwise. | ||
| 155 | bool computeOSLogBufferLayout(clang::ASTContext &Ctx, const clang::CallExpr *E, | ||
| 156 | OSLogBufferLayout &layout); | ||
| 157 | |||
| 158 | } // namespace analyze_os_log | ||
| 159 | } // namespace clang | ||
| 160 | #endif |