huangcm
2025-02-24 69ed55dec4b2116a19e4cca4393cbc014fce5fb2
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
/*
 * Copyright 2015 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 <keymaster/km_openssl/kdf.h>
 
namespace keymaster {
 
Kdf::Kdf() : is_initialized_(false) {}
 
bool Kdf::Init(keymaster_digest_t digest_type, const uint8_t* secret, size_t secret_len,
               const uint8_t* salt, size_t salt_len) {
    is_initialized_ = false;
 
    switch (digest_type) {
    case KM_DIGEST_SHA1:
        digest_size_ = 20;
        digest_type_ = digest_type;
        break;
    case KM_DIGEST_SHA_2_256:
        digest_size_ = 32;
        digest_type_ = digest_type;
        break;
    default:
        return false;
    }
 
    if (!secret || secret_len == 0)
        return false;
 
    secret_key_len_ = secret_len;
    secret_key_.reset(dup_buffer(secret, secret_len));
    if (!secret_key_.get())
        return false;
 
    salt_len_ = salt_len;
    if (salt && salt_len > 0) {
        salt_.reset(dup_buffer(salt, salt_len));
        if (!salt_.get())
            return false;
    } else {
        salt_.reset();
    }
 
    is_initialized_ = true;
    return true;
}
 
bool Kdf::Uint32ToBigEndianByteArray(uint32_t number, uint8_t* output) {
    if (!output)
        return false;
 
    output[0] = (number >> 24) & 0xff;
    output[1] = (number >> 16) & 0xff;
    output[2] = (number >> 8) & 0xff;
    output[3] = (number)&0xff;
    return true;
}
 
}  // namespace keymaster