2003-02-10 Nick Drochak <ndrochak@gol.com>
[mono.git] / mcs / class / Mono.CSharp.Debugger / MonoSymbolFile.cs
1 //
2 // System.Diagnostics.SymbolStore/MonoSymbolFile.cs
3 //
4 // Author:
5 //   Martin Baulig (martin@gnome.org)
6 //
7 // (C) 2003 Ximian, Inc.  http://www.ximian.com
8 //
9
10 using System;
11 using System.Reflection;
12 using System.Reflection.Emit;
13 using System.Collections;
14 using System.Text;
15 using System.IO;
16         
17 namespace Mono.CSharp.Debugger
18 {
19         public class MonoSymbolFileException : Exception
20         {
21                 public MonoSymbolFileException ()
22                         : base ()
23                 { }
24
25                 public MonoSymbolFileException (string message, params object[] args)
26                         : base (String.Format (message, args))
27                 { }
28         }
29
30         internal class MyMemoryStream : Stream
31         {
32                 int length;
33                 int real_length;
34                 int position;
35
36                 int chunk_size = 4096;
37                 ArrayList chunks = new ArrayList ();
38
39                 private struct Chunk {
40                         public readonly int Offset;
41                         public readonly int Length;
42                         public byte[] Buffer;
43
44                         public Chunk (int offset, int length)
45                         {
46                                 this.Offset = offset;
47                                 this.Length = length;
48                                 this.Buffer = new Byte [length];
49                         }
50                 }
51
52                 public override long Position {
53                         get { return position; }
54
55                         set {
56                                 if (value > length)
57                                         throw new ArgumentOutOfRangeException ();
58
59                                 position = (int) value;
60                         }
61                 }
62
63                 public override long Length {
64                         get { return length; }
65                 }
66
67                 public override bool CanRead {
68                         get { return true; }
69                 }
70
71                 public override bool CanWrite {
72                         get { return true; }
73                 }
74
75                 public override bool CanSeek {
76                         get { return true; }
77                 }
78
79                 public override void SetLength (long new_length)
80                 {
81                         if (new_length < length)
82                                 throw new ArgumentException ();
83
84                         while (new_length >= real_length) {
85                                 Chunk new_chunk = new Chunk (real_length, chunk_size);
86                                 chunks.Add (new_chunk);
87                                 real_length += chunk_size;
88                         }
89
90                         length = (int) new_length;
91                 }
92
93                 public override void Flush ()
94                 { }
95
96                 public override long Seek (long offset, SeekOrigin origin)
97                 {
98                         int ref_point;
99
100                         switch (origin) {
101                         case SeekOrigin.Begin:
102                                 ref_point = 0;
103                                 break;
104                         case SeekOrigin.Current:
105                                 ref_point = position;
106                                 break;
107                         case SeekOrigin.End:
108                                 ref_point = length;
109                                 break;
110                         default:
111                                 throw new ArgumentException ("Invalid SeekOrigin");
112                         }
113
114                         if ((ref_point + offset < 0) || (offset > real_length))
115                                 throw new ArgumentOutOfRangeException ();
116
117                         position = ref_point + (int) offset;
118
119                         return position;
120                 }
121
122                 Chunk FindChunk (int offset)
123                 {
124                         return (Chunk) chunks [offset / chunk_size];
125                 }
126
127                 public override int Read (byte[] buffer, int offset, int count)
128                 {
129                         int old_count = count;
130
131                         while (count > 0) {
132                                 Chunk chunk = FindChunk (position);
133                                 int coffset = position - chunk.Offset;
134                                 int rest = chunk.Length - coffset;
135                                 int size = System.Math.Min (count, rest);
136
137                                 Array.Copy (chunk.Buffer, coffset, buffer, offset, size);
138                                 position += size;
139                                 offset += size;
140                                 count -= size;
141                         }
142
143                         return old_count;
144                 }
145
146                 public override void Write (byte[] buffer, int offset, int count)
147                 {
148                         if (position + count > length)
149                                 SetLength (position + count);
150
151                         while (count > 0) {
152                                 Chunk chunk = FindChunk (position);
153                                 int coffset = position - chunk.Offset;
154                                 int rest = chunk.Length - coffset;
155                                 int size = System.Math.Min (count, rest);
156
157                                 Array.Copy (buffer, offset, chunk.Buffer, coffset, size);
158                                 position += size;
159                                 offset += size;
160                                 count -= size;
161                         }
162                 }
163
164                 public byte[] GetContents ()
165                 {
166                         byte[] retval = new byte [length];
167                         position = 0;
168                         Read (retval, 0, length);
169                         return retval;
170                 }
171         }
172
173         public class MonoSymbolFile : IDisposable
174         {
175                 ArrayList methods = new ArrayList ();
176                 ArrayList sources = new ArrayList ();
177                 Hashtable method_source_hash = new Hashtable ();
178                 Hashtable type_hash = new Hashtable ();
179
180                 OffsetTable ot;
181                 int last_type_index;
182                 int last_method_index;
183                 int last_source_index;
184                 int last_namespace_index;
185
186                 public MonoSymbolFile ()
187                 { }
188
189                 internal int AddSource (SourceFileEntry source)
190                 {
191                         sources.Add (source);
192                         return ++last_source_index;
193                 }
194
195                 internal int DefineType (Type type)
196                 {
197                         if (type_hash.Contains (type))
198                                 return (int) type_hash [type];
199
200                         int index = ++last_type_index;
201                         type_hash.Add (type, index);
202                         return index;
203                 }
204
205                 internal void AddMethod (MethodEntry entry)
206                 {
207                         methods.Add (entry);
208                 }
209
210                 internal int GetNextTypeIndex ()
211                 {
212                         return ++last_type_index;
213                 }
214
215                 internal int GetNextMethodIndex ()
216                 {
217                         return ++last_method_index;
218                 }
219
220                 internal int GetNextNamespaceIndex ()
221                 {
222                         return ++last_namespace_index;
223                 }
224
225                 internal void WriteString (BinaryWriter bw, string text)
226                 {
227                         byte[] data = Encoding.UTF8.GetBytes (text);
228                         bw.Write ((int) data.Length);
229                         bw.Write (data);
230                         StringSize += data.Length;
231                 }
232
233                 internal string ReadString (int offset)
234                 {
235                         int old_pos = (int) reader.BaseStream.Position;
236                         reader.BaseStream.Position = offset;
237                         int length = reader.ReadInt32 ();
238
239                         byte[] data = reader.ReadBytes (length);
240                         string text = Encoding.UTF8.GetString (data);
241                         reader.BaseStream.Position = old_pos;
242                         return text;
243                 }
244
245                 void Write (BinaryWriter bw)
246                 {
247                         // Magic number and file version.
248                         bw.Write (OffsetTable.Magic);
249                         bw.Write (OffsetTable.Version);
250
251                         //
252                         // Offsets of file sections; we must write this after we're done
253                         // writing the whole file, so we just reserve the space for it here.
254                         //
255                         long offset_table_offset = bw.BaseStream.Position;
256                         ot.Write (bw);
257
258                         //
259                         // Write data sections.
260                         //
261                         ot.DataSectionOffset = (int) bw.BaseStream.Position;
262                         foreach (SourceFileEntry source in sources)
263                                 source.WriteData (bw);
264                         ot.DataSectionSize = (int) bw.BaseStream.Position - ot.DataSectionOffset;
265
266                         //
267                         // Write method table.
268                         //
269                         ot.MethodTableOffset = (int) bw.BaseStream.Position;
270                         for (int i = 0; i < methods.Count; i++) {
271                                 MethodEntry entry = (MethodEntry) methods [i];
272                                 entry.WriteIndex (bw);
273                         }
274                         ot.MethodTableSize = (int) bw.BaseStream.Position - ot.MethodTableOffset;
275
276                         //
277                         // Write source table.
278                         //
279                         ot.SourceTableOffset = (int) bw.BaseStream.Position;
280                         for (int i = 0; i < sources.Count; i++) {
281                                 SourceFileEntry source = (SourceFileEntry) sources [i];
282                                 source.Write (bw);
283                         }
284                         ot.SourceTableSize = (int) bw.BaseStream.Position - ot.SourceTableOffset;
285
286                         //
287                         // Fixup offset table.
288                         //
289                         ot.TypeCount = last_type_index;
290                         ot.MethodCount = methods.Count;
291                         ot.SourceCount = sources.Count;
292
293                         //
294                         // Write offset table.
295                         //
296                         ot.TotalFileSize = (int) bw.BaseStream.Position;
297                         bw.Seek ((int) offset_table_offset, SeekOrigin.Begin);
298                         ot.Write (bw);
299                         bw.Seek (0, SeekOrigin.End);
300                 }
301
302                 public byte[] CreateSymbolFile ()
303                 {
304                         if (reader != null)
305                                 throw new InvalidOperationException ();
306
307                         using (MyMemoryStream stream = new MyMemoryStream ()) {
308                                 Write (new BinaryWriter (stream));
309                                 Console.WriteLine ("WROTE SYMFILE: {0} sources, {1} methods, {2} types, " +
310                                                    "{3} line numbers, {4} locals, {5} namespaces, " +
311                                                    "{6} bytes of string data",
312                                                    SourceCount, MethodCount, TypeCount, LineNumberCount,
313                                                    LocalCount, NamespaceCount, StringSize);
314                                 Console.WriteLine (ot);
315                                 return stream.GetContents ();
316                         }
317                 }
318
319                 // Stream stream;
320                 BinaryReader reader;
321                 Hashtable method_hash;
322                 Hashtable source_file_hash;
323
324                 Hashtable method_name_hash;
325                 Hashtable source_name_hash;
326
327                 protected MonoSymbolFile (Stream stream)
328                 {
329                         reader = new BinaryReader (stream);
330
331                         try {
332                                 long magic = reader.ReadInt64 ();
333                                 long version = reader.ReadInt32 ();
334                                 if ((magic != OffsetTable.Magic) || (version != OffsetTable.Version))
335                                         throw new MonoSymbolFileException ();
336                                 ot = new OffsetTable (reader);
337                         } catch {
338                                 throw new MonoSymbolFileException ();
339                         }
340
341                         method_hash = new Hashtable ();
342                         source_file_hash = new Hashtable ();
343                 }
344
345                 public static MonoSymbolFile ReadSymbolFile (Assembly assembly)
346                 {
347                         Stream stream = assembly.GetManifestResourceStream ("MonoSymbolFile");
348                         if (stream == null)
349                                 return null;
350
351                         return new MonoSymbolFile (stream);
352                 }
353
354                 public int SourceCount {
355                         get { return ot.SourceCount; }
356                 }
357
358                 public int MethodCount {
359                         get { return ot.MethodCount; }
360                 }
361
362                 public int TypeCount {
363                         get { return ot.TypeCount; }
364                 }
365
366                 public int NamespaceCount {
367                         get { return last_namespace_index; }
368                 }
369
370                 internal int LineNumberCount = 0;
371                 internal int LocalCount = 0;
372                 internal int StringSize = 0;
373
374                 public SourceFileEntry GetSourceFile (int index)
375                 {
376                         if ((index < 1) || (index > ot.SourceCount))
377                                 throw new ArgumentException ();
378                         if (reader == null)
379                                 throw new InvalidOperationException ();
380
381                         SourceFileEntry source = (SourceFileEntry) source_file_hash [index];
382                         if (source != null)
383                                 return source;
384
385                         reader.BaseStream.Position = ot.SourceTableOffset +
386                                 SourceFileEntry.Size * (index - 1);
387                         source = new SourceFileEntry (this, reader);
388                         source_file_hash.Add (index, source);
389                         return source;
390                 }
391
392                 public SourceFileEntry[] Sources {
393                         get {
394                                 if (reader == null)
395                                         throw new InvalidOperationException ();
396
397                                 SourceFileEntry[] retval = new SourceFileEntry [SourceCount];
398                                 for (int i = 0; i < SourceCount; i++)
399                                         retval [i] = GetSourceFile (i + 1);
400                                 return retval;
401                         }
402                 }
403
404                 public MethodEntry GetMethod (int index)
405                 {
406                         if ((index < 1) || (index > ot.MethodCount))
407                                 throw new ArgumentException ();
408                         if (reader == null)
409                                 throw new InvalidOperationException ();
410
411                         MethodEntry entry = (MethodEntry) method_hash [index];
412                         if (entry != null)
413                                 return entry;
414
415                         reader.BaseStream.Position = ot.MethodTableOffset + 8 * (index - 1);
416                         reader.BaseStream.Position = reader.ReadInt32 ();
417
418                         entry = new MethodEntry (this, reader, index);
419                         method_hash.Add (index, entry);
420                         return entry;
421                 }
422
423                 public MethodEntry[] Methods {
424                         get {
425                                 if (reader == null)
426                                         throw new InvalidOperationException ();
427
428                                 MethodEntry[] retval = new MethodEntry [MethodCount];
429                                 for (int i = 0; i < MethodCount; i++)
430                                         retval [i] = GetMethod (i + 1);
431                                 return retval;
432                         }
433                 }
434
435                 public MethodSourceEntry GetMethodSource (int index)
436                 {
437                         if ((index < 1) || (index > ot.MethodCount))
438                                 throw new ArgumentException ();
439                         if (reader == null)
440                                 throw new InvalidOperationException ();
441
442                         object entry = method_source_hash [index];
443                         if (entry != null)
444                                 return (MethodSourceEntry) entry;
445
446                         MethodEntry method = GetMethod (index);
447                         foreach (MethodSourceEntry source in method.SourceFile.Methods) {
448                                 if (source.Index == index) {
449                                         method_source_hash.Add (index, source);
450                                         return source;
451                                 }
452                         }
453
454                         throw new MonoSymbolFileException ("Internal error.");
455                 }
456
457                 public int FindMethod (string full_name)
458                 {
459                         if (reader == null)
460                                 throw new InvalidOperationException ();
461
462                         if (method_name_hash == null) {
463                                 method_name_hash = new Hashtable ();
464
465                                 for (int i = 0; i < ot.MethodCount; i++) {
466                                         reader.BaseStream.Position = ot.MethodTableOffset + 8 * i;
467
468                                         int offset = reader.ReadInt32 ();
469                                         int name_offset = reader.ReadInt32 ();
470                                         string name = ReadString (name_offset);
471
472                                         method_name_hash.Add (name, i + 1);
473                                 }
474                         }
475
476                         object value = method_name_hash [full_name];
477                         if (value == null)
478                                 return -1;
479                         return (int) value;
480                 }
481
482                 public int FindSource (string file_name)
483                 {
484                         if (reader == null)
485                                 throw new InvalidOperationException ();
486
487                         if (source_name_hash == null) {
488                                 source_name_hash = new Hashtable ();
489
490                                 for (int i = 0; i < ot.SourceCount; i++) {
491                                         SourceFileEntry source = GetSourceFile (i + 1);
492
493                                         source_name_hash.Add (source.FileName, i);
494                                 }
495                         }
496
497                         object value = source_name_hash [file_name];
498                         if (value == null)
499                                 return -1;
500                         return (int) value;
501                 }
502
503                 internal BinaryReader BinaryReader {
504                         get {
505                                 if (reader == null)
506                                         throw new InvalidOperationException ();
507
508                                 return reader;
509                         }
510                 }
511
512                 void IDisposable.Dispose ()
513                 {
514                         Dispose (true);
515                 }
516
517                 protected virtual void Dispose (bool disposing)
518                 {
519                         if (disposing) {
520                                 if (reader != null) {
521                                         reader.Close ();
522                                         reader = null;
523                                 }
524                         }
525                 }
526         }
527 }