- //= OSLog.h - Analysis of calls to os_log builtins --*- C++ -*-===============// 
- // 
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 
- // See https://llvm.org/LICENSE.txt for license information. 
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 
- // 
- //===----------------------------------------------------------------------===// 
- // 
- // This file defines APIs for determining the layout of the data buffer for 
- // os_log() and os_trace(). 
- // 
- //===----------------------------------------------------------------------===// 
-   
- #ifndef LLVM_CLANG_AST_OSLOG_H 
- #define LLVM_CLANG_AST_OSLOG_H 
-   
- #include "clang/AST/ASTContext.h" 
- #include "clang/AST/Expr.h" 
-   
- namespace clang { 
- namespace analyze_os_log { 
-   
- /// An OSLogBufferItem represents a single item in the data written by a call 
- /// to os_log() or os_trace(). 
- class OSLogBufferItem { 
- public: 
-   enum Kind { 
-     // The item is a scalar (int, float, raw pointer, etc.). No further copying 
-     // is required. This is the only kind allowed by os_trace(). 
-     ScalarKind = 0, 
-   
-     // The item is a count, which describes the length of the following item to 
-     // be copied. A count may only be followed by an item of kind StringKind, 
-     // WideStringKind, or PointerKind. 
-     CountKind, 
-   
-     // The item is a pointer to a C string. If preceded by a count 'n', 
-     // os_log() will copy at most 'n' bytes from the pointer. 
-     StringKind, 
-   
-     // The item is a pointer to a block of raw data. This item must be preceded 
-     // by a count 'n'. os_log() will copy exactly 'n' bytes from the pointer. 
-     PointerKind, 
-   
-     // The item is a pointer to an Objective-C object. os_log() may retain the 
-     // object for later processing. 
-     ObjCObjKind, 
-   
-     // The item is a pointer to wide-char string. 
-     WideStringKind, 
-   
-     // The item is corresponding to the '%m' format specifier, no value is 
-     // populated in the buffer and the runtime is loading the errno value. 
-     ErrnoKind, 
-   
-     // The item is a mask type. 
-     MaskKind 
-   }; 
-   
-   enum { 
-     // The item is marked "private" in the format string. 
-     IsPrivate = 0x1, 
-   
-     // The item is marked "public" in the format string. 
-     IsPublic = 0x2, 
-   
-     // The item is marked "sensitive" in the format string. 
-     IsSensitive = 0x4 | IsPrivate 
-   }; 
-   
- private: 
-   Kind TheKind = ScalarKind; 
-   const Expr *TheExpr = nullptr; 
-   CharUnits ConstValue; 
-   CharUnits Size; // size of the data, not including the header bytes 
-   unsigned Flags = 0; 
-   StringRef MaskType; 
-   
- public: 
-   OSLogBufferItem(Kind kind, const Expr *expr, CharUnits size, unsigned flags, 
-                   StringRef maskType = StringRef()) 
-       : TheKind(kind), TheExpr(expr), Size(size), Flags(flags), 
-         MaskType(maskType) { 
-     assert(((Flags == 0) || (Flags == IsPrivate) || (Flags == IsPublic) || 
-             (Flags == IsSensitive)) && 
-            "unexpected privacy flag"); 
-   } 
-   
-   OSLogBufferItem(ASTContext &Ctx, CharUnits value, unsigned flags) 
-       : TheKind(CountKind), ConstValue(value), 
-         Size(Ctx.getTypeSizeInChars(Ctx.IntTy)), Flags(flags) {} 
-   
-   unsigned char getDescriptorByte() const { 
-     unsigned char result = Flags; 
-     result |= ((unsigned)getKind()) << 4; 
-     return result; 
-   } 
-   
-   unsigned char getSizeByte() const { return size().getQuantity(); } 
-   
-   Kind getKind() const { return TheKind; } 
-   bool getIsPrivate() const { return (Flags & IsPrivate) != 0; } 
-   
-   const Expr *getExpr() const { return TheExpr; } 
-   CharUnits getConstValue() const { return ConstValue; } 
-   CharUnits size() const { return Size; } 
-   
-   StringRef getMaskType() const { return MaskType; } 
- }; 
-   
- class OSLogBufferLayout { 
- public: 
-   SmallVector<OSLogBufferItem, 4> Items; 
-   
-   enum Flags { HasPrivateItems = 1, HasNonScalarItems = 1 << 1 }; 
-   
-   CharUnits size() const { 
-     CharUnits result; 
-     result += CharUnits::fromQuantity(2); // summary byte, num-args byte 
-     for (auto &item : Items) { 
-       // descriptor byte, size byte 
-       result += item.size() + CharUnits::fromQuantity(2); 
-     } 
-     return result; 
-   } 
-   
-   bool hasPrivateItems() const { 
-     return llvm::any_of( 
-         Items, [](const OSLogBufferItem &Item) { return Item.getIsPrivate(); }); 
-   } 
-   
-   bool hasNonScalarOrMask() const { 
-     return llvm::any_of(Items, [](const OSLogBufferItem &Item) { 
-       return Item.getKind() != OSLogBufferItem::ScalarKind || 
-              !Item.getMaskType().empty(); 
-     }); 
-   } 
-   
-   unsigned char getSummaryByte() const { 
-     unsigned char result = 0; 
-     if (hasPrivateItems()) 
-       result |= HasPrivateItems; 
-     if (hasNonScalarOrMask()) 
-       result |= HasNonScalarItems; 
-     return result; 
-   } 
-   
-   unsigned char getNumArgsByte() const { return Items.size(); } 
- }; 
-   
- // Given a call 'E' to one of the builtins __builtin_os_log_format() or 
- // __builtin_os_log_format_buffer_size(), compute the layout of the buffer that 
- // the call will write into and store it in 'layout'. Returns 'false' if there 
- // was some error encountered while computing the layout, and 'true' otherwise. 
- bool computeOSLogBufferLayout(clang::ASTContext &Ctx, const clang::CallExpr *E, 
-                               OSLogBufferLayout &layout); 
-   
- } // namespace analyze_os_log 
- } // namespace clang 
- #endif 
-