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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
#include <stdio.h>
#include <stdint.h>
#include <stdbool.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <zlib.h>
#include <mbedtls/error.h>
#include "pack.h"
#include "util_rt.h"
static void report_unpack_bin_archive_err (const prne_unpack_bin_archive_result_t *r) {
const char *err_str, *err_msg = NULL;
char err_buf[1024];
switch (r->result) {
case PRNE_UNPACK_BIN_ARCHIVE_OK:
err_str = "ok";
break;
case PRNE_UNPACK_BIN_ARCHIVE_CRYPTO_ERR:
err_str = "crypto error";
mbedtls_strerror(r->err, err_buf, 1024);
err_msg = err_buf;
break;
case PRNE_UNPACK_BIN_ARCHIVE_Z_ERR:
err_str = "zlib error";
err_msg = zError((int)r->err);
break;
case PRNE_UNPACK_BIN_ARCHIVE_ERRNO:
err_str = "errno";
err_msg = strerror((int)r->err);
break;
case PRNE_UNPACK_BIN_ARCHIVE_MEM_ERR:
err_str = "memory error";
err_msg = strerror((int)r->err);
break;
case PRNE_UNPACK_BIN_ARCHIVE_FMT_ERR:
err_str = "format error";
break;
default:
err_str = "* unknown";
}
if (err_msg == NULL) {
fprintf(stderr, "%s.\n", err_str);
}
else {
fprintf(stderr, "%s: %s\n", err_str, err_msg);
}
}
static void report_index_bin_archive_err (const prne_index_bin_archive_result_code_t c) {
const char *msg;
switch (c) {
case PRNE_INDEX_BIN_ARCHIVE_OK:
msg = "ok"; break;
case PRNE_INDEX_BIN_ARCHIVE_FMT_ERR:
msg = "format error"; break;
case PRNE_INDEX_BIN_ARCHIVE_MEM_ERR:
msg = "memory error"; break;
default:
msg = "* unknown"; break;
}
fprintf(stderr, "%s.\n", msg);
}
int main (const int argc, const char **args) {
int exit_code = 0;
const char *path_prefix;
size_t path_prefix_len;
prne_unpack_bin_archive_result_t unpack_ret;
prne_bin_archive_t bin_archive;
prne_index_bin_archive_result_code_t index_ret;
size_t i;
const char *arch_str;
char *path = NULL;
size_t path_size;
void *ny_buf;
int fd = -1;
if (argc <= 1) {
fprintf(stderr, "Usage: %s <prefix>\n", args[0]);
return 1;
}
path_prefix = args[1];
path_prefix_len = strlen(path_prefix);
prne_init_bin_archive(&bin_archive);
do { // fake loop
unpack_ret = prne_unpack_bin_archive(STDIN_FILENO);
if (unpack_ret.result != PRNE_UNPACK_BIN_ARCHIVE_OK) {
report_unpack_bin_archive_err(&unpack_ret);
exit_code = 2;
break;
}
index_ret = prne_index_bin_archive(&unpack_ret, &bin_archive);
if (index_ret != PRNE_INDEX_BIN_ARCHIVE_OK) {
report_index_bin_archive_err(index_ret);
exit_code = 2;
break;
}
for (i = 0; i < bin_archive.nb_binaries; i += 1) {
arch_str = prne_arch2str(bin_archive.arch_arr[i]);
if (arch_str == NULL) {
fprintf(stderr, "** unrecognised arch!");
exit_code = 2;
break;
}
path_size = 2 + path_prefix_len + strlen(arch_str);
ny_buf = prne_realloc(path, 1, path_size);
if (ny_buf == NULL) {
perror("prne_realloc()");
exit_code = 2;
break;
}
path = (char*)ny_buf;
sprintf(path, "%s.%s", path_prefix, arch_str);
fd = open(path, O_CREAT | O_WRONLY | O_TRUNC, 0666);
if (fd < 0) {
perror("open()");
exit_code = 2;
break;
}
if (write(fd, bin_archive.data + bin_archive.offset_arr[i], bin_archive.size_arr[i]) != (ssize_t)bin_archive.size_arr[i]) {
perror("write()");
exit_code = 2;
break;
}
close(fd);
}
} while (false);
prne_free(path);
close(fd);
prne_free_unpack_bin_archive_result(&unpack_ret);
prne_free_bin_archive(&bin_archive);
return exit_code;
}
|
