1 /*
2 * Copyright (c) 2013, The Linux Foundation. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met:
7 * * Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * * Redistributions in binary form must reproduce the above
10 * copyright notice, this list of conditions and the following
11 * disclaimer in the documentation and/or other materials provided
12 * with the distribution.
13 * * Neither the name of The Linux Foundation nor the names of its
14 * contributors may be used to endorse or promote products derived
15 * from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED
18 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
21 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
24 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
25 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
26 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include "edify/expr.h"
34 #include "dec.h"
35 #include "gpt-utils.h"
36
DecryptFn(const char * name,State * state,int argc,Expr * argv[])37 Value* DecryptFn(const char* name, State* state, int argc, Expr* argv[]) {
38 int rc = -1;
39 char *src_file, *dst_file;
40
41 if (argc != 2)
42 return ErrorAbort(state, "%s expects 2 args, got %d", name, argc);
43
44 if (ReadArgs(state, argv, 2, &src_file, &dst_file))
45 return NULL;
46
47 rc = decrypt_image(src_file, dst_file);
48
49 free(src_file);
50 free(dst_file);
51
52 return StringValue(strdup(rc >= 0 ? "t" : ""));
53 }
54
BootUpdateFn(const char * name,State * state,int argc,Expr * argv[])55 Value* BootUpdateFn(const char* name, State* state, int argc, Expr* argv[])
56 {
57 int rc = 0;
58 char *stageStr;
59 enum boot_update_stage stage;
60
61 if (argc != 1)
62 return ErrorAbort(state, "%s() expects 1 args, got %d", name, argc);
63
64 if (ReadArgs(state, argv, 1, &stageStr))
65 return NULL;
66
67 if (!strcmp(stageStr, "main"))
68 stage = UPDATE_MAIN;
69 else if (!strcmp(stageStr, "backup"))
70 stage = UPDATE_BACKUP;
71 else if (!strcmp(stageStr, "finalize"))
72 stage = UPDATE_FINALIZE;
73 else {
74 fprintf(stderr, "Unrecognized boot update stage, exitting\n");
75 rc = -1;
76 }
77
78 free(stageStr);
79
80 if (!rc)
81 rc = prepare_boot_update(stage);
82
83 return StringValue(strdup(rc ? "" : "t"));
84 }
85
Register_librecovery_updater_msm()86 void Register_librecovery_updater_msm() {
87 RegisterFunction("msm.decrypt", DecryptFn);
88 RegisterFunction("msm.boot_update", BootUpdateFn);
89 }
90