Update copyright to LGPL on all files
[dyninst.git] / symtabAPI / tests / addsyms.C
1 /*
2  * Copyright (c) 1996-2009 Barton P. Miller
3  * 
4  * We provide the Paradyn Parallel Performance Tools (below
5  * described as "Paradyn") on an AS IS basis, and do not warrant its
6  * validity or performance.  We reserve the right to update, modify,
7  * or discontinue this software at any time.  We shall have no
8  * obligation to supply such updates or modifications or any other
9  * form of support to you.
10  * 
11  * By your use of Paradyn, you understand and agree that we (or any
12  * other person or entity with proprietary rights in Paradyn) are
13  * under no obligation to provide either maintenance services,
14  * update services, notices of latent defects, or correction of
15  * defects for Paradyn.
16  * 
17  * This library is free software; you can redistribute it and/or
18  * modify it under the terms of the GNU Lesser General Public
19  * License as published by the Free Software Foundation; either
20  * version 2.1 of the License, or (at your option) any later version.
21  * 
22  * This library is distributed in the hope that it will be useful,
23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
25  * Lesser General Public License for more details.
26  * 
27  * You should have received a copy of the GNU Lesser General Public
28  * License along with this library; if not, write to the Free Software
29  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
30  */
31 /* Given a binary with missing symbols as input, produce a new
32    binary with new symbols filled in for functions discovered
33    by Dyninst's techniques.
34 */
35
36 #include <stdio.h>
37 #include <stdlib.h>
38 #include <string.h>
39
40 #include <BPatch.h>
41 #include <Dyn_Symtab.h>
42 #include <BPatch_function.h>
43
44 using namespace std;
45
46 BPatch bpatch;
47
48 void usage(char *name)
49 {
50     printf("Usage: %s <binary> [output binary]\n",name);
51 }
52
53 int main(int argc, char**argv)
54 {
55     char * inName;
56     char * outName;
57     string inStr;
58     string outStr;
59
60     BPatch_process * proc;
61     BPatch_image * img;
62     const char * bpatch_argv[1];
63     BPatch_Vector<BPatch_function *> * dyn_funcs;
64     BPatch_Vector<BPatch_module *> * dyn_mods;
65
66     Dyn_Symtab *symtab;
67
68     char fname[128];
69
70     vector<Dyn_Symbol *> allSyms; 
71     vector<Dyn_Module *> sym_mods;
72
73     if(argc < 2) {
74         usage(argv[0]);
75         exit(1);
76     }
77
78     inName = argv[1];
79     inStr = inName;
80     if(argc == 3) {
81         outName = argv[2];
82     } else {
83         outName = new char[strlen(argv[1]) + 5];
84         snprintf(outName, strlen(argv[1]) + 5, "%s-out",argv[1]);
85     }
86     outStr = outName;
87
88     /* 1. Load the binary and parse with Dyninst */
89     printf("*** Loading image...\n");
90     
91     bpatch_argv[0] = NULL;
92     proc = bpatch.processCreate(inName,bpatch_argv);
93     if(!proc) {
94         fprintf(stderr,"Unable to create process %s, exiting\n",inName);
95         exit(1);
96     }
97     img = proc->getImage();
98
99     printf("*** Parsing binary image...\n");
100
101         // cause Dyninst to do a full parse of the binary
102         dyn_funcs = img->getProcedures();
103         if(!dyn_funcs)  {
104             fprintf(stderr,"Unable to parse binary %s, exiting\n",inName);
105             exit(1);
106         }
107
108     BPatch_module *mod = img->findModule("DEFAULT_MODULE");
109     if(!mod) {
110         fprintf(stderr,"unable to find default module, exiting\n");
111         exit(1);
112     }
113
114     Dyn_Symtab * st = mod->getDynSymtab();
115     if(st) {
116       
117         printf("*** Emitting new binary\n"); 
118        /* st->getAllModules(sym_mods);
119
120         for(unsigned m=0;m<sym_mods.size(); m++)
121         {
122             // for each module, find the analogue in Dyninst
123
124             printf("module: %s (%s)\n",sym_mods[m]->fileName().c_str(),
125                 sym_mods[m]->fullName().c_str());
126             
127         } */
128         st->emitSymbols(outStr);
129     } else {   
130         printf("*** Failed to locate Dyn_Symtab object\n"); 
131     }
132
133     return 0;
134 }