File: cached_metadata.cc

package info (click to toggle)
chromium 138.0.7204.183-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,908 kB
  • sloc: cpp: 34,937,088; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (107 lines) | stat: -rw-r--r-- 3,625 bytes parent folder | download | duplicates (5)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "third_party/blink/renderer/platform/loader/fetch/cached_metadata.h"

#include <utility>
#include <variant>

#include "base/memory/scoped_refptr.h"
#include "base/numerics/safe_conversions.h"
#include "third_party/blink/renderer/platform/loader/fetch/url_loader/cached_metadata_handler.h"

namespace blink {

namespace {

template <typename DataType>
bool CheckSizeAndMarker(const DataType& data) {
  // Ensure the data is big enough.
  if (data.size() <= sizeof(CachedMetadataHeader)) {
    return false;
  }
  // Ensure the marker matches.
  if (reinterpret_cast<const CachedMetadataHeader*>(data.data())->marker !=
      CachedMetadataHandler::kSingleEntryWithTag) {
    return false;
  }
  return true;
}

Vector<uint8_t> GetSerializedData(uint32_t data_type_id,
                                  base::span<const uint8_t> data,
                                  uint64_t tag) {
  // Don't allow an ID of 0, it is used internally to indicate errors.
  DCHECK(data_type_id);
  DCHECK(data.data());

  Vector<uint8_t> vector =
      CachedMetadata::GetSerializedDataHeader(data_type_id, data.size(), tag);
  vector.AppendSpan(data);
  return vector;
}

}  // namespace

scoped_refptr<CachedMetadata> CachedMetadata::Create(
    uint32_t data_type_id,
    base::span<const uint8_t> data,
    uint64_t tag) {
  return base::MakeRefCounted<CachedMetadata>(data_type_id, data, tag,
                                              base::PassKey<CachedMetadata>());
}

scoped_refptr<CachedMetadata> CachedMetadata::CreateFromSerializedData(
    Vector<uint8_t> data) {
  if (!CheckSizeAndMarker(data)) {
    return nullptr;
  }
  return base::MakeRefCounted<CachedMetadata>(std::move(data),
                                              base::PassKey<CachedMetadata>());
}

scoped_refptr<CachedMetadata> CachedMetadata::CreateFromSerializedData(
    mojo_base::BigBuffer& data,
    uint32_t offset) {
  if (data.size() < offset ||
      !CheckSizeAndMarker(base::as_byte_span(data).subspan(offset))) {
    return nullptr;
  }
  return base::MakeRefCounted<CachedMetadata>(std::move(data), offset,
                                              base::PassKey<CachedMetadata>());
}

CachedMetadata::CachedMetadata(Vector<uint8_t> data,
                               base::PassKey<CachedMetadata>)
    : buffer_(std::move(data)) {}

CachedMetadata::CachedMetadata(uint32_t data_type_id,
                               base::span<const uint8_t> data,
                               uint64_t tag,
                               base::PassKey<CachedMetadata>)
    : buffer_(GetSerializedData(data_type_id, data, tag)) {}

CachedMetadata::CachedMetadata(mojo_base::BigBuffer data,
                               uint32_t offset,
                               base::PassKey<CachedMetadata>)
    : buffer_(std::move(data)), offset_(offset) {}

base::span<const uint8_t> CachedMetadata::SerializedData() const {
  base::span<const uint8_t> span_including_offset;
  if (std::holds_alternative<Vector<uint8_t>>(buffer_)) {
    span_including_offset = std::get<Vector<uint8_t>>(buffer_);
  } else {
    CHECK(std::holds_alternative<mojo_base::BigBuffer>(buffer_));
    span_including_offset = std::get<mojo_base::BigBuffer>(buffer_);
  }
  CHECK_GE(span_including_offset.size(), offset_);
  return span_including_offset.subspan(offset_);
}

std::variant<Vector<uint8_t>, mojo_base::BigBuffer>
CachedMetadata::DrainSerializedData() && {
  return std::move(buffer_);
}

}  // namespace blink