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                                 return stream.GetContents ();
 310                         }
 311                 }
 312
 313                 Assembly assembly;
 314                 BinaryReader reader;
 315                 Hashtable method_hash;
 316                 Hashtable source_file_hash;
 317
 318                 Hashtable method_token_hash;
 319                 Hashtable method_name_hash;
 320                 Hashtable method_full_name_hash;
 321                 Hashtable source_name_hash;
 322
 323                 protected MonoSymbolFile (Assembly assembly, Stream stream)
 324                 {
 325                         this.assembly = assembly;
 326
 327                         reader = new BinaryReader (stream);
 328
 329                         try {
 330                                 long magic = reader.ReadInt64 ();
 331                                 long version = reader.ReadInt32 ();
 332                                 if ((magic != OffsetTable.Magic) || (version != OffsetTable.Version))
 333                                         throw new MonoSymbolFileException ();
 334                                 ot = new OffsetTable (reader);
 335                         } catch {
 336                                 throw new MonoSymbolFileException ();
 337                         }
 338
 339                         method_hash = new Hashtable ();
 340                         source_file_hash = new Hashtable ();
 341                 }
 342
 343                 public static MonoSymbolFile ReadSymbolFile (Assembly assembly)
 344                 {
 345                         Stream stream = assembly.GetManifestResourceStream ("MonoSymbolFile");
 346                         if (stream == null)
 347                                 return null;
 348
 349                         return new MonoSymbolFile (assembly, stream);
 350                 }
 351
 352                 public Assembly Assembly {
 353                         get { return assembly; }
 354                 }
 355
 356                 public int SourceCount {
 357                         get { return ot.SourceCount; }
 358                 }
 359
 360                 public int MethodCount {
 361                         get { return ot.MethodCount; }
 362                 }
 363
 364                 public int TypeCount {
 365                         get { return ot.TypeCount; }
 366                 }
 367
 368                 public int NamespaceCount {
 369                         get { return last_namespace_index; }
 370                 }
 371
 372                 internal int LineNumberCount = 0;
 373                 internal int LocalCount = 0;
 374                 internal int StringSize = 0;
 375
 376                 public SourceFileEntry GetSourceFile (int index)
 377                 {
 378                         if ((index < 1) || (index > ot.SourceCount))
 379                                 throw new ArgumentException ();
 380                         if (reader == null)
 381                                 throw new InvalidOperationException ();
 382
 383                         SourceFileEntry source = (SourceFileEntry) source_file_hash [index];
 384                         if (source != null)
 385                                 return source;
 386
 387                         reader.BaseStream.Position = ot.SourceTableOffset +
 388                                 SourceFileEntry.Size * (index - 1);
 389                         source = new SourceFileEntry (this, reader);
 390                         source_file_hash.Add (index, source);
 391                         return source;
 392                 }
 393
 394                 public SourceFileEntry[] Sources {
 395                         get {
 396                                 if (reader == null)
 397                                         throw new InvalidOperationException ();
 398
 399                                 SourceFileEntry[] retval = new SourceFileEntry [SourceCount];
 400                                 for (int i = 0; i < SourceCount; i++)
 401                                         retval [i] = GetSourceFile (i + 1);
 402                                 return retval;
 403                         }
 404                 }
 405
 406                 public MethodIndexEntry GetMethodIndexEntry (int index)
 407                 {
 408                         int old_pos = (int) reader.BaseStream.Position;
 409                         reader.BaseStream.Position = ot.MethodTableOffset +
 410                                 MethodIndexEntry.Size * (index - 1);
 411                         MethodIndexEntry ie = new MethodIndexEntry (reader);
 412                         reader.BaseStream.Position = old_pos;
 413                         return ie;
 414                 }
 415
 416                 public MethodEntry GetMethodByToken (int token)
 417                 {
 418                         if (reader == null)
 419                                 throw new InvalidOperationException ();
 420
 421                         if (method_token_hash == null) {
 422                                 method_token_hash = new Hashtable ();
 423
 424                                 for (int i = 0; i < MethodCount; i++) {
 425                                         MethodIndexEntry ie = GetMethodIndexEntry (i + 1);
 426
 427                                         method_token_hash.Add (ie.Token, i + 1);
 428                                 }
 429                         }
 430
 431                         object value = method_token_hash [token];
 432                         if (value == null)
 433                                 return null;
 434
 435                         return GetMethod ((int) value);
 436                 }
 437
 438                 public MethodEntry GetMethod (MethodBase method)
 439                 {
 440                         if (reader == null)
 441                                 throw new InvalidOperationException ();
 442                         int token = assembly.MonoDebugger_GetMethodToken (method);
 443                         return GetMethodByToken (token);
 444                 }
 445
 446                 public MethodEntry GetMethod (int index)
 447                 {
 448                         if ((index < 1) || (index > ot.MethodCount))
 449                                 throw new ArgumentException ();
 450                         if (reader == null)
 451                                 throw new InvalidOperationException ();
 452
 453                         MethodEntry entry = (MethodEntry) method_hash [index];
 454                         if (entry != null)
 455                                 return entry;
 456
 457                         MethodIndexEntry ie = GetMethodIndexEntry (index);
 458                         reader.BaseStream.Position = ie.FileOffset;
 459
 460                         entry = new MethodEntry (this, reader, index);
 461                         method_hash.Add (index, entry);
 462                         return entry;
 463                 }
 464
 465                 public MethodEntry[] Methods {
 466                         get {
 467                                 if (reader == null)
 468                                         throw new InvalidOperationException ();
 469
 470                                 MethodEntry[] retval = new MethodEntry [MethodCount];
 471                                 for (int i = 0; i < MethodCount; i++)
 472                                         retval [i] = GetMethod (i + 1);
 473                                 return retval;
 474                         }
 475                 }
 476
 477                 public MethodSourceEntry GetMethodSource (int index)
 478                 {
 479                         if ((index < 1) || (index > ot.MethodCount))
 480                                 throw new ArgumentException ();
 481                         if (reader == null)
 482                                 throw new InvalidOperationException ();
 483
 484                         object entry = method_source_hash [index];
 485                         if (entry != null)
 486                                 return (MethodSourceEntry) entry;
 487
 488                         MethodEntry method = GetMethod (index);
 489                         foreach (MethodSourceEntry source in method.SourceFile.Methods) {
 490                                 if (source.Index == index) {
 491                                         method_source_hash.Add (index, source);
 492                                         return source;
 493                                 }
 494                         }
 495
 496                         throw new MonoSymbolFileException ("Internal error.");
 497                 }
 498
 499                 public int FindMethod (string full_name)
 500                 {
 501                         if (reader == null)
 502                                 throw new InvalidOperationException ();
 503
 504                         if (method_full_name_hash == null) {
 505                                 method_full_name_hash = new Hashtable ();
 506
 507                                 for (int i = 0; i < ot.MethodCount; i++) {
 508                                         MethodIndexEntry ie = GetMethodIndexEntry (i + 1);
 509                                         string name = ReadString (ie.FullNameOffset);
 510
 511                                         method_full_name_hash.Add (name, i + 1);
 512                                 }
 513                         }
 514
 515                         object value = method_full_name_hash [full_name];
 516                         if (value == null)
 517                                 return -1;
 518                         return (int) value;
 519                 }
 520
 521                 public int[] MethodLookup (string query)
 522                 {
 523                         if (reader == null)
 524                                 throw new InvalidOperationException ();
 525
 526                         ArrayList list;
 527                         if (method_name_hash == null) {
 528                                 method_name_hash = new Hashtable ();
 529
 530                                 for (int i = 0; i < ot.MethodCount; i++) {
 531                                         MethodIndexEntry ie = GetMethodIndexEntry (i + 1);
 532                                         string full_name = ReadString (ie.FullNameOffset);
 533
 534                                         int pos = full_name.IndexOf ('(');
 535                                         string name = full_name.Substring (0, pos);
 536
 537                                         list = method_name_hash [name] as ArrayList;
 538                                         if (list == null) {
 539                                                 list = new ArrayList ();
 540                                                 method_name_hash.Add (name, list);
 541                                         }
 542
 543                                         list.Add (i + 1);
 544                                 }
 545                         }
 546
 547                         list = method_name_hash [query] as ArrayList;
 548                         if (list == null)
 549                                 return new int [0];
 550
 551                         int[] retval = new int [list.Count];
 552                         list.CopyTo (retval, 0);
 553                         return retval;
 554                 }
 555
 556                 public int FindSource (string file_name)
 557                 {
 558                         if (reader == null)
 559                                 throw new InvalidOperationException ();
 560
 561                         if (source_name_hash == null) {
 562                                 source_name_hash = new Hashtable ();
 563
 564                                 for (int i = 0; i < ot.SourceCount; i++) {
 565                                         SourceFileEntry source = GetSourceFile (i + 1);
 566
 567                                         source_name_hash.Add (source.FileName, i);
 568                                 }
 569                         }
 570
 571                         object value = source_name_hash [file_name];
 572                         if (value == null)
 573                                 return -1;
 574                         return (int) value;
 575                 }
 576
 577                 internal BinaryReader BinaryReader {
 578                         get {
 579                                 if (reader == null)
 580                                         throw new InvalidOperationException ();
 581
 582                                 return reader;
 583                         }
 584                 }
 585
 586                 void IDisposable.Dispose ()
 587                 {
 588                         Dispose (true);
 589                 }
 590
 591                 protected virtual void Dispose (bool disposing)
 592                 {
 593                         if (disposing) {
 594                                 if (reader != null) {
 595                                         reader.Close ();
 596                                         reader = null;
 597                                 }
 598                         }
 599                 }
 600         }
 601 }