71cca1dcbf93697352c057e86bb74b8d54f4f4cb
[strongswan.git] / src / checksum / checksum_builder.c
1 /*
2 * Copyright (C) 2009 Martin Willi
3 * Hochschule fuer Technik Rapperswil, Switzerland
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13 * for more details.
14 */
15
16 #include <stdlib.h>
17 #include <stdio.h>
18 #include <dlfcn.h>
19
20 #include <library.h>
21
22 /* we need to fake some charon symbols to dlopen() its plugins */
23 void *charon, *eap_type_names, *auth_class_names, *protocol_id_names,
24 *action_names, *ipsec_mode_names, *ike_sa_state_names, *child_sa_state_names,
25 *policy_dir_names, *ipcomp_transform_names, *debug_names, *controller_cb_empty;
26
27 int main(int argc, char* argv[])
28 {
29 int i;
30 integrity_checker_t *integrity;
31
32 /* avoid confusing leak reports in build process */
33 setenv("LEAK_DETECTIVE_DISABLE", "1", 0);
34 library_init(NULL);
35 atexit(library_deinit);
36
37 integrity = integrity_checker_create(NULL);
38
39 printf("/**\n");
40 printf(" * checksums of files and loaded code segments.\n");
41 printf(" * created by %s\n", argv[0]);
42 printf(" */\n");
43 printf("\n");
44 printf("#include <library.h>\n");
45 printf("\n");
46 printf("integrity_checksum_t checksums[] = {\n");
47 fprintf(stderr, "integrity test data:\n");
48 fprintf(stderr, "module name, file size / checksum segment size / checksum\n");
49 for (i = 1; i < argc; i++)
50 {
51 char *name, *path, *sname = NULL;
52 void *handle, *symbol;
53 u_int32_t fsum, ssum;
54 size_t fsize = 0;
55 size_t ssize = 0;
56
57 path = argv[i];
58
59 if ((name = strstr(path, "libstrongswan-")))
60 {
61 name = strdup(name + strlen("libstrongswan-"));
62 name[strlen(name) - 3] = '"';
63 name[strlen(name) - 2] = ',';
64 name[strlen(name) - 1] = '\0';
65 sname = "plugin_create";
66 }
67 else if (strstr(path, "libstrongswan.so"))
68 {
69 name = strdup("libstrongswan\",");
70 sname = "library_init";
71 }
72 else if (strstr(path, "charon"))
73 {
74 name = strdup("charon\",");
75 }
76 else if (strstr(path, "pluto"))
77 {
78 name = strdup("pluto\",");
79 }
80 else
81 {
82 fprintf(stderr, "don't know how to handle '%s', ignored", path);
83 continue;
84 }
85
86 fsum = integrity->build_file(integrity, path, &fsize);
87 ssum = 0;
88 if (sname)
89 {
90 handle = dlopen(path, RTLD_LAZY);
91 if (handle)
92 {
93 symbol = dlsym(handle, sname);
94 if (symbol)
95 {
96 ssum = integrity->build_segment(integrity, symbol, &ssize);
97 }
98 else
99 {
100 fprintf(stderr, "symbol lookup failed: %s\n", dlerror());
101 }
102 dlclose(handle);
103 }
104 else
105 {
106 fprintf(stderr, "dlopen failed: %s\n", dlerror());
107 }
108 }
109 printf("\t{\"%-20s%7u, 0x%08x, %6u, 0x%08x},\n",
110 name, fsize, fsum, ssize, ssum);
111 fprintf(stderr, "\"%-20s%7u / 0x%08x %6u / 0x%08x\n",
112 name, fsize, fsum, ssize, ssum);
113 free(name);
114 }
115 printf("};\n");
116 printf("\n");
117 printf("int checksum_count = countof(checksums);\n");
118 printf("\n");
119 integrity->destroy(integrity);
120
121 exit(0);
122 }
123