streams.inc 28 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198
  1. {
  2. This file is part of the Free Component Library (FCL)
  3. Copyright (c) 1999-2000 by the Free Pascal development team
  4. See the file COPYING.FPC, included in this distribution,
  5. for details about the copyright.
  6. This program is distributed in the hope that it will be useful,
  7. but WITHOUT ANY WARRANTY; without even the implied warranty of
  8. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  9. **********************************************************************}
  10. {****************************************************************************}
  11. {* TStream *}
  12. {****************************************************************************}
  13. procedure TStream.ReadNotImplemented;
  14. begin
  15. raise EStreamError.CreateFmt(SStreamNoReading, [ClassName]) at get_caller_addr(get_frame);
  16. end;
  17. procedure TStream.WriteNotImplemented;
  18. begin
  19. raise EStreamError.CreateFmt(SStreamNoWriting, [ClassName]) at get_caller_addr(get_frame);
  20. end;
  21. function TStream.Read(var Buffer; Count: Longint): Longint;
  22. begin
  23. ReadNotImplemented;
  24. Result := 0;
  25. end;
  26. function TStream.Write(const Buffer; Count: Longint): Longint;
  27. begin
  28. WriteNotImplemented;
  29. Result := 0;
  30. end;
  31. function TStream.GetPosition: Int64;
  32. begin
  33. Result:=Seek(0,soCurrent);
  34. end;
  35. procedure TStream.SetPosition(const Pos: Int64);
  36. begin
  37. Seek(pos,soBeginning);
  38. end;
  39. procedure TStream.SetSize64(const NewSize: Int64);
  40. begin
  41. // Required because can't use overloaded functions in properties
  42. SetSize(NewSize);
  43. end;
  44. function TStream.GetSize: Int64;
  45. var
  46. p : int64;
  47. begin
  48. p:=Seek(0,soCurrent);
  49. GetSize:=Seek(0,soEnd);
  50. Seek(p,soBeginning);
  51. end;
  52. procedure TStream.SetSize(NewSize: Longint);
  53. begin
  54. // We do nothing. Pipe streams don't support this
  55. // As wel as possible read-ony streams !!
  56. end;
  57. procedure TStream.SetSize(const NewSize: Int64);
  58. begin
  59. // Backwards compatibility that calls the longint SetSize
  60. if (NewSize<Low(longint)) or
  61. (NewSize>High(longint)) then
  62. raise ERangeError.Create(SRangeError);
  63. SetSize(longint(NewSize));
  64. end;
  65. function TStream.Seek(Offset: Longint; Origin: Word): Longint;
  66. type
  67. TSeek64 = function(const offset:Int64;Origin:TSeekorigin):Int64 of object;
  68. var
  69. CurrSeek,
  70. TStreamSeek : TSeek64;
  71. CurrClass : TClass;
  72. begin
  73. // Redirect calls to 64bit Seek, but we can't call the 64bit Seek
  74. // from TStream, because then we end up in an infinite loop
  75. CurrSeek:=nil;
  76. CurrClass:=Classtype;
  77. while (CurrClass<>nil) and
  78. (CurrClass<>TStream) do
  79. CurrClass:=CurrClass.Classparent;
  80. if CurrClass<>nil then
  81. begin
  82. CurrSeek:[email protected];
  83. TStreamSeek:=@TStream(@CurrClass).Seek;
  84. if TMethod(TStreamSeek).Code=TMethod(CurrSeek).Code then
  85. CurrSeek:=nil;
  86. end;
  87. if CurrSeek<>nil then
  88. Result:=Seek(Int64(offset),TSeekOrigin(origin))
  89. else
  90. raise EStreamError.CreateFmt(SSeekNotImplemented,[ClassName]);
  91. end;
  92. procedure TStream.Discard(const Count: Int64);
  93. const
  94. CSmallSize =255;
  95. CLargeMaxBuffer =32*1024; // 32 KiB
  96. var
  97. Buffer: array[1..CSmallSize] of Byte;
  98. begin
  99. if Count=0 then
  100. Exit;
  101. if Count<=SizeOf(Buffer) then
  102. ReadBuffer(Buffer,Count)
  103. else
  104. DiscardLarge(Count,CLargeMaxBuffer);
  105. end;
  106. procedure TStream.DiscardLarge(Count: int64; const MaxBufferSize: Longint);
  107. var
  108. Buffer: array of Byte;
  109. begin
  110. if Count=0 then
  111. Exit;
  112. if Count>MaxBufferSize then
  113. SetLength(Buffer,MaxBufferSize)
  114. else
  115. SetLength(Buffer,Count);
  116. while (Count>=Length(Buffer)) do
  117. begin
  118. ReadBuffer(Buffer[0],Length(Buffer));
  119. Dec(Count,Length(Buffer));
  120. end;
  121. if Count>0 then
  122. ReadBuffer(Buffer[0],Count);
  123. end;
  124. procedure TStream.InvalidSeek;
  125. begin
  126. raise EStreamError.CreateFmt(SStreamInvalidSeek, [ClassName]) at get_caller_addr(get_frame);
  127. end;
  128. procedure TStream.FakeSeekForward(Offset: Int64; const Origin: TSeekOrigin; const Pos: Int64);
  129. var
  130. Buffer: Pointer;
  131. BufferSize, i: LongInt;
  132. begin
  133. if Origin=soBeginning then
  134. Dec(Offset,Pos);
  135. if (Offset<0) or (Origin=soEnd) then
  136. InvalidSeek;
  137. if Offset>0 then
  138. Discard(Offset);
  139. end;
  140. function TStream.Seek(const Offset: Int64; Origin: TSeekorigin): Int64;
  141. begin
  142. // Backwards compatibility that calls the longint Seek
  143. if (Offset<Low(longint)) or
  144. (Offset>High(longint)) then
  145. raise ERangeError.Create(SRangeError);
  146. Result:=Seek(longint(Offset),ord(Origin));
  147. end;
  148. procedure TStream.ReadBuffer(var Buffer; Count: Longint);
  149. begin
  150. if Read(Buffer,Count)<Count then
  151. Raise EReadError.Create(SReadError);
  152. end;
  153. procedure TStream.WriteBuffer(const Buffer; Count: Longint);
  154. begin
  155. if Write(Buffer,Count)<Count then
  156. Raise EWriteError.Create(SWriteError);
  157. end;
  158. function TStream.CopyFrom(Source: TStream; Count: Int64): Int64;
  159. var
  160. Buffer: Pointer;
  161. BufferSize, i: LongInt;
  162. const
  163. MaxSize = $20000;
  164. begin
  165. Result:=0;
  166. if Count=0 then
  167. Source.Position:=0; // This WILL fail for non-seekable streams...
  168. BufferSize:=MaxSize;
  169. if (Count>0) and (Count<BufferSize) then
  170. BufferSize:=Count; // do not allocate more than needed
  171. GetMem(Buffer,BufferSize);
  172. try
  173. if Count=0 then
  174. repeat
  175. i:=Source.Read(buffer^,BufferSize);
  176. if i>0 then
  177. WriteBuffer(buffer^,i);
  178. Inc(Result,i);
  179. until i<BufferSize
  180. else
  181. while Count>0 do
  182. begin
  183. if Count>BufferSize then
  184. i:=BufferSize
  185. else
  186. i:=Count;
  187. Source.ReadBuffer(buffer^,i);
  188. WriteBuffer(buffer^,i);
  189. Dec(count,i);
  190. Inc(Result,i);
  191. end;
  192. finally
  193. FreeMem(Buffer);
  194. end;
  195. end;
  196. function TStream.ReadComponent(Instance: TComponent): TComponent;
  197. var
  198. Reader: TReader;
  199. begin
  200. Reader := TReader.Create(Self, 4096);
  201. try
  202. Result := Reader.ReadRootComponent(Instance);
  203. finally
  204. Reader.Free;
  205. end;
  206. end;
  207. function TStream.ReadComponentRes(Instance: TComponent): TComponent;
  208. begin
  209. ReadResHeader;
  210. Result := ReadComponent(Instance);
  211. end;
  212. procedure TStream.WriteComponent(Instance: TComponent);
  213. begin
  214. WriteDescendent(Instance, nil);
  215. end;
  216. procedure TStream.WriteComponentRes(const ResName: string; Instance: TComponent);
  217. begin
  218. WriteDescendentRes(ResName, Instance, nil);
  219. end;
  220. procedure TStream.WriteDescendent(Instance, Ancestor: TComponent);
  221. var
  222. Driver : TAbstractObjectWriter;
  223. Writer : TWriter;
  224. begin
  225. Driver := TBinaryObjectWriter.Create(Self, 4096);
  226. Try
  227. Writer := TWriter.Create(Driver);
  228. Try
  229. Writer.WriteDescendent(Instance, Ancestor);
  230. Finally
  231. Writer.Destroy;
  232. end;
  233. Finally
  234. Driver.Free;
  235. end;
  236. end;
  237. procedure TStream.WriteDescendentRes(const ResName: string; Instance, Ancestor: TComponent);
  238. var
  239. FixupInfo: Integer;
  240. begin
  241. { Write a resource header }
  242. WriteResourceHeader(ResName, FixupInfo);
  243. { Write the instance itself }
  244. WriteDescendent(Instance, Ancestor);
  245. { Insert the correct resource size into the resource header }
  246. FixupResourceHeader(FixupInfo);
  247. end;
  248. procedure TStream.WriteResourceHeader(const ResName: string; {!!!: out} var FixupInfo: Integer);
  249. var
  250. ResType, Flags : word;
  251. begin
  252. ResType:=NtoLE(word($000A));
  253. Flags:=NtoLE(word($1030));
  254. { Note: This is a Windows 16 bit resource }
  255. { Numeric resource type }
  256. WriteByte($ff);
  257. { Application defined data }
  258. WriteWord(ResType);
  259. { write the name as asciiz }
  260. WriteBuffer(ResName[1],length(ResName));
  261. WriteByte(0);
  262. { Movable, Pure and Discardable }
  263. WriteWord(Flags);
  264. { Placeholder for the resource size }
  265. WriteDWord(0);
  266. { Return current stream position so that the resource size can be
  267. inserted later }
  268. FixupInfo := Position;
  269. end;
  270. procedure TStream.FixupResourceHeader(FixupInfo: Integer);
  271. var
  272. ResSize,TmpResSize : Integer;
  273. begin
  274. ResSize := Position - FixupInfo;
  275. TmpResSize := NtoLE(longword(ResSize));
  276. { Insert the correct resource size into the placeholder written by
  277. WriteResourceHeader }
  278. Position := FixupInfo - 4;
  279. WriteDWord(TmpResSize);
  280. { Seek back to the end of the resource }
  281. Position := FixupInfo + ResSize;
  282. end;
  283. procedure TStream.ReadResHeader;
  284. var
  285. ResType, Flags : word;
  286. begin
  287. try
  288. { Note: This is a Windows 16 bit resource }
  289. { application specific resource ? }
  290. if ReadByte<>$ff then
  291. raise EInvalidImage.Create(SInvalidImage);
  292. ResType:=LEtoN(ReadWord);
  293. if ResType<>$000a then
  294. raise EInvalidImage.Create(SInvalidImage);
  295. { read name }
  296. while ReadByte<>0 do
  297. ;
  298. { check the access specifier }
  299. Flags:=LEtoN(ReadWord);
  300. if Flags<>$1030 then
  301. raise EInvalidImage.Create(SInvalidImage);
  302. { ignore the size }
  303. ReadDWord;
  304. except
  305. on EInvalidImage do
  306. raise;
  307. else
  308. raise EInvalidImage.create(SInvalidImage);
  309. end;
  310. end;
  311. function TStream.ReadByte : Byte;
  312. var
  313. b : Byte;
  314. begin
  315. ReadBuffer(b,1);
  316. ReadByte:=b;
  317. end;
  318. function TStream.ReadWord : Word;
  319. var
  320. w : Word;
  321. begin
  322. ReadBuffer(w,2);
  323. ReadWord:=w;
  324. end;
  325. function TStream.ReadDWord : Cardinal;
  326. var
  327. d : Cardinal;
  328. begin
  329. ReadBuffer(d,4);
  330. ReadDWord:=d;
  331. end;
  332. function TStream.ReadQWord: QWord;
  333. var
  334. q: QWord;
  335. begin
  336. ReadBuffer(q,8);
  337. ReadQWord:=q;
  338. end;
  339. Function TStream.ReadAnsiString : String;
  340. Var
  341. TheSize : Longint;
  342. P : PByte ;
  343. begin
  344. ReadBuffer (TheSize,SizeOf(TheSize));
  345. SetLength(Result,TheSize);
  346. // Illegal typecast if no AnsiStrings defined.
  347. if TheSize>0 then
  348. begin
  349. ReadBuffer (Pointer(Result)^,TheSize);
  350. P:=Pointer(Result)+TheSize;
  351. p^:=0;
  352. end;
  353. end;
  354. Procedure TStream.WriteAnsiString (const S : String);
  355. Var L : Longint;
  356. begin
  357. L:=Length(S);
  358. WriteBuffer (L,SizeOf(L));
  359. WriteBuffer (Pointer(S)^,L);
  360. end;
  361. procedure TStream.WriteByte(b : Byte);
  362. begin
  363. WriteBuffer(b,1);
  364. end;
  365. procedure TStream.WriteWord(w : Word);
  366. begin
  367. WriteBuffer(w,2);
  368. end;
  369. procedure TStream.WriteDWord(d : Cardinal);
  370. begin
  371. WriteBuffer(d,4);
  372. end;
  373. procedure TStream.WriteQWord(q: QWord);
  374. begin
  375. WriteBuffer(q,8);
  376. end;
  377. {****************************************************************************}
  378. {* THandleStream *}
  379. {****************************************************************************}
  380. Constructor THandleStream.Create(AHandle: THandle);
  381. begin
  382. Inherited Create;
  383. FHandle:=AHandle;
  384. end;
  385. function THandleStream.Read(var Buffer; Count: Longint): Longint;
  386. begin
  387. Result:=FileRead(FHandle,Buffer,Count);
  388. If Result=-1 then Result:=0;
  389. end;
  390. function THandleStream.Write(const Buffer; Count: Longint): Longint;
  391. begin
  392. Result:=FileWrite (FHandle,Buffer,Count);
  393. If Result=-1 then Result:=0;
  394. end;
  395. Procedure THandleStream.SetSize(NewSize: Longint);
  396. begin
  397. SetSize(Int64(NewSize));
  398. end;
  399. Procedure THandleStream.SetSize(const NewSize: Int64);
  400. begin
  401. FileTruncate(FHandle,NewSize);
  402. end;
  403. function THandleStream.Seek(const Offset: Int64; Origin: TSeekOrigin): Int64;
  404. begin
  405. Result:=FileSeek(FHandle,Offset,ord(Origin));
  406. end;
  407. {****************************************************************************}
  408. {* TFileStream *}
  409. {****************************************************************************}
  410. constructor TFileStream.Create(const AFileName: string; Mode: Word);
  411. begin
  412. Create(AFileName,Mode,438);
  413. end;
  414. constructor TFileStream.Create(const AFileName: string; Mode: Word; Rights: Cardinal);
  415. begin
  416. FFileName:=AFileName;
  417. If (Mode and fmCreate) > 0 then
  418. FHandle:=FileCreate(AFileName,Mode,Rights)
  419. else
  420. FHAndle:=FileOpen(AFileName,Mode);
  421. If (THandle(FHandle)=feInvalidHandle) then
  422. If Mode=fmcreate then
  423. raise EFCreateError.createfmt(SFCreateError,[AFileName])
  424. else
  425. raise EFOpenError.Createfmt(SFOpenError,[AFilename]);
  426. end;
  427. destructor TFileStream.Destroy;
  428. begin
  429. FileClose(FHandle);
  430. end;
  431. {****************************************************************************}
  432. {* TCustomMemoryStream *}
  433. {****************************************************************************}
  434. procedure TCustomMemoryStream.SetPointer(Ptr: Pointer; ASize: PtrInt);
  435. begin
  436. FMemory:=Ptr;
  437. FSize:=ASize;
  438. end;
  439. function TCustomMemoryStream.GetSize: Int64;
  440. begin
  441. Result:=FSize;
  442. end;
  443. function TCustomMemoryStream.GetPosition: Int64;
  444. begin
  445. Result:=FPosition;
  446. end;
  447. function TCustomMemoryStream.Read(var Buffer; Count: LongInt): LongInt;
  448. begin
  449. Result:=0;
  450. If (FSize>0) and (FPosition<Fsize) and (FPosition>=0) then
  451. begin
  452. Result:=FSize-FPosition;
  453. If Result>Count then Result:=Count;
  454. Move ((FMemory+FPosition)^,Buffer,Result);
  455. FPosition:=Fposition+Result;
  456. end;
  457. end;
  458. function TCustomMemoryStream.Seek(const Offset: Int64; Origin: TSeekOrigin): Int64;
  459. begin
  460. Case Word(Origin) of
  461. soFromBeginning : FPosition:=Offset;
  462. soFromEnd : FPosition:=FSize+Offset;
  463. soFromCurrent : FPosition:=FPosition+Offset;
  464. end;
  465. Result:=FPosition;
  466. {$IFDEF DEBUG}
  467. if Result < 0 then
  468. raise Exception.Create('TCustomMemoryStream');
  469. {$ENDIF}
  470. end;
  471. procedure TCustomMemoryStream.SaveToStream(Stream: TStream);
  472. begin
  473. if FSize>0 then Stream.WriteBuffer (FMemory^,FSize);
  474. end;
  475. procedure TCustomMemoryStream.SaveToFile(const FileName: string);
  476. Var S : TFileStream;
  477. begin
  478. S:=TFileStream.Create (FileName,fmCreate);
  479. Try
  480. SaveToStream(S);
  481. finally
  482. S.free;
  483. end;
  484. end;
  485. {****************************************************************************}
  486. {* TMemoryStream *}
  487. {****************************************************************************}
  488. Const TMSGrow = 4096; { Use 4k blocks. }
  489. procedure TMemoryStream.SetCapacity(NewCapacity: PtrInt);
  490. begin
  491. SetPointer (Realloc(NewCapacity),Fsize);
  492. FCapacity:=NewCapacity;
  493. end;
  494. function TMemoryStream.Realloc(var NewCapacity: PtrInt): Pointer;
  495. begin
  496. If NewCapacity<0 Then
  497. NewCapacity:=0
  498. else
  499. begin
  500. // if growing, grow at least a quarter
  501. if (NewCapacity>FCapacity) and (NewCapacity < (5*FCapacity) div 4) then
  502. NewCapacity := (5*FCapacity) div 4;
  503. // round off to block size.
  504. NewCapacity := (NewCapacity + (TMSGrow-1)) and not (TMSGROW-1);
  505. end;
  506. // Only now check !
  507. If NewCapacity=FCapacity then
  508. Result:=FMemory
  509. else
  510. begin
  511. Result:=Reallocmem(FMemory,Newcapacity);
  512. If (Result=Nil) and (Newcapacity>0) then
  513. Raise EStreamError.Create(SMemoryStreamError);
  514. end;
  515. end;
  516. destructor TMemoryStream.Destroy;
  517. begin
  518. Clear;
  519. Inherited Destroy;
  520. end;
  521. procedure TMemoryStream.Clear;
  522. begin
  523. FSize:=0;
  524. FPosition:=0;
  525. SetCapacity (0);
  526. end;
  527. procedure TMemoryStream.LoadFromStream(Stream: TStream);
  528. begin
  529. Stream.Position:=0;
  530. SetSize(Stream.Size);
  531. If FSize>0 then Stream.ReadBuffer(FMemory^,FSize);
  532. end;
  533. procedure TMemoryStream.LoadFromFile(const FileName: string);
  534. Var S : TFileStream;
  535. begin
  536. S:=TFileStream.Create (FileName,fmOpenRead or fmShareDenyWrite);
  537. Try
  538. LoadFromStream(S);
  539. finally
  540. S.free;
  541. end;
  542. end;
  543. procedure TMemoryStream.SetSize({$ifdef CPU64}const{$endif CPU64} NewSize: PtrInt);
  544. begin
  545. SetCapacity (NewSize);
  546. FSize:=NewSize;
  547. IF FPosition>FSize then
  548. FPosition:=FSize;
  549. end;
  550. function TMemoryStream.Write(const Buffer; Count: LongInt): LongInt;
  551. Var NewPos : PtrInt;
  552. begin
  553. If (Count=0) or (FPosition<0) then
  554. exit(0);
  555. NewPos:=FPosition+Count;
  556. If NewPos>Fsize then
  557. begin
  558. IF NewPos>FCapacity then
  559. SetCapacity (NewPos);
  560. FSize:=Newpos;
  561. end;
  562. System.Move (Buffer,(FMemory+FPosition)^,Count);
  563. FPosition:=NewPos;
  564. Result:=Count;
  565. end;
  566. {****************************************************************************}
  567. {* TBytesStream *}
  568. {****************************************************************************}
  569. constructor TBytesStream.Create(const ABytes: TBytes);
  570. begin
  571. inherited Create;
  572. FBytes:=ABytes;
  573. SetPointer(Pointer(FBytes),Length(FBytes));
  574. FCapacity:=Length(FBytes);
  575. end;
  576. function TBytesStream.Realloc(var NewCapacity: PtrInt): Pointer;
  577. begin
  578. // adapt TMemoryStream code to use with dynamic array
  579. if NewCapacity<0 Then
  580. NewCapacity:=0
  581. else
  582. begin
  583. if (NewCapacity>Capacity) and (NewCapacity < (5*Capacity) div 4) then
  584. NewCapacity := (5*Capacity) div 4;
  585. NewCapacity := (NewCapacity + (TMSGrow-1)) and not (TMSGROW-1);
  586. end;
  587. if NewCapacity=Capacity then
  588. Result:=Pointer(FBytes)
  589. else
  590. begin
  591. SetLength(FBytes,Newcapacity);
  592. Result:=Pointer(FBytes);
  593. if (Result=nil) and (Newcapacity>0) then
  594. raise EStreamError.Create(SMemoryStreamError);
  595. end;
  596. end;
  597. {****************************************************************************}
  598. {* TStringStream *}
  599. {****************************************************************************}
  600. function TStringStream.GetSize: Int64;
  601. begin
  602. Result:=Length(FDataString);
  603. end;
  604. function TStringStream.GetPosition: Int64;
  605. begin
  606. Result:=FPosition;
  607. end;
  608. procedure TStringStream.SetSize(NewSize: Longint);
  609. begin
  610. Setlength(FDataString,NewSize);
  611. If FPosition>NewSize then FPosition:=NewSize;
  612. end;
  613. constructor TStringStream.Create(const AString: string);
  614. begin
  615. Inherited create;
  616. FDataString:=AString;
  617. end;
  618. function TStringStream.Read(var Buffer; Count: Longint): Longint;
  619. begin
  620. Result:=Length(FDataString)-FPosition;
  621. If Result>Count then Result:=Count;
  622. // This supposes FDataString to be of type AnsiString !
  623. Move (Pchar(FDataString)[FPosition],Buffer,Result);
  624. FPosition:=FPosition+Result;
  625. end;
  626. function TStringStream.ReadString(Count: Longint): string;
  627. Var NewLen : Longint;
  628. begin
  629. NewLen:=Length(FDataString)-FPosition;
  630. If NewLen>Count then NewLen:=Count;
  631. SetLength(Result,NewLen);
  632. Read (Pointer(Result)^,NewLen);
  633. end;
  634. function TStringStream.Seek(Offset: Longint; Origin: Word): Longint;
  635. begin
  636. Case Origin of
  637. soFromBeginning : FPosition:=Offset;
  638. soFromEnd : FPosition:=Length(FDataString)+Offset;
  639. soFromCurrent : FPosition:=FPosition+Offset;
  640. end;
  641. If FPosition>Length(FDataString) then
  642. FPosition:=Length(FDataString)
  643. else If FPosition<0 then
  644. FPosition:=0;
  645. Result:=FPosition;
  646. end;
  647. function TStringStream.Write(const Buffer; Count: Longint): Longint;
  648. begin
  649. Result:=Count;
  650. SetSize(FPosition+Count);
  651. // This supposes that FDataString is of type AnsiString)
  652. Move (Buffer,PChar(FDataString)[Fposition],Count);
  653. FPosition:=FPosition+Count;
  654. end;
  655. procedure TStringStream.WriteString(const AString: string);
  656. begin
  657. Write (PChar(Astring)[0],Length(AString));
  658. end;
  659. {****************************************************************************}
  660. {* TResourceStream *}
  661. {****************************************************************************}
  662. {$ifdef UNICODE}
  663. procedure TResourceStream.Initialize(Instance: THandle; Name, ResType: PWideChar; NameIsID: Boolean);
  664. begin
  665. Res:=FindResource(Instance, Name, ResType);
  666. if Res=0 then
  667. if NameIsID then
  668. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  669. else
  670. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  671. Handle:=LoadResource(Instance,Res);
  672. if Handle=0 then
  673. if NameIsID then
  674. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  675. else
  676. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  677. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  678. end;
  679. constructor TResourceStream.Create(Instance: THandle; const ResName: WideString; ResType: PWideChar);
  680. begin
  681. inherited create;
  682. Initialize(Instance,PWideChar(ResName),ResType,False);
  683. end;
  684. constructor TResourceStream.CreateFromID(Instance: THandle; ResID: Integer; ResType: PWideChar);
  685. begin
  686. inherited create;
  687. Initialize(Instance,PWideChar(ResID),ResType,True);
  688. end;
  689. {$else UNICODE}
  690. procedure TResourceStream.Initialize(Instance: THandle; Name, ResType: PChar; NameIsID: Boolean);
  691. begin
  692. Res:=FindResource(Instance, Name, ResType);
  693. if Res=0 then
  694. if NameIsID then
  695. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  696. else
  697. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  698. Handle:=LoadResource(Instance,Res);
  699. if Handle=0 then
  700. if NameIsID then
  701. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  702. else
  703. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  704. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  705. end;
  706. constructor TResourceStream.Create(Instance: THandle; const ResName: string; ResType: PChar);
  707. begin
  708. inherited create;
  709. Initialize(Instance,pchar(ResName),ResType,False);
  710. end;
  711. constructor TResourceStream.CreateFromID(Instance: THandle; ResID: Integer; ResType: PChar);
  712. begin
  713. inherited create;
  714. Initialize(Instance,pchar(PtrInt(ResID)),ResType,True);
  715. end;
  716. {$endif UNICODE}
  717. destructor TResourceStream.Destroy;
  718. begin
  719. UnlockResource(Handle);
  720. FreeResource(Handle);
  721. inherited destroy;
  722. end;
  723. {****************************************************************************}
  724. {* TOwnerStream *}
  725. {****************************************************************************}
  726. constructor TOwnerStream.Create(ASource: TStream);
  727. begin
  728. FSource:=ASource;
  729. end;
  730. destructor TOwnerStream.Destroy;
  731. begin
  732. If FOwner then
  733. FreeAndNil(FSource);
  734. inherited Destroy;
  735. end;
  736. {****************************************************************************}
  737. {* TStreamAdapter *}
  738. {****************************************************************************}
  739. constructor TStreamAdapter.Create(Stream: TStream; Ownership: TStreamOwnership = soReference);
  740. begin
  741. inherited Create;
  742. FStream:=Stream;
  743. FOwnership:=Ownership;
  744. m_bReverted:=false; // mantis 15003
  745. // http://www.tech-archive.net/Archive/German/microsoft.public.de.vc/2005-08/msg00791.html
  746. // http://code.google.com/p/ddab-lib/wiki/TPJIStreamWrapper
  747. end;
  748. destructor TStreamAdapter.Destroy;
  749. begin
  750. if StreamOwnership=soOwned then
  751. FreeAndNil(FStream);
  752. inherited Destroy;
  753. end;
  754. {$warnings off}
  755. function TStreamAdapter.Read(pv: Pointer; cb: DWORD; pcbRead: PDWORD): HResult; stdcall;
  756. var
  757. readcount: Longint;
  758. begin
  759. if m_bReverted then
  760. begin
  761. Result := STG_E_REVERTED;
  762. Exit;
  763. end;
  764. if pv = nil then
  765. begin
  766. Result := STG_E_INVALIDPOINTER;
  767. Exit;
  768. end;
  769. readcount := FStream.Read(pv^, cb);
  770. if pcbRead <> nil then pcbRead^ := readcount;
  771. Result := S_OK;
  772. end;
  773. function TStreamAdapter.Write(pv: Pointer; cb: DWORD; pcbWritten: PDWORD): HResult; stdcall;
  774. var
  775. writecount: Longint;
  776. begin
  777. if m_bReverted then
  778. begin
  779. Result := STG_E_REVERTED;
  780. Exit;
  781. end;
  782. if pv = nil then
  783. begin
  784. Result := STG_E_INVALIDPOINTER;
  785. Exit;
  786. end;
  787. writecount := FStream.Write(pv^, cb);
  788. if pcbWritten <> nil then pcbWritten^ := writecount;
  789. Result := S_OK;
  790. end;
  791. function TStreamAdapter.Seek(dlibMove: Largeint; dwOrigin: Longint; out libNewPosition: Largeint): HResult; stdcall;
  792. var
  793. newpos: Int64;
  794. begin
  795. if m_bReverted then
  796. begin
  797. Result := STG_E_REVERTED;
  798. Exit;
  799. end;
  800. case dwOrigin of
  801. STREAM_SEEK_SET: newpos := FStream.Seek(dlibMove, soBeginning);
  802. STREAM_SEEK_CUR: newpos := FStream.Seek(dlibMove, soCurrent);
  803. STREAM_SEEK_END: newpos := FStream.Seek(dlibMove, soEnd);
  804. else
  805. begin
  806. Result := STG_E_INVALIDFUNCTION;
  807. Exit;
  808. end;
  809. end;
  810. if @libNewPosition <> nil then
  811. libNewPosition := newpos;
  812. Result := S_OK;
  813. end;
  814. function TStreamAdapter.SetSize(libNewSize: Largeint): HResult; stdcall;
  815. begin
  816. if m_bReverted then
  817. begin
  818. Result := STG_E_REVERTED;
  819. Exit;
  820. end;
  821. if libNewSize<0 then
  822. begin
  823. Result := STG_E_INVALIDFUNCTION;
  824. Exit;
  825. end;
  826. try
  827. FStream.Size := libNewSize;
  828. Result := S_OK;
  829. except
  830. // TODO: return different error value according to exception like STG_E_MEDIUMFULL
  831. Result := E_FAIL;
  832. end;
  833. end;
  834. function TStreamAdapter.CopyTo(stm: IStream; cb: Largeint; out cbRead: Largeint; out cbWritten: Largeint): HResult; stdcall;
  835. var
  836. sz: dword;
  837. buffer : array[0..1023] of byte;
  838. begin
  839. if m_bReverted then
  840. begin
  841. Result := STG_E_REVERTED;
  842. Exit;
  843. end;
  844. // the method is similar to TStream.CopyFrom => use CopyFrom implementation
  845. cbWritten := 0;
  846. cbRead := 0;
  847. while cb > 0 do
  848. begin
  849. if (cb > sizeof(buffer)) then
  850. sz := sizeof(Buffer)
  851. else
  852. sz := cb;
  853. sz := FStream.Read(buffer, sz);
  854. inc(cbRead, sz);
  855. stm.Write(@buffer[0], sz, @sz);
  856. inc(cbWritten, sz);
  857. if sz = 0 then
  858. begin
  859. Result := E_FAIL;
  860. Exit;
  861. end;
  862. dec(cb, sz);
  863. end;
  864. Result := S_OK;
  865. end;
  866. function TStreamAdapter.Commit(grfCommitFlags: Longint): HResult; stdcall;
  867. begin
  868. if m_bReverted then
  869. Result := STG_E_REVERTED
  870. else
  871. Result := S_OK;
  872. end;
  873. function TStreamAdapter.Revert: HResult; stdcall;
  874. begin
  875. m_bReverted := True;
  876. Result := S_OK;
  877. end;
  878. function TStreamAdapter.LockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  879. begin
  880. Result := STG_E_INVALIDFUNCTION;
  881. end;
  882. function TStreamAdapter.UnlockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  883. begin
  884. Result := STG_E_INVALIDFUNCTION;
  885. end;
  886. function TStreamAdapter.Stat(out statstg: TStatStg; grfStatFlag: Longint): HResult; stdcall;
  887. begin
  888. if m_bReverted then
  889. begin
  890. Result := STG_E_REVERTED;
  891. Exit;
  892. end;
  893. if grfStatFlag in [STATFLAG_DEFAULT,STATFLAG_NOOPEN,STATFLAG_NONAME] then
  894. begin
  895. if @statstg <> nil then
  896. begin
  897. fillchar(statstg, sizeof(TStatStg),#0);
  898. { //TODO handle pwcsName
  899. if grfStatFlag = STATFLAG_DEFAULT then
  900. runerror(217) //Result :={$ifdef windows} STG_E_INVALIDFLAG{$else}E_INVALID_FLAG{$endif}
  901. }
  902. statstg.dwType := STGTY_STREAM;
  903. statstg.cbSize := FStream.Size;
  904. statstg.grfLocksSupported := LOCK_WRITE;
  905. end;
  906. Result := S_OK;
  907. end else
  908. Result := STG_E_INVALIDFLAG
  909. end;
  910. function TStreamAdapter.Clone(out stm: IStream): HResult; stdcall;
  911. begin
  912. if m_bReverted then
  913. begin
  914. Result := STG_E_REVERTED;
  915. Exit;
  916. end;
  917. // don't raise an exception here return error value that function is not implemented
  918. // to implement this we need a clone method for TStream class
  919. Result := STG_E_UNIMPLEMENTEDFUNCTION;
  920. end;
  921. constructor TProxyStream.Create(const Stream: IStream);
  922. begin
  923. FStream := Stream;
  924. end;
  925. function TProxyStream.Read(var Buffer; Count: Longint): Longint;
  926. begin
  927. Check(FStream.Read(@Buffer, Count, @Result));
  928. end;
  929. function TProxyStream. Seek(const Offset: int64; Origin: TSeekOrigin): int64;
  930. begin
  931. Check(FStream.Seek(Offset, ord(Origin), result));
  932. end;
  933. function TProxyStream.Write(const Buffer; Count: Longint): Longint;
  934. begin
  935. Check(FStream.Write(@Buffer, Count, @Result));
  936. end;
  937. function TProxyStream.GetIStream: IStream;
  938. begin
  939. Result := FStream;
  940. end;
  941. procedure TProxyStream.Check(err:integer);
  942. var e : EInOutError;
  943. begin
  944. e:= EInOutError.Create('Proxystream.Check');
  945. e.Errorcode:=err;
  946. raise e;
  947. end;
  948. {$warnings on}