首页 诗词 字典 板报 句子 名言 友答 励志 学校 网站地图
当前位置: 首页 > 教程频道 > .NET > .NET >

请帮忙修改下代码,增加区段大小。结贴100%,多谢了

2013-11-01 
请帮忙修改下代码,增加区段大小。。。。。。。。。。。。。。。结贴100%,谢谢了这个代码是在网上找到,实现给PE文件增加区

请帮忙修改下代码,增加区段大小。。。。。。。。。。。。。。。结贴100%,谢谢了
这个代码是在网上找到,
实现给PE文件增加区段,
对应的参数是   文件路径,新建区段名和新建的区段大小


请帮忙修改一下,修改成不增加区段,而是在最后一个区段的末尾加上0x2000大小的空间。
也就是相当于把新建的区段的空间加到最后一个区段的末尾。



源码如下:修改后可发送到 xiaoruirui01@163.com
也可以直接回贴。
谢谢了
结贴率100%




procedure AddSection(lFileName,secname,secsize: string; lBackup: boolean); //打开exe文件,是否备份
var
    hFile: THandle; //文件句柄
    ImageDosHeader: IMAGE_DOS_HEADER; //DOS部首
    ImageNtHeaders: IMAGE_NT_HEADERS; //映象头
    ImageSectionHeader: IMAGE_SECTION_HEADER; //块表
    lPointerToRawData: dword; //指向文件中的偏移
    lVirtualAddress: dword; //指向内存中的偏移
    i: integer; //循环变量
    p: integer;//自定义区段的字符串的长
    S: String;//除去自定义区段空格


    BytesRead, ByteSWrite: Cardinal; //读写用参数
    AttachSize: dword; //附加段大小
    AttachData, ChangeData: integer; //附加段填充数据
    OEP: integer; //使用过程中用到的OEP
    lpBuffer: array[0..1024 * 400] of byte; {数据存储缓冲区}
    StartEN, SizeEN, StartCr: dword; //PE修改后的物理地址和大小
    ret: boolean;

begin
    

  //定义附加段填充数据
 AttachData := 0;
 S := Trim(secname);//除去自定义区段空格
 AttachSize := strtoint(secsize);//获取自定义的长度
 p :=Length(S);//得到自定义区段名字长度
 SetLength(arrText, p+2); //动态数组初始化
 SetLength(arrChar,p);

 arrText[0] :='.';
 for   i:=0   to   p-1  do
    begin
    arrText[i+1] :=secname[i+1] //动态数组下标从0开始,String类数组下标从1开始
    end;
    arrText[p+1] := #0;

   //打开文件
    hFile := CreateFile(PChar(lFileName), GENERIC_READ or GENERIC_WRITE, FILE_SHARE_READ or FILE_SHARE_WRITE, nil, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);

  //校验
  if hFile = INVALID_HANDLE_VALUE then 
    begin
       ShowMessage('打开文件失败');
      exit;
   end;

  //确认备份
  if lBackup then CopyFile(PChar(lFileName), PChar(lFileName + '.bak'), False);
    ret := False;
    try

 //读取DOS部首到ImageDosHeader
    ReadFile(hFile, ImageDosHeader, SizeOf(ImageDosHeader), BytesRead, nil);

     //校验
   if ImageDosHeader.e_magic <> IMAGE_DOS_SIGNATURE then
        begin
            ShowMessage('不是有效的PE文件!');
            exit;
        end;

    //指向映象头
        SetFilePointer(hFile, ImageDosHeader._lfanew, nil, FILE_BEGIN); 

   //读取映向头到ImageNtHeaders
        ReadFile(hFile, ImageNtHeaders, SizeOf(ImageNtHeaders), BytesRead, nil);

    //校验
   if ImageNtHeaders.Signature <> IMAGE_NT_SIGNATURE then
        begin
           ShowMessage('不是有效的PE文件');
           exit;
        end;
{********************************}
{OEP=基址+原EP}
        OEP := ImageNtHeaders.OptionalHeader.ImageBase + ImageNtHeaders.OptionalHeader.AddressOfEntryPoint;

 {********************************}




   //初始化文件中偏移和映象中偏移
        lPointerToRawData := 0;
        lVirtualAddress := 0;

        for i := 0 to ImageNtHeaders.FileHeader.NumberOfSections - 1 do
        begin

      //读取块表中信息
            ReadFile(hFile, ImageSectionHeader, SizeOf(ImageSectionHeader), BytesRead, nil);

{********************************}
{查找原EP所在区段(原EP所在区段),记录物理偏移(初始地址),物理大小(长度)}
        if ImageNtHeaders.OptionalHeader.AddressOfEntryPoint > ImageSectionHeader.VirtualAddress then;
            begin
                StartEN := ImageSectionHeader.PointerToRawData;


                SecData := StartEN;
                SizeEN := ImageSectionHeader.SizeOfRawData; 
                StartCr := ImageNtHeaders.OptionalHeader.ImageBase + ImageSectionHeader.VirtualAddress;
           end;

 {********************************}

  //计算文件中偏移
            if lPointerToRawData < ImageSectionHeader.PointerToRawData + ImageSectionHeader.SizeOfRawData then
                lPointerToRawData := ImageSectionHeader.PointerToRawData + ImageSectionHeader.SizeOfRawData;

      //计算映象中偏移
      if lVirtualAddress < ImageSectionHeader.VirtualAddress + ImageSectionHeader.Misc.VirtualSize
 then

                lVirtualAddress := ImageSectionHeader.VirtualAddress + ImageSectionHeader.Misc.VirtualSize;
        end;

{增加块,定义块各项属性}


        Move(arrText[0], ImageSectionHeader.Name, p+2);

  //设置初始属性
        ImageSectionHeader.Misc.VirtualSize := AttachSize;
        ImageSectionHeader.VirtualAddress := lVirtualAddress;
        ImageSectionHeader.SizeOfRawData := AttachSize;
        ImageSectionHeader.PointerToRawData := lPointerToRawData;
        ImageSectionHeader.PointerToRelocations := 0;
        ImageSectionHeader.PointerToLinenumbers := 0;
        ImageSectionHeader.NumberOfRelocations := 0;

    //校正新节物理偏移(物理区块对齐)
        if ImageSectionHeader.VirtualAddress mod ImageNtHeaders.OptionalHeader.SectionAlignment > 0 then
          ImageSectionHeader.VirtualAddress := (ImageSectionHeader.VirtualAddress div ImageNtHeaders.OptionalHeader.SectionAlignment + 1) * ImageNtHeaders.OptionalHeader.SectionAlignment;

    //校正新节映象偏移(映象中区块对齐)
        if ImageSectionHeader.Misc.VirtualSize mod ImageNtHeaders.OptionalHeader.SectionAlignment > 0 then
            ImageSectionHeader.Misc.VirtualSize := (ImageSectionHeader.Misc.VirtualSize div ImageNtHeaders.OptionalHeader.SectionAlignment + 1) * ImageNtHeaders.OptionalHeader.SectionAlignment;

  //设置区块属性
        ImageSectionHeader.Characteristics := $E00000E0;

    //保存区块信息
        WriteFile(hFile, ImageSectionHeader, SizeOf(ImageSectionHeader), ByteSWrite, nil);
 //校正内存映象大小
    ImageNtHeaders.OptionalHeader.SizeOfImage := ImageNtHeaders.OptionalHeader.SizeOfImage + ImageSectionHeader.Misc.VirtualSize;
  //更
    //校正块数目
      Inc(ImageNtHeaders.FileHeader.NumberOfSections);

    //定位到映象头
      SetFilePointer(hFile, ImageDosHeader._lfanew, nil, FILE_BEGIN);

  //保存校正过的映象头
       WriteFile(hFile, ImageNtHeaders, SizeOf(ImageNtHeaders), ByteSWrite, nil);

    //定位到新节开始处
       SetFilePointer(hFile, ImageSectionHeader.PointerToRawData, nil, FILE_BEGIN);

    //用00数据填充满新节
        for i := 1 to AttachSize do
        begin 
            WriteFile(hFile, PByte(@AttachData)^, 1, ByteSWrite, nil);
        end;

  {填充自定义数据}
    //指向新节开始处
        SetFilePointer(hFile, ImageSectionHeader.PointerToRawData, nil, FILE_BEGIN);

    
        WriteFile(hFile, OEP, 4, ByteSWrite, nil); //跳回OEP

    //没有异常,显示增加区块成功!
        ShowMessage('增加区块成功!');

    finally

 {8.退出}
   //关闭文件
     CloseHandle(hFile);
   end;

end;


[解决办法]
这个没弄过。帮顶把

热点排行