1 /*
2  * Copyright (c) 2023-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 
16 #include <dirent.h>
17 #include <errno.h>
18 #include <unistd.h>
19 #include <sys/types.h>
20 #include <stdio.h>
21 #include <sys/stat.h>
22 #include <stdbool.h>
23 
24 #include <securec.h>
25 
26 /**
27  * This bin file is used in l1 device triggered when device initializing, in order to support the compatibility for old
28  * keys with version 1.
29 */
30 
31 #define HUKS_SERVICE_UID 12
32 #define DIR_TYPE 4
33 #define DEFAULT_PATH_LEN 1024
34 #define DEFAULT_READ_BUFFER 4096
35 #define DEFAULT_HUKS_PATH_PERMISSION 0700
36 
ChangeDirAndFilesPerm(const char * path)37 static void ChangeDirAndFilesPerm(const char *path)
38 {
39     DIR *dir;
40     struct dirent *ptr;
41     dir = opendir(path);
42     if (dir == NULL) {
43         return;
44     }
45     int ret = EOK;
46     while ((ptr = readdir(dir)) != NULL) {
47         if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0) {
48             continue;
49         }
50         char curPath[DEFAULT_PATH_LEN] = { 0 };
51 
52         ret = strcpy_s(curPath, DEFAULT_PATH_LEN, path);
53         if (ret != EOK) {
54             break;
55         }
56         ret = strcat_s(curPath, DEFAULT_PATH_LEN, "/");
57         if (ret != EOK) {
58             break;
59         }
60         ret = strcat_s(curPath, DEFAULT_PATH_LEN, ptr->d_name);
61         if (ret != EOK) {
62             break;
63         }
64 
65         ret = chown(curPath, HUKS_SERVICE_UID, HUKS_SERVICE_UID);
66         if (ret != EOK) {
67             break;
68         }
69         if ((ptr->d_type != DIR_TYPE)) {
70             if (chmod(curPath, S_IRUSR | S_IWUSR) < 0) {
71                 break;
72             }
73         } else {
74             if (chmod(curPath, S_IRWXU) < 0) {
75                 break;
76             }
77             ChangeDirAndFilesPerm(curPath);
78         }
79     }
80     if (ret != EOK) {
81         printf("chmod dir and file failed! errno = 0x%x \n", errno);
82     }
83     (void)closedir(dir);
84 }
85 
MoveOldFileToNew(const char * srcPath,const char * tarPath)86 static void MoveOldFileToNew(const char *srcPath, const char *tarPath)
87 {
88     char buffer[DEFAULT_READ_BUFFER];
89     FILE *in, *out;
90     if ((in = fopen(srcPath, "r")) == NULL) {
91         printf("open source file failed !\n");
92         return;
93     }
94     if ((out = fopen(tarPath, "w")) == NULL) {
95         printf("open target file failed !\n");
96         (void)fclose(in);
97         return;
98     }
99     int len;
100     while ((len = fread(buffer, 1, DEFAULT_READ_BUFFER, in)) > 0) {
101         int size = fwrite(buffer, 1, len, out);
102         if (size != len) {
103             printf("move old file to new path failed!");
104             (void)fclose(out);
105             (void)fclose(in);
106             return;
107         }
108     }
109     (void)fclose(out);
110     (void)fclose(in);
111 
112     (void)remove(srcPath);
113 }
114 
ConstructSrcAndTargetPath(char * curPath,char * desPath,struct dirent * ptr,const char * srcPath,const char * tarPath)115 static int ConstructSrcAndTargetPath(char *curPath, char *desPath, struct dirent *ptr,
116     const char *srcPath, const char *tarPath)
117 {
118     int ret = strcpy_s(curPath, DEFAULT_PATH_LEN, srcPath);
119     if (ret != EOK) {
120         return ret;
121     }
122     ret = strcat_s(curPath, DEFAULT_PATH_LEN, "/");
123     if (ret != EOK) {
124         return ret;
125     }
126 
127     ret = strcat_s(curPath, DEFAULT_PATH_LEN, ptr->d_name);
128     if (ret != EOK) {
129         return ret;
130     }
131 
132     ret = strcpy_s(desPath, DEFAULT_PATH_LEN, tarPath);
133     if (ret != EOK) {
134         return ret;
135     }
136     ret = strcat_s(desPath, DEFAULT_PATH_LEN, "/");
137     if (ret != EOK) {
138         return ret;
139     }
140 
141     ret = strcat_s(desPath, DEFAULT_PATH_LEN, ptr->d_name);
142     if (ret != EOK) {
143         return ret;
144     }
145     return EOK;
146 }
147 
MoveOldFolderToNew(const char * srcPath,const char * tarPath)148 static void MoveOldFolderToNew(const char *srcPath, const char *tarPath)
149 {
150     if (!opendir(tarPath)) {
151         if (mkdir(tarPath, DEFAULT_HUKS_PATH_PERMISSION) != 0) {
152             printf("mkdir failed! errno = 0x%x \n", errno);
153             return;
154         }
155     }
156     struct dirent *ptr;
157     DIR *dir = opendir(srcPath);
158     int ret = EOK;
159     while ((ptr = readdir(dir)) != NULL) {
160         if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0) {
161             continue;
162         }
163         char curPath[DEFAULT_PATH_LEN] = { 0 };
164         char desPath[DEFAULT_PATH_LEN] = { 0 };
165 
166         ret = ConstructSrcAndTargetPath(curPath, desPath, ptr, srcPath, tarPath);
167         if (ret != EOK) {
168             printf("construct src and target path failed!");
169             break;
170         }
171         if (ptr->d_type == DIR_TYPE) {
172             MoveOldFolderToNew(curPath, desPath);
173         } else {
174             MoveOldFileToNew(curPath, desPath);
175         }
176     }
177     (void)closedir(dir);
178     if (ret != EOK) {
179         printf("chmod dir and file failed! errno = 0x%x \n", errno);
180     }
181     (void)rmdir(srcPath);
182 }
183 
main(void)184 int main(void)
185 {
186     const char *oldPath = "/storage/maindata";
187     const char *newPath = "/storage/data/service/el1/public/huks_service/maindata";
188 
189     // move directories and files form old path to new path
190     MoveOldFolderToNew(oldPath, newPath);
191 
192     // change directories and files permission
193     ChangeDirAndFilesPerm(newPath);
194 }
195