19990502 sourceware import
[external/binutils.git] / gprof / source.c
1 /*
2  * Keeps track of source files.
3  */
4 #include "gprof.h"
5 #include "libiberty.h"
6 #include "search_list.h"
7 #include "source.h"
8
9 #define EXT_ANNO "-ann"         /* postfix of annotated files */
10
11 /*
12  * Default option values:
13  */
14 bool create_annotation_files = FALSE;
15
16 Search_List src_search_list =
17 {0, 0};
18 Source_File *first_src_file = 0;
19
20
21 Source_File *
22 DEFUN (source_file_lookup_path, (path), const char *path)
23 {
24   Source_File *sf;
25
26   for (sf = first_src_file; sf; sf = sf->next)
27     {
28       if (strcmp (path, sf->name) == 0)
29         {
30           break;
31         }
32     }
33   if (!sf)
34     {
35       /* create a new source file descriptor: */
36
37       sf = (Source_File *) xmalloc (sizeof (*sf));
38       memset (sf, 0, sizeof (*sf));
39       sf->name = xstrdup (path);
40       sf->next = first_src_file;
41       first_src_file = sf;
42     }
43   return sf;
44 }
45
46
47 Source_File *
48 DEFUN (source_file_lookup_name, (filename), const char *filename)
49 {
50   const char *fname;
51   Source_File *sf;
52   /*
53    * The user cannot know exactly how a filename will be stored in
54    * the debugging info (e.g., ../include/foo.h
55    * vs. /usr/include/foo.h).  So we simply compare the filename
56    * component of a path only:
57    */
58   for (sf = first_src_file; sf; sf = sf->next)
59     {
60       fname = strrchr (sf->name, '/');
61       if (fname)
62         {
63           ++fname;
64         }
65       else
66         {
67           fname = sf->name;
68         }
69       if (strcmp (filename, fname) == 0)
70         {
71           break;
72         }
73     }
74   return sf;
75 }
76
77
78 FILE *
79 DEFUN (annotate_source, (sf, max_width, annote, arg),
80        Source_File * sf AND int max_width
81        AND void (*annote) PARAMS ((char *buf, int w, int l, void *arg))
82        AND void *arg)
83 {
84   static bool first_file = TRUE;
85   int i, line_num, nread;
86   bool new_line;
87   char buf[8192];
88   char fname[PATH_MAX];
89   char *annotation, *name_only;
90   FILE *ifp, *ofp;
91   Search_List_Elem *sle = src_search_list.head;
92
93   /*
94    * Open input file.  If open fails, walk along search-list until
95    * open succeeds or reaching end of list:
96    */
97   strcpy (fname, sf->name);
98   if (sf->name[0] == '/')
99     {
100       sle = 0;                  /* don't use search list for absolute paths */
101     }
102   name_only = 0;
103   while (TRUE)
104     {
105       DBG (SRCDEBUG, printf ("[annotate_source]: looking for %s, trying %s\n",
106                              sf->name, fname));
107       ifp = fopen (fname, FOPEN_RB);
108       if (ifp)
109         {
110           break;
111         }
112       if (!sle && !name_only)
113         {
114           name_only = strrchr (sf->name, '/');
115           if (name_only)
116             {
117               /* try search-list again, but this time with name only: */
118               ++name_only;
119               sle = src_search_list.head;
120             }
121         }
122       if (sle)
123         {
124           strcpy (fname, sle->path);
125           strcat (fname, "/");
126           if (name_only)
127             {
128               strcat (fname, name_only);
129             }
130           else
131             {
132               strcat (fname, sf->name);
133             }
134           sle = sle->next;
135         }
136       else
137         {
138           if (errno == ENOENT)
139             {
140               fprintf (stderr, _("%s: could not locate `%s'\n"),
141                        whoami, sf->name);
142             }
143           else
144             {
145               perror (sf->name);
146             }
147           return 0;
148         }
149     }
150
151   ofp = stdout;
152   if (create_annotation_files)
153     {
154       /* try to create annotated source file: */
155       const char *filename;
156
157       /* create annotation files in the current working directory: */
158       filename = strrchr (sf->name, '/');
159       if (filename)
160         {
161           ++filename;
162         }
163       else
164         {
165           filename = sf->name;
166         }
167
168       strcpy (fname, filename);
169       strcat (fname, EXT_ANNO);
170       ofp = fopen (fname, "w");
171       if (!ofp)
172         {
173           perror (fname);
174           return 0;
175         }
176     }
177
178   /*
179    * Print file names if output goes to stdout and there are
180    * more than one source file:
181    */
182   if (ofp == stdout)
183     {
184       if (first_file)
185         {
186           first_file = FALSE;
187         }
188       else
189         {
190           fputc ('\n', ofp);
191         }
192       if (first_output)
193         {
194           first_output = FALSE;
195         }
196       else
197         {
198           fprintf (ofp, "\f\n");
199         }
200       fprintf (ofp, _("*** File %s:\n"), sf->name);
201     }
202
203   annotation = xmalloc (max_width + 1);
204   line_num = 1;
205   new_line = TRUE;
206   while ((nread = fread (buf, 1, sizeof (buf), ifp)) > 0)
207     {
208       for (i = 0; i < nread; ++i)
209         {
210           if (new_line)
211             {
212               (*annote) (annotation, max_width, line_num, arg);
213               fputs (annotation, ofp);
214               ++line_num;
215               new_line = FALSE;
216             }
217           new_line = (buf[i] == '\n');
218           fputc (buf[i], ofp);
219         }
220     }
221   free (annotation);
222   return ofp;
223 }