summaryrefslogtreecommitdiff
path: root/cmds/incidentd/src/proto_util.cpp
blob: 4e8ff71b377763808c576321aefe6a8b6e373323 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "proto_util.h"

#include <google/protobuf/io/zero_copy_stream_impl.h>

#include <android/util/protobuf.h>
#include <android/util/ProtoOutputStream.h>
#include <android-base/file.h>

namespace android {
namespace os {
namespace incidentd {

using namespace android::base;
using namespace android::util;
using google::protobuf::io::FileOutputStream;

// special section ids
const int FIELD_ID_INCIDENT_HEADER = 1;

status_t write_header_section(int fd, const uint8_t* buf, size_t bufSize) {
    status_t err;

    if (bufSize == 0) {
        return NO_ERROR;
    }

    err = write_section_header(fd, FIELD_ID_INCIDENT_HEADER, bufSize);
    if (err != NO_ERROR) {
        return err;
    }

    err = WriteFully(fd, buf, bufSize);
    if (err != NO_ERROR) {
        return err;
    }

    return NO_ERROR;
}

status_t write_section_header(int fd, int sectionId, size_t size) {
    uint8_t buf[20];
    uint8_t* p = write_length_delimited_tag_header(buf, sectionId, size);
    return WriteFully(fd, buf, p - buf) ? NO_ERROR : -errno;
}

status_t write_section(int fd, int sectionId, const MessageLite& message) {
    status_t err;

    err = write_section_header(fd, sectionId, message.ByteSize());
    if (err != NO_ERROR) {
        return err;
    }

    FileOutputStream stream(fd);
    if (!message.SerializeToZeroCopyStream(&stream)) {
        return stream.GetErrno();
    } else {
        return NO_ERROR;
    }
}



}  // namespace incidentd
}  // namespace os
}  // namespace android