xref: /aosp_15_r20/external/tink/cc/util/keyset_util.cc (revision e7b1675dde1b92d52ec075b0a92829627f2c52a5)
1*e7b1675dSTing-Kang Chang // Copyright 2018 Google Inc.
2*e7b1675dSTing-Kang Chang //
3*e7b1675dSTing-Kang Chang // Licensed under the Apache License, Version 2.0 (the "License");
4*e7b1675dSTing-Kang Chang // you may not use this file except in compliance with the License.
5*e7b1675dSTing-Kang Chang // You may obtain a copy of the License at
6*e7b1675dSTing-Kang Chang //
7*e7b1675dSTing-Kang Chang //     http://www.apache.org/licenses/LICENSE-2.0
8*e7b1675dSTing-Kang Chang //
9*e7b1675dSTing-Kang Chang // Unless required by applicable law or agreed to in writing, software
10*e7b1675dSTing-Kang Chang // distributed under the License is distributed on an "AS IS" BASIS,
11*e7b1675dSTing-Kang Chang // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*e7b1675dSTing-Kang Chang // See the License for the specific language governing permissions and
13*e7b1675dSTing-Kang Chang // limitations under the License.
14*e7b1675dSTing-Kang Chang //
15*e7b1675dSTing-Kang Chang ///////////////////////////////////////////////////////////////////////////////
16*e7b1675dSTing-Kang Chang 
17*e7b1675dSTing-Kang Chang #include "tink/util/keyset_util.h"
18*e7b1675dSTing-Kang Chang 
19*e7b1675dSTing-Kang Chang #include <cstdint>
20*e7b1675dSTing-Kang Chang #include <random>
21*e7b1675dSTing-Kang Chang 
22*e7b1675dSTing-Kang Chang #include "proto/tink.pb.h"
23*e7b1675dSTing-Kang Chang 
24*e7b1675dSTing-Kang Chang namespace crypto {
25*e7b1675dSTing-Kang Chang namespace tink {
26*e7b1675dSTing-Kang Chang 
27*e7b1675dSTing-Kang Chang namespace {
28*e7b1675dSTing-Kang Chang 
29*e7b1675dSTing-Kang Chang using google::crypto::tink::Keyset;
30*e7b1675dSTing-Kang Chang 
NewKeyId()31*e7b1675dSTing-Kang Chang uint32_t NewKeyId() {
32*e7b1675dSTing-Kang Chang   std::random_device rd;
33*e7b1675dSTing-Kang Chang   std::minstd_rand0 gen(rd());
34*e7b1675dSTing-Kang Chang   std::uniform_int_distribution<uint32_t> dist;
35*e7b1675dSTing-Kang Chang   return dist(gen);
36*e7b1675dSTing-Kang Chang }
37*e7b1675dSTing-Kang Chang 
38*e7b1675dSTing-Kang Chang }  // namespace
39*e7b1675dSTing-Kang Chang 
GenerateUnusedKeyId(const Keyset & keyset)40*e7b1675dSTing-Kang Chang uint32_t GenerateUnusedKeyId(const Keyset& keyset) {
41*e7b1675dSTing-Kang Chang   while (true) {
42*e7b1675dSTing-Kang Chang     uint32_t key_id = NewKeyId();
43*e7b1675dSTing-Kang Chang     bool already_exists = false;
44*e7b1675dSTing-Kang Chang     for (auto& key : keyset.key()) {
45*e7b1675dSTing-Kang Chang       if (key.key_id() == key_id) {
46*e7b1675dSTing-Kang Chang         already_exists = true;
47*e7b1675dSTing-Kang Chang         break;
48*e7b1675dSTing-Kang Chang       }
49*e7b1675dSTing-Kang Chang     }
50*e7b1675dSTing-Kang Chang     if (!already_exists) return key_id;
51*e7b1675dSTing-Kang Chang   }
52*e7b1675dSTing-Kang Chang }
53*e7b1675dSTing-Kang Chang 
54*e7b1675dSTing-Kang Chang }  // namespace tink
55*e7b1675dSTing-Kang Chang }  // namespace crypto
56