public static void GetEmbeddedModule(ModuleReader reader) { // Read the MZ signature. if (reader.ReadByte() != 'M' || reader.ReadByte() != 'Z') { throw new ModuleException("Invalid PE signature."); } // Read the PE offset. reader.SetPosition(0x3c); uint peOffset = reader.ReadUInt(); // Read the PE\0\0 sinature reader.SetPosition(peOffset); if (reader.ReadByte() != 'P' || reader.ReadByte() != 'E' || reader.ReadByte() != 0 || reader.ReadByte() != 0) { throw new ModuleException("Unsupported MS DOS programs."); } // Read the COFF header. CoffHeader header = new CoffHeader(); header.Read(reader); // Ignore the optional header. reader.Skip(header.optionalHeaderSize); // Read the sections until finding the .cbm section. CoffSectionHeader sectionHeader = new CoffSectionHeader(); for (int i = 0; i < header.numSections; ++i) { // Read the section header. sectionHeader.Read(reader); // If this is the .cbm section, done. if (sectionHeader.name == ".cbm") { reader.SetPosition(sectionHeader.rawDataPointer); return; } } // Couldn't find embedded module. throw new ModuleException("Couldn't find embedded Chela module."); }
public static void GetEmbeddedModule(ModuleReader reader) { // Read the MZ signature. if(reader.ReadByte() != 'M' || reader.ReadByte() != 'Z') throw new ModuleException("Invalid PE signature."); // Read the PE offset. reader.SetPosition(0x3c); uint peOffset = reader.ReadUInt(); // Read the PE\0\0 sinature reader.SetPosition(peOffset); if(reader.ReadByte() != 'P' || reader.ReadByte() != 'E' || reader.ReadByte() != 0 || reader.ReadByte() != 0) throw new ModuleException("Unsupported MS DOS programs."); // Read the COFF header. CoffHeader header = new CoffHeader(); header.Read(reader); // Ignore the optional header. reader.Skip(header.optionalHeaderSize); // Read the sections until finding the .cbm section. CoffSectionHeader sectionHeader = new CoffSectionHeader(); for(int i = 0; i < header.numSections; ++i) { // Read the section header. sectionHeader.Read(reader); // If this is the .cbm section, done. if(sectionHeader.name == ".cbm") { reader.SetPosition(sectionHeader.rawDataPointer); return; } } // Couldn't find embedded module. throw new ModuleException("Couldn't find embedded Chela module."); }
private static void GetEmbeddedModule64(ModuleReader reader, byte[] ident) { // Read the endiannes. bool big = ident[(int)ElfIdent.EI_DATA] == (int)ElfData.ELFDATA2MSB; // Read the header. Elf64_Ehdr header = new Elf64_Ehdr(); header.Read(reader, big); // Reject files without section names. if (header.e_shstrndex == (int)ElfSectionNumber.SHN_UNDEF) { throw new ModuleException("Unsupported elfs without section names"); } // Read the "section names" section. reader.SetPosition(header.e_shoff + (ulong)header.e_shstrndex * header.e_shentsize); Elf64_Shdr namesHeader = new Elf64_Shdr(); namesHeader.Read(reader, big); // Reject files without real section names. if (namesHeader.sh_size == 0) { throw new ModuleException("This elf doesn't have section names."); } // Read the name table. byte[] nameTable; reader.SetPosition(namesHeader.sh_offset); reader.Read(out nameTable, (int)namesHeader.sh_size); // Move to the section header table. reader.SetPosition(header.e_shoff); // Read the section until hit '.cbm'. Elf64_Shdr sectionHeader = new Elf64_Shdr(); bool found = false; StringBuilder builder = new StringBuilder(); for (int i = 0; i < header.e_shnum; ++i) { // Read the section header. sectionHeader.Read(reader, big); // Check the section name. if (sectionHeader.sh_name >= nameTable.Length) { throw new ModuleException("Invalid section name."); } // Build the section name. builder.Length = 0; int pos = (int)sectionHeader.sh_name; while (nameTable[pos] != 0) { builder.Append((char)nameTable[pos++]); } string sectionName = builder.ToString(); // Compare the section name. if (sectionName == ".cbm") { found = true; break; } // Skip the extra data. reader.Skip(header.e_shentsize - Elf64_Shdr.Size); } // Make sure the section was found. if (!found) { throw new ModuleException("The elf doesn't have a chela module."); } // Make sure the section type is PROGBITS. if (sectionHeader.sh_type != (int)ElfSectionType.SHT_PROGBITS) { throw new ModuleException("The elf section that can have the module is not supported."); } // Move to the section offset. reader.SetPosition(sectionHeader.sh_offset); }
public static ChelaModule LoadModule(ModuleReader reader, bool fullLoad) { // Try to guess the file type. uint moduleStart = reader.GetPosition(); byte[] signature; reader.Read(out signature, 8); // Check for elf file type. bool isElf = true; for(int i = 0; i < 4; ++i) { if(signature[i] != ElfFormat.ElfMagic[i]) { isElf = false; break; } } // Check for PE. bool isPE = signature[0] == 'M' && signature[1] == 'Z'; // Find the module content. reader.SetPosition(moduleStart); if(isElf) { // Move to the embedded module. ElfFormat.GetEmbeddedModule(reader); reader.FixBase(); } else if(isPE) { // Move to the embedded module. PEFormat.GetEmbeddedModule(reader); reader.FixBase(); } // Read the module header. ModuleHeader header = new ModuleHeader(); header.Read(reader); // Create the output module. ChelaModule result = new ChelaModule(); // Read the string table. reader.SetPosition(header.stringTableOffset); for(uint i = 0; i < header.stringTableEntries; ++i) { uint id = i+1u; string s = reader.ReadString(); result.stringTable.Add(s); result.registeredStrings.Add(s, id); } // Read myself data. ModuleReference modref = new ModuleReference(); reader.SetPosition(header.moduleRefTableOffset); modref.Read(reader); result.SetName(result.GetString(modref.moduleName)); // Read the module type. result.moduleType = (ModuleType)header.moduleType; // Read the referenced modules. for(uint i = 1; i < header.moduleRefTableEntries; ++i) { // Read the referenced module. modref.Read(reader); // Load it. ChelaModule referencedModule = LoadNamedModule(result.GetString(modref.moduleName)); result.referencedModules.Add(referencedModule); } // Preload the members. reader.SetPosition(header.memberTableOffset); uint end = header.memberTableOffset + header.memberTableSize; MemberHeader mheader = new MemberHeader(); while(reader.GetPosition() < end) { // Read the member header. mheader.Read(reader); // Skip the attributes. SkipAttributes(reader, mheader.memberAttributes); // Preload the member. MemberHeaderType mtype = (MemberHeaderType)mheader.memberType; switch(mtype) { case MemberHeaderType.Namespace: Namespace.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Structure: Structure.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Class: Class.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Interface: Interface.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Function: Function.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Field: FieldVariable.PreloadMember(result, reader, mheader); break; case MemberHeaderType.FunctionGroup: FunctionGroup.PreloadMember(result, reader, mheader); break; case MemberHeaderType.TypeInstance: StructureInstance.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Property: PropertyVariable.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Event: EventVariable.PreloadMember(result, reader, mheader); break; case MemberHeaderType.TypeName: TypeNameMember.PreloadMember(result, reader, mheader); break; case MemberHeaderType.MemberInstance: case MemberHeaderType.FunctionInstance: // Ignore the member instance. result.memberTable.Add(null); reader.Skip(mheader.memberSize); break; case MemberHeaderType.TypeGroup: TypeGroup.PreloadMember(result, reader, mheader); break; case MemberHeaderType.Reference: { // Read the module id. byte moduleId = reader.ReadByte(); // Get the referenced module. ChelaModule refMod = result.GetReferenced(moduleId); // Find the member there. ScopeMember member = refMod.GetMember(result.GetString(mheader.memberName)); result.memberTable.Add(member); } break; default: // Unknown member. System.Console.WriteLine("add unknown member {0}", mtype); result.memberTable.Add(null); reader.Skip(mheader.memberSize); break; } } // Read the type table. reader.SetPosition(header.typeTableOffset); for(uint i = 0; i < header.typeTableEntries; ++i) { // Read the type reference. TypeReference typeRef = new TypeReference(); typeRef.Read(reader); // Try to load direct members. TypeKind kind = (ChelaModule.TypeKind)typeRef.typeKind; if(kind == TypeKind.Class || kind == TypeKind.Interface || kind == TypeKind.Structure || kind == TypeKind.Instance) { // Store the actual type. result.typeTable.Add(result.GetMember(typeRef.memberId)); } else { // Store the reference for delayed loading. result.typeTable.Add(typeRef); } } // Read the anonymous types. reader.SetPosition(header.anonTypeTableOffset); end = header.anonTypeTableOffset + header.anonTypeTableSize; while(reader.GetPosition() < end) { // Read the anonymous type data. AnonTypeHeader anonType = new AnonTypeHeader(); anonType.Read(reader); // Store it in the anonymous type table. result.anonymousTypes.Add(anonType); } // Now, read the actual members. reader.SetPosition(header.memberTableOffset); for(int i = 0; i < result.memberTable.Count; ++i) { // Get the preloaded member. ScopeMember member = result.memberTable[i]; // Read his header. mheader.Read(reader); // Skip the attributes. SkipAttributes(reader, mheader.memberAttributes); // Read the member. if(member != null && mheader.memberType != (int)MemberHeaderType.Reference) member.Read(reader, mheader); else reader.Skip(mheader.memberSize); // Skip unknown members. } // The first member must be the global namespace. result.globalNamespace = (Namespace)result.GetMember(1); // Update the parent children relationship. result.globalNamespace.UpdateParent(null); // Now, perform second read pass, required by function groups. reader.SetPosition(header.memberTableOffset); for(int i = 0; i < result.memberTable.Count; ++i) { // Get the preloaded member. ScopeMember member = result.memberTable[i]; // Read his header. mheader.Read(reader); // Skip the attributes. SkipAttributes(reader, mheader.memberAttributes); // Read the member. if(member != null && mheader.memberType != (int)MemberHeaderType.Reference) member.Read2(reader, mheader); else reader.Skip(mheader.memberSize); // Skip unknown members. } // Finish all of the loading. result.globalNamespace.FinishLoad(); // Register the member names. for(int i = 0; i < result.memberTable.Count; ++i) { ScopeMember member = result.memberTable[i]; if(member != null) result.memberNameTable[member.GetFullName()] = member; } // Load the runtime classes. if(result.referencedModules.Count == 0) result.LoadRuntimeReferences(result); // Dump the readed module. //result.Dump(); // Return the result module. return result; }
private static void GetEmbeddedModule64(ModuleReader reader, byte[] ident) { // Read the endiannes. bool big = ident[(int)ElfIdent.EI_DATA] == (int)ElfData.ELFDATA2MSB; // Read the header. Elf64_Ehdr header = new Elf64_Ehdr(); header.Read(reader, big); // Reject files without section names. if(header.e_shstrndex == (int)ElfSectionNumber.SHN_UNDEF) throw new ModuleException("Unsupported elfs without section names"); // Read the "section names" section. reader.SetPosition(header.e_shoff + (ulong)header.e_shstrndex*header.e_shentsize); Elf64_Shdr namesHeader = new Elf64_Shdr(); namesHeader.Read(reader, big); // Reject files without real section names. if(namesHeader.sh_size == 0) throw new ModuleException("This elf doesn't have section names."); // Read the name table. byte[] nameTable; reader.SetPosition(namesHeader.sh_offset); reader.Read(out nameTable, (int)namesHeader.sh_size); // Move to the section header table. reader.SetPosition(header.e_shoff); // Read the section until hit '.cbm'. Elf64_Shdr sectionHeader = new Elf64_Shdr(); bool found = false; StringBuilder builder = new StringBuilder(); for(int i = 0; i < header.e_shnum; ++i) { // Read the section header. sectionHeader.Read(reader, big); // Check the section name. if(sectionHeader.sh_name >= nameTable.Length) throw new ModuleException("Invalid section name."); // Build the section name. builder.Length = 0; int pos = (int)sectionHeader.sh_name; while(nameTable[pos] != 0) builder.Append((char)nameTable[pos++]); string sectionName = builder.ToString(); // Compare the section name. if(sectionName == ".cbm") { found = true; break; } // Skip the extra data. reader.Skip(header.e_shentsize - Elf64_Shdr.Size); } // Make sure the section was found. if(!found) throw new ModuleException("The elf doesn't have a chela module."); // Make sure the section type is PROGBITS. if(sectionHeader.sh_type != (int)ElfSectionType.SHT_PROGBITS) throw new ModuleException("The elf section that can have the module is not supported."); // Move to the section offset. reader.SetPosition(sectionHeader.sh_offset); }