1/*
2 * Copyright (C) 2017 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16#pragma once
17
18#ifndef PRIVACY_BUFFER_H
19#define PRIVACY_BUFFER_H
20
21#include "Privacy.h"
22
23#include <android/util/EncodedBuffer.h>
24#include <android/util/ProtoOutputStream.h>
25#include <stdint.h>
26#include <utils/Errors.h>
27
28namespace android {
29namespace os {
30namespace incidentd {
31
32using namespace android::util;
33
34/**
35 * PrivacyBuffer holds the original protobuf data and strips PII-sensitive fields
36 * based on the request and holds stripped data in its own buffer for output.
37 */
38class PrivacyBuffer {
39public:
40    PrivacyBuffer(const Privacy* policy, EncodedBuffer::iterator data);
41    ~PrivacyBuffer();
42
43    /**
44     * Strip based on the request and hold data in its own buffer. Return NO_ERROR if strip
45     * succeeds.
46     */
47    status_t strip(const PrivacySpec& spec);
48
49    /**
50     * Clear encoded buffer so it can be reused by another request.
51     */
52    void clear();
53
54    /**
55     * Return the size of the stripped data.
56     */
57    size_t size() const;
58
59    /**
60     * Flush buffer to the given fd. NO_ERROR is returned if the flush succeeds.
61     */
62    status_t flush(int fd);
63
64private:
65    const Privacy* mPolicy;
66    EncodedBuffer::iterator mData;
67
68    ProtoOutputStream mProto;
69    size_t mSize;
70
71    status_t stripField(const Privacy* parentPolicy, const PrivacySpec& spec, int depth);
72    void writeFieldOrSkip(uint32_t fieldTag, bool skip);
73};
74
75}  // namespace incidentd
76}  // namespace os
77}  // namespace android
78
79#endif  // PRIVACY_BUFFER_H