mcs/class/Mono.Cecil/Mono.Cecil.Metadata/StringHeap.cs

   1 //
   2 // StringHeap.cs
   3 //
   4 // Author:
   5 //   Jb Evain (jbevain@gmail.com)
   6 //
   7 // Copyright (c) 2008 - 2010 Jb Evain
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System;
  30 using System.Collections.Generic;
  31 using System.Text;
  32
  33 using Mono.Cecil.PE;
  34
  35 namespace Mono.Cecil.Metadata {
  36
  37         class StringHeap : Heap {
  38
  39                 readonly Dictionary<uint, string> strings = new Dictionary<uint, string> ();
  40
  41                 public StringHeap (Section section, uint start, uint size)
  42                         : base (section, start, size)
  43                 {
  44                 }
  45
  46                 public string Read (uint index)
  47                 {
  48                         if (index == 0)
  49                                 return string.Empty;
  50
  51                         string @string;
  52                         if (strings.TryGetValue (index, out @string))
  53                                 return @string;
  54
  55                         if (index > Size - 1)
  56                                 return string.Empty;
  57
  58                         @string = ReadStringAt (index);
  59                         if (@string.Length != 0)
  60                                 strings.Add (index, @string);
  61
  62                         return @string;
  63                 }
  64
  65                 protected virtual string ReadStringAt (uint index)
  66                 {
  67                         int length = 0;
  68                         byte [] data = Section.Data;
  69                         int start = (int) (index + Offset);
  70
  71                         for (int i = start; ; i++) {
  72                                 if (data [i] == 0)
  73                                         break;
  74
  75                                 length++;
  76                         }
  77
  78                         return Encoding.UTF8.GetString (data, start, length);
  79                 }
  80         }
  81 }