1/*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15#include "udid.h"
16#ifndef OHOS_LITE
17#include "init_module_engine.h"
18#endif
19#include "init_param.h"
20#include "param_comm.h"
21#include "securec.h"
22#include "sysparam_errno.h"
23
24#include "mbedtls/sha256.h"
25
26#define BUF_MAX_LEN 132
27#define PARAM_VALUE_MAX_LEN 36
28
29INIT_UDID_LOCAL_API int GetSha256Value(const char *input, char *udid, uint32_t udidSize)
30{
31    if (input == NULL) {
32        return EC_FAILURE;
33    }
34    char buf[DEV_BUF_LENGTH] = { 0 };
35    unsigned char hash[HASH_LENGTH] = { 0 };
36
37    mbedtls_sha256_context context;
38    mbedtls_sha256_init(&context);
39    mbedtls_sha256_starts(&context, 0);
40    mbedtls_sha256_update(&context, (const unsigned char *)input, strlen(input));
41    mbedtls_sha256_finish(&context, hash);
42
43    for (size_t i = 0; i < HASH_LENGTH; i++) {
44        unsigned char value = hash[i];
45        memset_s(buf, DEV_BUF_LENGTH, 0, DEV_BUF_LENGTH);
46        int len = sprintf_s(buf, sizeof(buf), "%02X", value);
47        if (len > 0 && strcat_s(udid, udidSize, buf) != 0) {
48            return EC_FAILURE;
49        }
50    }
51    return EC_SUCCESS;
52}
53
54INIT_UDID_LOCAL_API int CalcDevUdid(char *udid, uint32_t size)
55{
56    char tmp[BUF_MAX_LEN] = {0};
57    uint32_t manufactureLen = PARAM_VALUE_MAX_LEN;
58    int ret = SystemReadParam("const.product.manufacturer", tmp, &manufactureLen);
59    BEGET_ERROR_CHECK(ret == 0, return -1, "Read param const.product.manufacturer failed!");
60
61    uint32_t modelLen = PARAM_VALUE_MAX_LEN;
62    ret = SystemReadParam("const.product.model", tmp + manufactureLen, &modelLen);
63    BEGET_ERROR_CHECK(ret == 0, return -1, "Read param const.product.model failed!");
64    const char *serial = GetSerial_();
65    BEGET_ERROR_CHECK(serial != NULL, return -1, "Read param serial failed!");
66    ret = strcat_s(tmp, BUF_MAX_LEN, serial);
67    BEGET_ERROR_CHECK(ret != -1, return -1, "Cat serial failed!");
68    ret = GetSha256Value(tmp, udid, size);
69    return ret;
70}
71
72#ifndef OHOS_LITE
73INIT_UDID_LOCAL_API void SetDevUdid()
74{
75    BEGET_LOGI("Begin set udid param");
76    char udid[UDID_LEN] = {0};
77    uint32_t size = (uint32_t)sizeof(udid);
78    int ret = GetUdidFromParam(udid, size);
79    if (ret != 0) {
80        BEGET_LOGI("Get udid from param failed, calculate udid from other param");
81        ret = CalcDevUdid(udid, size);
82        BEGET_ERROR_CHECK(ret == 0, return, "calculate udid is failed!")
83    }
84    ret = SystemWriteParam("const.product.devUdid", udid);
85    BEGET_ERROR_CHECK(ret == 0, return, "write param const.product.devUdid failed!");
86}
87
88MODULE_CONSTRUCTOR(void)
89{
90    SetDevUdid();
91}
92#endif
93