• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2009 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include <ctype.h>
18 #include <errno.h>
19 #include <stdarg.h>
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <string.h>
23 #include <sys/mount.h>
24 #include <sys/stat.h>
25 #include <sys/types.h>
26 #include <sys/wait.h>
27 #include <unistd.h>
28 
29 #include "cutils/misc.h"
30 #include "cutils/properties.h"
31 #include "edify/expr.h"
32 #include "minzip/DirUtil.h"
33 #include "mtdutils/mounts.h"
34 #include "mtdutils/mtdutils.h"
35 #include "updater.h"
36 
37 
38 // mount(type, location, mount_point)
39 //
40 //   what:  type="MTD"   location="<partition>"            to mount a yaffs2 filesystem
41 //          type="vfat"  location="/dev/block/<whatever>"  to mount a device
MountFn(const char * name,State * state,int argc,Expr * argv[])42 char* MountFn(const char* name, State* state, int argc, Expr* argv[]) {
43     char* result = NULL;
44     if (argc != 3) {
45         return ErrorAbort(state, "%s() expects 3 args, got %d", name, argc);
46     }
47     char* type;
48     char* location;
49     char* mount_point;
50     if (ReadArgs(state, argv, 3, &type, &location, &mount_point) < 0) {
51         return NULL;
52     }
53 
54     if (strlen(type) == 0) {
55         ErrorAbort(state, "type argument to %s() can't be empty", name);
56         goto done;
57     }
58     if (strlen(location) == 0) {
59         ErrorAbort(state, "location argument to %s() can't be empty", name);
60         goto done;
61     }
62     if (strlen(mount_point) == 0) {
63         ErrorAbort(state, "mount_point argument to %s() can't be empty", name);
64         goto done;
65     }
66 
67     mkdir(mount_point, 0755);
68 
69     if (strcmp(type, "MTD") == 0) {
70         mtd_scan_partitions();
71         const MtdPartition* mtd;
72         mtd = mtd_find_partition_by_name(location);
73         if (mtd == NULL) {
74             fprintf(stderr, "%s: no mtd partition named \"%s\"",
75                     name, location);
76             result = strdup("");
77             goto done;
78         }
79         if (mtd_mount_partition(mtd, mount_point, "yaffs2", 0 /* rw */) != 0) {
80             fprintf(stderr, "mtd mount of %s failed: %s\n",
81                     location, strerror(errno));
82             result = strdup("");
83             goto done;
84         }
85         result = mount_point;
86     } else {
87         if (mount(location, mount_point, type,
88                   MS_NOATIME | MS_NODEV | MS_NODIRATIME, "") < 0) {
89             fprintf(stderr, "%s: failed to mount %s at %s: %s\n",
90                     name, location, mount_point, strerror(errno));
91             result = strdup("");
92         } else {
93             result = mount_point;
94         }
95     }
96 
97 done:
98     free(type);
99     free(location);
100     if (result != mount_point) free(mount_point);
101     return result;
102 }
103 
104 
105 // is_mounted(mount_point)
IsMountedFn(const char * name,State * state,int argc,Expr * argv[])106 char* IsMountedFn(const char* name, State* state, int argc, Expr* argv[]) {
107     char* result = NULL;
108     if (argc != 1) {
109         return ErrorAbort(state, "%s() expects 1 arg, got %d", name, argc);
110     }
111     char* mount_point;
112     if (ReadArgs(state, argv, 1, &mount_point) < 0) {
113         return NULL;
114     }
115     if (strlen(mount_point) == 0) {
116         ErrorAbort(state, "mount_point argument to unmount() can't be empty");
117         goto done;
118     }
119 
120     scan_mounted_volumes();
121     const MountedVolume* vol = find_mounted_volume_by_mount_point(mount_point);
122     if (vol == NULL) {
123         result = strdup("");
124     } else {
125         result = mount_point;
126     }
127 
128 done:
129     if (result != mount_point) free(mount_point);
130     return result;
131 }
132 
133 
UnmountFn(const char * name,State * state,int argc,Expr * argv[])134 char* UnmountFn(const char* name, State* state, int argc, Expr* argv[]) {
135     char* result = NULL;
136     if (argc != 1) {
137         return ErrorAbort(state, "%s() expects 1 arg, got %d", name, argc);
138     }
139     char* mount_point;
140     if (ReadArgs(state, argv, 1, &mount_point) < 0) {
141         return NULL;
142     }
143     if (strlen(mount_point) == 0) {
144         ErrorAbort(state, "mount_point argument to unmount() can't be empty");
145         goto done;
146     }
147 
148     scan_mounted_volumes();
149     const MountedVolume* vol = find_mounted_volume_by_mount_point(mount_point);
150     if (vol == NULL) {
151         fprintf(stderr, "unmount of %s failed; no such volume\n", mount_point);
152         result = strdup("");
153     } else {
154         unmount_mounted_volume(vol);
155         result = mount_point;
156     }
157 
158 done:
159     if (result != mount_point) free(mount_point);
160     return result;
161 }
162 
163 
164 // format(type, location)
165 //
166 //    type="MTD"  location=partition
FormatFn(const char * name,State * state,int argc,Expr * argv[])167 char* FormatFn(const char* name, State* state, int argc, Expr* argv[]) {
168     char* result = NULL;
169     if (argc != 2) {
170         return ErrorAbort(state, "%s() expects 2 args, got %d", name, argc);
171     }
172     char* type;
173     char* location;
174     if (ReadArgs(state, argv, 2, &type, &location) < 0) {
175         return NULL;
176     }
177 
178     if (strlen(type) == 0) {
179         ErrorAbort(state, "type argument to %s() can't be empty", name);
180         goto done;
181     }
182     if (strlen(location) == 0) {
183         ErrorAbort(state, "location argument to %s() can't be empty", name);
184         goto done;
185     }
186 
187     if (strcmp(type, "MTD") == 0) {
188         mtd_scan_partitions();
189         const MtdPartition* mtd = mtd_find_partition_by_name(location);
190         if (mtd == NULL) {
191             fprintf(stderr, "%s: no mtd partition named \"%s\"",
192                     name, location);
193             result = strdup("");
194             goto done;
195         }
196         MtdWriteContext* ctx = mtd_write_partition(mtd);
197         if (ctx == NULL) {
198             fprintf(stderr, "%s: can't write \"%s\"", name, location);
199             result = strdup("");
200             goto done;
201         }
202         if (mtd_erase_blocks(ctx, -1) == -1) {
203             mtd_write_close(ctx);
204             fprintf(stderr, "%s: failed to erase \"%s\"", name, location);
205             result = strdup("");
206             goto done;
207         }
208         if (mtd_write_close(ctx) != 0) {
209             fprintf(stderr, "%s: failed to close \"%s\"", name, location);
210             result = strdup("");
211             goto done;
212         }
213         result = location;
214     } else {
215         fprintf(stderr, "%s: unsupported type \"%s\"", name, type);
216     }
217 
218 done:
219     free(type);
220     if (result != location) free(location);
221     return result;
222 }
223 
224 
DeleteFn(const char * name,State * state,int argc,Expr * argv[])225 char* DeleteFn(const char* name, State* state, int argc, Expr* argv[]) {
226     char** paths = malloc(argc * sizeof(char*));
227     int i;
228     for (i = 0; i < argc; ++i) {
229         paths[i] = Evaluate(state, argv[i]);
230         if (paths[i] == NULL) {
231             int j;
232             for (j = 0; j < i; ++i) {
233                 free(paths[j]);
234             }
235             free(paths);
236             return NULL;
237         }
238     }
239 
240     bool recursive = (strcmp(name, "delete_recursive") == 0);
241 
242     int success = 0;
243     for (i = 0; i < argc; ++i) {
244         if ((recursive ? dirUnlinkHierarchy(paths[i]) : unlink(paths[i])) == 0)
245             ++success;
246         free(paths[i]);
247     }
248     free(paths);
249 
250     char buffer[10];
251     sprintf(buffer, "%d", success);
252     return strdup(buffer);
253 }
254 
255 
ShowProgressFn(const char * name,State * state,int argc,Expr * argv[])256 char* ShowProgressFn(const char* name, State* state, int argc, Expr* argv[]) {
257     if (argc != 2) {
258         return ErrorAbort(state, "%s() expects 2 args, got %d", name, argc);
259     }
260     char* frac_str;
261     char* sec_str;
262     if (ReadArgs(state, argv, 2, &frac_str, &sec_str) < 0) {
263         return NULL;
264     }
265 
266     double frac = strtod(frac_str, NULL);
267     int sec = strtol(sec_str, NULL, 10);
268 
269     UpdaterInfo* ui = (UpdaterInfo*)(state->cookie);
270     fprintf(ui->cmd_pipe, "progress %f %d\n", frac, sec);
271 
272     free(sec_str);
273     return frac_str;
274 }
275 
SetProgressFn(const char * name,State * state,int argc,Expr * argv[])276 char* SetProgressFn(const char* name, State* state, int argc, Expr* argv[]) {
277     if (argc != 1) {
278         return ErrorAbort(state, "%s() expects 1 arg, got %d", name, argc);
279     }
280     char* frac_str;
281     if (ReadArgs(state, argv, 1, &frac_str) < 0) {
282         return NULL;
283     }
284 
285     double frac = strtod(frac_str, NULL);
286 
287     UpdaterInfo* ui = (UpdaterInfo*)(state->cookie);
288     fprintf(ui->cmd_pipe, "set_progress %f\n", frac);
289 
290     return frac_str;
291 }
292 
293 // package_extract_dir(package_path, destination_path)
PackageExtractDirFn(const char * name,State * state,int argc,Expr * argv[])294 char* PackageExtractDirFn(const char* name, State* state,
295                           int argc, Expr* argv[]) {
296     if (argc != 2) {
297         return ErrorAbort(state, "%s() expects 2 args, got %d", name, argc);
298     }
299     char* zip_path;
300     char* dest_path;
301     if (ReadArgs(state, argv, 2, &zip_path, &dest_path) < 0) return NULL;
302 
303     ZipArchive* za = ((UpdaterInfo*)(state->cookie))->package_zip;
304 
305     // To create a consistent system image, never use the clock for timestamps.
306     struct utimbuf timestamp = { 1217592000, 1217592000 };  // 8/1/2008 default
307 
308     bool success = mzExtractRecursive(za, zip_path, dest_path,
309                                       MZ_EXTRACT_FILES_ONLY, &timestamp,
310                                       NULL, NULL);
311     free(zip_path);
312     free(dest_path);
313     return strdup(success ? "t" : "");
314 }
315 
316 
317 // package_extract_file(package_path, destination_path)
PackageExtractFileFn(const char * name,State * state,int argc,Expr * argv[])318 char* PackageExtractFileFn(const char* name, State* state,
319                            int argc, Expr* argv[]) {
320     if (argc != 2) {
321         return ErrorAbort(state, "%s() expects 2 args, got %d", name, argc);
322     }
323     char* zip_path;
324     char* dest_path;
325     if (ReadArgs(state, argv, 2, &zip_path, &dest_path) < 0) return NULL;
326 
327     bool success = false;
328 
329     ZipArchive* za = ((UpdaterInfo*)(state->cookie))->package_zip;
330     const ZipEntry* entry = mzFindZipEntry(za, zip_path);
331     if (entry == NULL) {
332         fprintf(stderr, "%s: no %s in package\n", name, zip_path);
333         goto done;
334     }
335 
336     FILE* f = fopen(dest_path, "wb");
337     if (f == NULL) {
338         fprintf(stderr, "%s: can't open %s for write: %s\n",
339                 name, dest_path, strerror(errno));
340         goto done;
341     }
342     success = mzExtractZipEntryToFile(za, entry, fileno(f));
343     fclose(f);
344 
345   done:
346     free(zip_path);
347     free(dest_path);
348     return strdup(success ? "t" : "");
349 }
350 
351 
352 // symlink target src1 src2 ...
353 //    unlinks any previously existing src1, src2, etc before creating symlinks.
SymlinkFn(const char * name,State * state,int argc,Expr * argv[])354 char* SymlinkFn(const char* name, State* state, int argc, Expr* argv[]) {
355     if (argc == 0) {
356         return ErrorAbort(state, "%s() expects 1+ args, got %d", name, argc);
357     }
358     char* target;
359     target = Evaluate(state, argv[0]);
360     if (target == NULL) return NULL;
361 
362     char** srcs = ReadVarArgs(state, argc-1, argv+1);
363     if (srcs == NULL) {
364         free(target);
365         return NULL;
366     }
367 
368     int i;
369     for (i = 0; i < argc-1; ++i) {
370         if (unlink(srcs[i]) < 0) {
371             if (errno != ENOENT) {
372                 fprintf(stderr, "%s: failed to remove %s: %s\n",
373                         name, srcs[i], strerror(errno));
374             }
375         }
376         if (symlink(target, srcs[i]) < 0) {
377             fprintf(stderr, "%s: failed to symlink %s to %s: %s\n",
378                     name, srcs[i], target, strerror(errno));
379         }
380         free(srcs[i]);
381     }
382     free(srcs);
383     return strdup("");
384 }
385 
386 
SetPermFn(const char * name,State * state,int argc,Expr * argv[])387 char* SetPermFn(const char* name, State* state, int argc, Expr* argv[]) {
388     char* result = NULL;
389     bool recursive = (strcmp(name, "set_perm_recursive") == 0);
390 
391     int min_args = 4 + (recursive ? 1 : 0);
392     if (argc < min_args) {
393         return ErrorAbort(state, "%s() expects %d+ args, got %d", name, argc);
394     }
395 
396     char** args = ReadVarArgs(state, argc, argv);
397     if (args == NULL) return NULL;
398 
399     char* end;
400     int i;
401 
402     int uid = strtoul(args[0], &end, 0);
403     if (*end != '\0' || args[0][0] == 0) {
404         ErrorAbort(state, "%s: \"%s\" not a valid uid", name, args[0]);
405         goto done;
406     }
407 
408     int gid = strtoul(args[1], &end, 0);
409     if (*end != '\0' || args[1][0] == 0) {
410         ErrorAbort(state, "%s: \"%s\" not a valid gid", name, args[1]);
411         goto done;
412     }
413 
414     if (recursive) {
415         int dir_mode = strtoul(args[2], &end, 0);
416         if (*end != '\0' || args[2][0] == 0) {
417             ErrorAbort(state, "%s: \"%s\" not a valid dirmode", name, args[2]);
418             goto done;
419         }
420 
421         int file_mode = strtoul(args[3], &end, 0);
422         if (*end != '\0' || args[3][0] == 0) {
423             ErrorAbort(state, "%s: \"%s\" not a valid filemode",
424                        name, args[3]);
425             goto done;
426         }
427 
428         for (i = 4; i < argc; ++i) {
429             dirSetHierarchyPermissions(args[i], uid, gid, dir_mode, file_mode);
430         }
431     } else {
432         int mode = strtoul(args[2], &end, 0);
433         if (*end != '\0' || args[2][0] == 0) {
434             ErrorAbort(state, "%s: \"%s\" not a valid mode", name, args[2]);
435             goto done;
436         }
437 
438         for (i = 3; i < argc; ++i) {
439             if (chown(args[i], uid, gid) < 0) {
440                 fprintf(stderr, "%s: chown of %s to %d %d failed: %s\n",
441                         name, args[i], uid, gid, strerror(errno));
442             }
443             if (chmod(args[i], mode) < 0) {
444                 fprintf(stderr, "%s: chmod of %s to %o failed: %s\n",
445                         name, args[i], mode, strerror(errno));
446             }
447         }
448     }
449     result = strdup("");
450 
451 done:
452     for (i = 0; i < argc; ++i) {
453         free(args[i]);
454     }
455     free(args);
456 
457     return result;
458 }
459 
460 
GetPropFn(const char * name,State * state,int argc,Expr * argv[])461 char* GetPropFn(const char* name, State* state, int argc, Expr* argv[]) {
462     if (argc != 1) {
463         return ErrorAbort(state, "%s() expects 1 arg, got %d", name, argc);
464     }
465     char* key;
466     key = Evaluate(state, argv[0]);
467     if (key == NULL) return NULL;
468 
469     char value[PROPERTY_VALUE_MAX];
470     property_get(key, value, "");
471     free(key);
472 
473     return strdup(value);
474 }
475 
476 
477 // file_getprop(file, key)
478 //
479 //   interprets 'file' as a getprop-style file (key=value pairs, one
480 //   per line, # comment lines and blank lines okay), and returns the value
481 //   for 'key' (or "" if it isn't defined).
FileGetPropFn(const char * name,State * state,int argc,Expr * argv[])482 char* FileGetPropFn(const char* name, State* state, int argc, Expr* argv[]) {
483     char* result = NULL;
484     char* buffer = NULL;
485     char* filename;
486     char* key;
487     if (ReadArgs(state, argv, 2, &filename, &key) < 0) {
488         return NULL;
489     }
490 
491     struct stat st;
492     if (stat(filename, &st) < 0) {
493         ErrorAbort(state, "%s: failed to stat \"%s\": %s",
494                    name, filename, strerror(errno));
495         goto done;
496     }
497 
498 #define MAX_FILE_GETPROP_SIZE    65536
499 
500     if (st.st_size > MAX_FILE_GETPROP_SIZE) {
501         ErrorAbort(state, "%s too large for %s (max %d)",
502                    filename, name, MAX_FILE_GETPROP_SIZE);
503         goto done;
504     }
505 
506     buffer = malloc(st.st_size+1);
507     if (buffer == NULL) {
508         ErrorAbort(state, "%s: failed to alloc %d bytes", name, st.st_size+1);
509         goto done;
510     }
511 
512     FILE* f = fopen(filename, "rb");
513     if (f == NULL) {
514         ErrorAbort(state, "%s: failed to open %s: %s",
515                    name, filename, strerror(errno));
516         goto done;
517     }
518 
519     if (fread(buffer, 1, st.st_size, f) != st.st_size) {
520         ErrorAbort(state, "%s: failed to read %d bytes from %s",
521                    name, st.st_size+1, filename);
522         fclose(f);
523         goto done;
524     }
525     buffer[st.st_size] = '\0';
526 
527     fclose(f);
528 
529     char* line = strtok(buffer, "\n");
530     do {
531         // skip whitespace at start of line
532         while (*line && isspace(*line)) ++line;
533 
534         // comment or blank line: skip to next line
535         if (*line == '\0' || *line == '#') continue;
536 
537         char* equal = strchr(line, '=');
538         if (equal == NULL) {
539             ErrorAbort(state, "%s: malformed line \"%s\": %s not a prop file?",
540                        name, line, filename);
541             goto done;
542         }
543 
544         // trim whitespace between key and '='
545         char* key_end = equal-1;
546         while (key_end > line && isspace(*key_end)) --key_end;
547         key_end[1] = '\0';
548 
549         // not the key we're looking for
550         if (strcmp(key, line) != 0) continue;
551 
552         // skip whitespace after the '=' to the start of the value
553         char* val_start = equal+1;
554         while(*val_start && isspace(*val_start)) ++val_start;
555 
556         // trim trailing whitespace
557         char* val_end = val_start + strlen(val_start)-1;
558         while (val_end > val_start && isspace(*val_end)) --val_end;
559         val_end[1] = '\0';
560 
561         result = strdup(val_start);
562         break;
563 
564     } while ((line = strtok(NULL, "\n")));
565 
566     if (result == NULL) result = strdup("");
567 
568   done:
569     free(filename);
570     free(key);
571     free(buffer);
572     return result;
573 }
574 
575 
write_raw_image_cb(const unsigned char * data,int data_len,void * ctx)576 static bool write_raw_image_cb(const unsigned char* data,
577                                int data_len, void* ctx) {
578     int r = mtd_write_data((MtdWriteContext*)ctx, (const char *)data, data_len);
579     if (r == data_len) return true;
580     fprintf(stderr, "%s\n", strerror(errno));
581     return false;
582 }
583 
584 // write_raw_image(file, partition)
WriteRawImageFn(const char * name,State * state,int argc,Expr * argv[])585 char* WriteRawImageFn(const char* name, State* state, int argc, Expr* argv[]) {
586     char* result = NULL;
587 
588     char* partition;
589     char* filename;
590     if (ReadArgs(state, argv, 2, &filename, &partition) < 0) {
591         return NULL;
592     }
593 
594     if (strlen(partition) == 0) {
595         ErrorAbort(state, "partition argument to %s can't be empty", name);
596         goto done;
597     }
598     if (strlen(filename) == 0) {
599         ErrorAbort(state, "file argument to %s can't be empty", name);
600         goto done;
601     }
602 
603     mtd_scan_partitions();
604     const MtdPartition* mtd = mtd_find_partition_by_name(partition);
605     if (mtd == NULL) {
606         fprintf(stderr, "%s: no mtd partition named \"%s\"\n", name, partition);
607         result = strdup("");
608         goto done;
609     }
610 
611     MtdWriteContext* ctx = mtd_write_partition(mtd);
612     if (ctx == NULL) {
613         fprintf(stderr, "%s: can't write mtd partition \"%s\"\n",
614                 name, partition);
615         result = strdup("");
616         goto done;
617     }
618 
619     bool success;
620 
621     FILE* f = fopen(filename, "rb");
622     if (f == NULL) {
623         fprintf(stderr, "%s: can't open %s: %s\n",
624                 name, filename, strerror(errno));
625         result = strdup("");
626         goto done;
627     }
628 
629     success = true;
630     char* buffer = malloc(BUFSIZ);
631     int read;
632     while (success && (read = fread(buffer, 1, BUFSIZ, f)) > 0) {
633         int wrote = mtd_write_data(ctx, buffer, read);
634         success = success && (wrote == read);
635         if (!success) {
636             fprintf(stderr, "mtd_write_data to %s failed: %s\n",
637                     partition, strerror(errno));
638         }
639     }
640     free(buffer);
641     fclose(f);
642 
643     if (mtd_erase_blocks(ctx, -1) == -1) {
644         fprintf(stderr, "%s: error erasing blocks of %s\n", name, partition);
645     }
646     if (mtd_write_close(ctx) != 0) {
647         fprintf(stderr, "%s: error closing write of %s\n", name, partition);
648     }
649 
650     printf("%s %s partition from %s\n",
651            success ? "wrote" : "failed to write", partition, filename);
652 
653     result = success ? partition : strdup("");
654 
655 done:
656     if (result != partition) free(partition);
657     free(filename);
658     return result;
659 }
660 
661 // write_firmware_image(file, partition)
662 //
663 //    partition is "radio" or "hboot"
664 //    file is not used until after updater exits
665 //
666 // TODO: this should live in some HTC-specific library
WriteFirmwareImageFn(const char * name,State * state,int argc,Expr * argv[])667 char* WriteFirmwareImageFn(const char* name, State* state,
668                            int argc, Expr* argv[]) {
669     char* result = NULL;
670 
671     char* partition;
672     char* filename;
673     if (ReadArgs(state, argv, 2, &filename, &partition) < 0) {
674         return NULL;
675     }
676 
677     if (strlen(partition) == 0) {
678         ErrorAbort(state, "partition argument to %s can't be empty", name);
679         goto done;
680     }
681     if (strlen(filename) == 0) {
682         ErrorAbort(state, "file argument to %s can't be empty", name);
683         goto done;
684     }
685 
686     FILE* cmd = ((UpdaterInfo*)(state->cookie))->cmd_pipe;
687     fprintf(cmd, "firmware %s %s\n", partition, filename);
688 
689     printf("will write %s firmware from %s\n", partition, filename);
690     result = partition;
691 
692 done:
693     if (result != partition) free(partition);
694     free(filename);
695     return result;
696 }
697 
698 
699 extern int applypatch(int argc, char** argv);
700 
701 // apply_patch(srcfile, tgtfile, tgtsha1, tgtsize, sha1:patch, ...)
702 // apply_patch_check(file, sha1, ...)
703 // apply_patch_space(bytes)
ApplyPatchFn(const char * name,State * state,int argc,Expr * argv[])704 char* ApplyPatchFn(const char* name, State* state, int argc, Expr* argv[]) {
705     printf("in applypatchfn (%s)\n", name);
706 
707     char* prepend = NULL;
708     if (strstr(name, "check") != NULL) {
709         prepend = "-c";
710     } else if (strstr(name, "space") != NULL) {
711         prepend = "-s";
712     }
713 
714     char** args = ReadVarArgs(state, argc, argv);
715     if (args == NULL) return NULL;
716 
717     // insert the "program name" argv[0] and a copy of the "prepend"
718     // string (if any) at the start of the args.
719 
720     int extra = 1 + (prepend != NULL ? 1 : 0);
721     char** temp = malloc((argc+extra) * sizeof(char*));
722     memcpy(temp+extra, args, argc * sizeof(char*));
723     temp[0] = strdup("updater");
724     if (prepend) {
725         temp[1] = strdup(prepend);
726     }
727     free(args);
728     args = temp;
729     argc += extra;
730 
731     printf("calling applypatch\n");
732     fflush(stdout);
733     int result = applypatch(argc, args);
734     printf("applypatch returned %d\n", result);
735 
736     int i;
737     for (i = 0; i < argc; ++i) {
738         free(args[i]);
739     }
740     free(args);
741 
742     switch (result) {
743         case 0:   return strdup("t");
744         case 1:   return strdup("");
745         default:  return ErrorAbort(state, "applypatch couldn't parse args");
746     }
747 }
748 
UIPrintFn(const char * name,State * state,int argc,Expr * argv[])749 char* UIPrintFn(const char* name, State* state, int argc, Expr* argv[]) {
750     char** args = ReadVarArgs(state, argc, argv);
751     if (args == NULL) {
752         return NULL;
753     }
754 
755     int size = 0;
756     int i;
757     for (i = 0; i < argc; ++i) {
758         size += strlen(args[i]);
759     }
760     char* buffer = malloc(size+1);
761     size = 0;
762     for (i = 0; i < argc; ++i) {
763         strcpy(buffer+size, args[i]);
764         size += strlen(args[i]);
765         free(args[i]);
766     }
767     free(args);
768     buffer[size] = '\0';
769 
770     char* line = strtok(buffer, "\n");
771     while (line) {
772         fprintf(((UpdaterInfo*)(state->cookie))->cmd_pipe,
773                 "ui_print %s\n", line);
774         line = strtok(NULL, "\n");
775     }
776     fprintf(((UpdaterInfo*)(state->cookie))->cmd_pipe, "ui_print\n");
777 
778     return buffer;
779 }
780 
RunProgramFn(const char * name,State * state,int argc,Expr * argv[])781 char* RunProgramFn(const char* name, State* state, int argc, Expr* argv[]) {
782     if (argc < 1) {
783         return ErrorAbort(state, "%s() expects at least 1 arg", name);
784     }
785     char** args = ReadVarArgs(state, argc, argv);
786     if (args == NULL) {
787         return NULL;
788     }
789 
790     char** args2 = malloc(sizeof(char*) * (argc+1));
791     memcpy(args2, args, sizeof(char*) * argc);
792     args2[argc] = NULL;
793 
794     fprintf(stderr, "about to run program [%s] with %d args\n", args2[0], argc);
795 
796     pid_t child = fork();
797     if (child == 0) {
798         execv(args2[0], args2);
799         fprintf(stderr, "run_program: execv failed: %s\n", strerror(errno));
800         _exit(1);
801     }
802     int status;
803     waitpid(child, &status, 0);
804     if (WIFEXITED(status)) {
805         if (WEXITSTATUS(status) != 0) {
806             fprintf(stderr, "run_program: child exited with status %d\n",
807                     WEXITSTATUS(status));
808         }
809     } else if (WIFSIGNALED(status)) {
810         fprintf(stderr, "run_program: child terminated by signal %d\n",
811                 WTERMSIG(status));
812     }
813 
814     int i;
815     for (i = 0; i < argc; ++i) {
816         free(args[i]);
817     }
818     free(args);
819     free(args2);
820 
821     char buffer[20];
822     sprintf(buffer, "%d", status);
823 
824     return strdup(buffer);
825 }
826 
827 
RegisterInstallFunctions()828 void RegisterInstallFunctions() {
829     RegisterFunction("mount", MountFn);
830     RegisterFunction("is_mounted", IsMountedFn);
831     RegisterFunction("unmount", UnmountFn);
832     RegisterFunction("format", FormatFn);
833     RegisterFunction("show_progress", ShowProgressFn);
834     RegisterFunction("set_progress", SetProgressFn);
835     RegisterFunction("delete", DeleteFn);
836     RegisterFunction("delete_recursive", DeleteFn);
837     RegisterFunction("package_extract_dir", PackageExtractDirFn);
838     RegisterFunction("package_extract_file", PackageExtractFileFn);
839     RegisterFunction("symlink", SymlinkFn);
840     RegisterFunction("set_perm", SetPermFn);
841     RegisterFunction("set_perm_recursive", SetPermFn);
842 
843     RegisterFunction("getprop", GetPropFn);
844     RegisterFunction("file_getprop", FileGetPropFn);
845     RegisterFunction("write_raw_image", WriteRawImageFn);
846     RegisterFunction("write_firmware_image", WriteFirmwareImageFn);
847 
848     RegisterFunction("apply_patch", ApplyPatchFn);
849     RegisterFunction("apply_patch_check", ApplyPatchFn);
850     RegisterFunction("apply_patch_space", ApplyPatchFn);
851 
852     RegisterFunction("ui_print", UIPrintFn);
853 
854     RegisterFunction("run_program", RunProgramFn);
855 }
856