src/game/g_res_downloader.pas

   1 unit g_res_downloader;
   2
   3 interface
   4
   5 uses sysutils, Classes, md5, g_net, g_netmsg, g_console, g_main, e_log;
   6
   7 function g_Res_SearchSameWAD(const path, filename: string; const resMd5: TMD5Digest): string;
   8 function g_Res_DownloadWAD(const FileName: string): string;
   9
  10 implementation
  11
  12 uses g_language, sfs, WADEDITOR;
  13
  14 const DOWNLOAD_DIR = 'downloads';
  15
  16 procedure FindFiles(const dirName, filename: string; var files: TStringList);
  17 var
  18   searchResult: TSearchRec;
  19 begin
  20   if FindFirst(dirName+'/*', faAnyFile, searchResult) = 0 then
  21   begin
  22     try
  23       repeat
  24         if (searchResult.Attr and faDirectory) = 0 then
  25         begin
  26           if SFSStrEqu(searchResult.Name, filename) then
  27           begin
  28             files.Add(dirName+'/'+filename);
  29             Exit;
  30           end;
  31         end
  32         else if (searchResult.Name <> '.') and (searchResult.Name <> '..') then
  33           FindFiles(IncludeTrailingPathDelimiter(dirName)+searchResult.Name, filename, files);
  34       until FindNext(searchResult) <> 0;
  35     finally
  36       FindClose(searchResult);
  37     end;
  38   end;
  39 end;
  40
  41 function CompareFileHash(const filename: string; const resMd5: TMD5Digest): Boolean;
  42 var
  43   gResHash: TMD5Digest;
  44   fname: string;
  45 begin
  46   fname := findDiskWad(filename);
  47   if length(fname) = 0 then begin result := false; exit; end;
  48   gResHash := MD5File(fname);
  49   Result := MD5Match(gResHash, resMd5);
  50 end;
  51
  52 function CheckFileHash(const path, filename: string; const resMd5: TMD5Digest): Boolean;
  53 var
  54   fname: string;
  55 begin
  56   fname := findDiskWad(path+filename);
  57   if length(fname) = 0 then begin result := false; exit; end;
  58   Result := FileExists(fname) and CompareFileHash(fname, resMd5);
  59 end;
  60
  61 function g_Res_SearchSameWAD(const path, filename: string; const resMd5: TMD5Digest): string;
  62 var
  63   res: string;
  64   files: TStringList;
  65   i: Integer;
  66 begin
  67   Result := '';
  68
  69   if CheckFileHash(path, filename, resMd5) then
  70   begin
  71     Result := path + filename;
  72     Exit;
  73   end;
  74
  75   files := TStringList.Create;
  76
  77   FindFiles(path, filename, files);
  78   for i := 0 to files.Count - 1 do
  79   begin
  80     res := files.Strings[i];
  81     if CompareFileHash(res, resMd5) then
  82     begin
  83       Result := res;
  84       Break;
  85     end;
  86   end;
  87
  88   files.Free;
  89 end;
  90
  91 function SaveWAD(const path, filename: string; const data: array of Byte): string;
  92 var
  93   resFile: TFileStream;
  94 begin
  95   try
  96     Result := path + DOWNLOAD_DIR + '/' + filename;
  97     if not DirectoryExists(path + DOWNLOAD_DIR) then
  98     begin
  99       CreateDir(path + DOWNLOAD_DIR);
 100     end;
 101     resFile := TFileStream.Create(Result, fmCreate);
 102     resFile.WriteBuffer(data[0], Length(data));
 103     resFile.Free
 104   except
 105     Result := '';
 106   end;
 107 end;
 108
 109 function g_Res_DownloadWAD(const FileName: string): string;
 110 var
 111   msgStream: TMemoryStream;
 112   resStream: TFileStream;
 113   mapData: TMapDataMsg;
 114   i: Integer;
 115   resData: TResDataMsg;
 116 begin
 117   SetLength(mapData.ExternalResources, 0);
 118   g_Console_Add(Format(_lc[I_NET_MAP_DL], [FileName]));
 119   e_WriteLog('Downloading map `' + FileName + '` from server', MSG_NOTIFY);
 120   MC_SEND_MapRequest();
 121
 122   msgStream := g_Net_Wait_Event(NET_MSG_MAP_RESPONSE);
 123   if msgStream <> nil then
 124   begin
 125     mapData := MapDataFromMsgStream(msgStream);
 126     msgStream.Free;
 127   end;
 128
 129   for i := 0 to High(mapData.ExternalResources) do
 130   begin
 131     if not CheckFileHash(GameDir + '/wads/',
 132                          mapData.ExternalResources[i].Name,
 133                          mapData.ExternalResources[i].md5) then
 134     begin
 135       g_Console_Add(Format(_lc[I_NET_WAD_DL],
 136                            [mapData.ExternalResources[i].Name]));
 137       e_WriteLog('Downloading Wad `' + mapData.ExternalResources[i].Name +
 138                  '` from server', MSG_NOTIFY);
 139       MC_SEND_ResRequest(mapData.ExternalResources[i].Name);
 140
 141       msgStream := g_Net_Wait_Event(NET_MSG_RES_RESPONSE);
 142       resData := ResDataFromMsgStream(msgStream);
 143
 144       resStream := TFileStream.Create(GameDir+'/wads/'+
 145                                       mapData.ExternalResources[i].Name,
 146                                       fmCreate);
 147       resStream.WriteBuffer(resData.FileData[0], resData.FileSize);
 148
 149       resData.FileData := nil;
 150       resStream.Free;
 151       msgStream.Free;
 152     end;
 153   end;
 154
 155   Result := SaveWAD(MapsDir, ExtractFileName(FileName), mapData.FileData);
 156 end;
 157
 158 end.