streams.inc 28 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201
  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), get_caller_frame(get_frame);
  16. end;
  17. procedure TStream.WriteNotImplemented;
  18. begin
  19. raise EStreamError.CreateFmt(SStreamNoWriting, [ClassName]) at get_caller_addr(get_frame), get_caller_frame(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), get_caller_frame(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:=Count;
  453. If (Result>(FSize-FPosition)) then
  454. Result:=(FSize-FPosition);
  455. Move ((FMemory+FPosition)^,Buffer,Result);
  456. FPosition:=Fposition+Result;
  457. end;
  458. end;
  459. function TCustomMemoryStream.Seek(const Offset: Int64; Origin: TSeekOrigin): Int64;
  460. begin
  461. Case Word(Origin) of
  462. soFromBeginning : FPosition:=Offset;
  463. soFromEnd : FPosition:=FSize+Offset;
  464. soFromCurrent : FPosition:=FPosition+Offset;
  465. end;
  466. Result:=FPosition;
  467. {$IFDEF DEBUG}
  468. if Result < 0 then
  469. raise Exception.Create('TCustomMemoryStream');
  470. {$ENDIF}
  471. end;
  472. procedure TCustomMemoryStream.SaveToStream(Stream: TStream);
  473. begin
  474. if FSize>0 then Stream.WriteBuffer (FMemory^,FSize);
  475. end;
  476. procedure TCustomMemoryStream.SaveToFile(const FileName: string);
  477. Var S : TFileStream;
  478. begin
  479. S:=TFileStream.Create (FileName,fmCreate);
  480. Try
  481. SaveToStream(S);
  482. finally
  483. S.free;
  484. end;
  485. end;
  486. {****************************************************************************}
  487. {* TMemoryStream *}
  488. {****************************************************************************}
  489. Const TMSGrow = 4096; { Use 4k blocks. }
  490. procedure TMemoryStream.SetCapacity(NewCapacity: PtrInt);
  491. begin
  492. SetPointer (Realloc(NewCapacity),Fsize);
  493. FCapacity:=NewCapacity;
  494. end;
  495. function TMemoryStream.Realloc(var NewCapacity: PtrInt): Pointer;
  496. begin
  497. If NewCapacity<0 Then
  498. NewCapacity:=0
  499. else
  500. begin
  501. // if growing, grow at least a quarter
  502. if (NewCapacity>FCapacity) and (NewCapacity < (5*FCapacity) div 4) then
  503. NewCapacity := (5*FCapacity) div 4;
  504. // round off to block size.
  505. NewCapacity := (NewCapacity + (TMSGrow-1)) and not (TMSGROW-1);
  506. end;
  507. // Only now check !
  508. If NewCapacity=FCapacity then
  509. Result:=FMemory
  510. else
  511. begin
  512. Result:=Reallocmem(FMemory,Newcapacity);
  513. If (Result=Nil) and (Newcapacity>0) then
  514. Raise EStreamError.Create(SMemoryStreamError);
  515. end;
  516. end;
  517. destructor TMemoryStream.Destroy;
  518. begin
  519. Clear;
  520. Inherited Destroy;
  521. end;
  522. procedure TMemoryStream.Clear;
  523. begin
  524. FSize:=0;
  525. FPosition:=0;
  526. SetCapacity (0);
  527. end;
  528. procedure TMemoryStream.LoadFromStream(Stream: TStream);
  529. begin
  530. Stream.Position:=0;
  531. SetSize(Stream.Size);
  532. If FSize>0 then Stream.ReadBuffer(FMemory^,FSize);
  533. end;
  534. procedure TMemoryStream.LoadFromFile(const FileName: string);
  535. Var S : TFileStream;
  536. begin
  537. S:=TFileStream.Create (FileName,fmOpenRead or fmShareDenyWrite);
  538. Try
  539. LoadFromStream(S);
  540. finally
  541. S.free;
  542. end;
  543. end;
  544. procedure TMemoryStream.SetSize({$ifdef CPU64}const NewSize: Int64{$else}NewSize: LongInt{$endif});
  545. begin
  546. SetCapacity (NewSize);
  547. FSize:=NewSize;
  548. IF FPosition>FSize then
  549. FPosition:=FSize;
  550. end;
  551. function TMemoryStream.Write(const Buffer; Count: LongInt): LongInt;
  552. Var NewPos : PtrInt;
  553. begin
  554. If (Count=0) or (FPosition<0) then
  555. exit(0);
  556. NewPos:=FPosition+Count;
  557. If NewPos>Fsize then
  558. begin
  559. IF NewPos>FCapacity then
  560. SetCapacity (NewPos);
  561. FSize:=Newpos;
  562. end;
  563. System.Move (Buffer,(FMemory+FPosition)^,Count);
  564. FPosition:=NewPos;
  565. Result:=Count;
  566. end;
  567. {****************************************************************************}
  568. {* TBytesStream *}
  569. {****************************************************************************}
  570. constructor TBytesStream.Create(const ABytes: TBytes);
  571. begin
  572. inherited Create;
  573. FBytes:=ABytes;
  574. SetPointer(Pointer(FBytes),Length(FBytes));
  575. FCapacity:=Length(FBytes);
  576. end;
  577. function TBytesStream.Realloc(var NewCapacity: PtrInt): Pointer;
  578. begin
  579. // adapt TMemoryStream code to use with dynamic array
  580. if NewCapacity<0 Then
  581. NewCapacity:=0
  582. else
  583. begin
  584. if (NewCapacity>Capacity) and (NewCapacity < (5*Capacity) div 4) then
  585. NewCapacity := (5*Capacity) div 4;
  586. NewCapacity := (NewCapacity + (TMSGrow-1)) and not (TMSGROW-1);
  587. end;
  588. if NewCapacity=Capacity then
  589. Result:=Pointer(FBytes)
  590. else
  591. begin
  592. SetLength(FBytes,Newcapacity);
  593. Result:=Pointer(FBytes);
  594. if (Result=nil) and (Newcapacity>0) then
  595. raise EStreamError.Create(SMemoryStreamError);
  596. end;
  597. end;
  598. {****************************************************************************}
  599. {* TStringStream *}
  600. {****************************************************************************}
  601. function TStringStream.GetSize: Int64;
  602. begin
  603. Result:=Length(FDataString);
  604. end;
  605. function TStringStream.GetPosition: Int64;
  606. begin
  607. Result:=FPosition;
  608. end;
  609. procedure TStringStream.SetSize(NewSize: Longint);
  610. begin
  611. Setlength(FDataString,NewSize);
  612. If FPosition>NewSize then FPosition:=NewSize;
  613. end;
  614. constructor TStringStream.Create(const AString: string);
  615. begin
  616. Inherited create;
  617. FDataString:=AString;
  618. UniqueString(FDataString);
  619. end;
  620. function TStringStream.Read(var Buffer; Count: Longint): Longint;
  621. begin
  622. Result:=Length(FDataString)-FPosition;
  623. If Result>Count then Result:=Count;
  624. // This supposes FDataString to be of type AnsiString !
  625. Move (Pchar(FDataString)[FPosition],Buffer,Result);
  626. FPosition:=FPosition+Result;
  627. end;
  628. function TStringStream.ReadString(Count: Longint): string;
  629. Var NewLen : Longint;
  630. begin
  631. NewLen:=Length(FDataString)-FPosition;
  632. If NewLen>Count then NewLen:=Count;
  633. SetLength(Result,NewLen);
  634. Read (Pointer(Result)^,NewLen);
  635. end;
  636. function TStringStream.Seek(Offset: Longint; Origin: Word): Longint;
  637. begin
  638. Case Origin of
  639. soFromBeginning : FPosition:=Offset;
  640. soFromEnd : FPosition:=Length(FDataString)+Offset;
  641. soFromCurrent : FPosition:=FPosition+Offset;
  642. end;
  643. If FPosition>Length(FDataString) then
  644. FPosition:=Length(FDataString)
  645. else If FPosition<0 then
  646. FPosition:=0;
  647. Result:=FPosition;
  648. end;
  649. function TStringStream.Write(const Buffer; Count: Longint): Longint;
  650. begin
  651. Result:=Count;
  652. if FPosition+Count>Size then
  653. SetSize(FPosition+Count);
  654. // This supposes that FDataString is of type AnsiString)
  655. Move (Buffer,PChar(FDataString)[Fposition],Count);
  656. FPosition:=FPosition+Count;
  657. end;
  658. procedure TStringStream.WriteString(const AString: string);
  659. begin
  660. Write (PChar(Astring)[0],Length(AString));
  661. end;
  662. {****************************************************************************}
  663. {* TResourceStream *}
  664. {****************************************************************************}
  665. {$ifdef FPC_OS_UNICODE}
  666. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PWideChar; NameIsID: Boolean);
  667. begin
  668. Res:=FindResource(Instance, Name, ResType);
  669. if Res=0 then
  670. if NameIsID then
  671. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  672. else
  673. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  674. Handle:=LoadResource(Instance,Res);
  675. if Handle=0 then
  676. if NameIsID then
  677. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  678. else
  679. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  680. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  681. end;
  682. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: WideString; ResType: PWideChar);
  683. begin
  684. inherited create;
  685. Initialize(Instance,PWideChar(ResName),ResType,False);
  686. end;
  687. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PWideChar);
  688. begin
  689. inherited create;
  690. Initialize(Instance,PWideChar(ResID),ResType,True);
  691. end;
  692. {$else FPC_OS_UNICODE}
  693. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PChar; NameIsID: Boolean);
  694. begin
  695. Res:=FindResource(Instance, Name, ResType);
  696. if Res=0 then
  697. if NameIsID then
  698. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  699. else
  700. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  701. Handle:=LoadResource(Instance,Res);
  702. if Handle=0 then
  703. if NameIsID then
  704. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  705. else
  706. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  707. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  708. end;
  709. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: string; ResType: PChar);
  710. begin
  711. inherited create;
  712. Initialize(Instance,pchar(ResName),ResType,False);
  713. end;
  714. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PChar);
  715. begin
  716. inherited create;
  717. Initialize(Instance,pchar(PtrInt(ResID)),ResType,True);
  718. end;
  719. {$endif FPC_OS_UNICODE}
  720. destructor TResourceStream.Destroy;
  721. begin
  722. UnlockResource(Handle);
  723. FreeResource(Handle);
  724. inherited destroy;
  725. end;
  726. {****************************************************************************}
  727. {* TOwnerStream *}
  728. {****************************************************************************}
  729. constructor TOwnerStream.Create(ASource: TStream);
  730. begin
  731. FSource:=ASource;
  732. end;
  733. destructor TOwnerStream.Destroy;
  734. begin
  735. If FOwner then
  736. FreeAndNil(FSource);
  737. inherited Destroy;
  738. end;
  739. {****************************************************************************}
  740. {* TStreamAdapter *}
  741. {****************************************************************************}
  742. constructor TStreamAdapter.Create(Stream: TStream; Ownership: TStreamOwnership = soReference);
  743. begin
  744. inherited Create;
  745. FStream:=Stream;
  746. FOwnership:=Ownership;
  747. m_bReverted:=false; // mantis 15003
  748. // http://www.tech-archive.net/Archive/German/microsoft.public.de.vc/2005-08/msg00791.html
  749. // http://code.google.com/p/ddab-lib/wiki/TPJIStreamWrapper
  750. end;
  751. destructor TStreamAdapter.Destroy;
  752. begin
  753. if StreamOwnership=soOwned then
  754. FreeAndNil(FStream);
  755. inherited Destroy;
  756. end;
  757. {$warnings off}
  758. function TStreamAdapter.Read(pv: Pointer; cb: DWORD; pcbRead: PDWORD): HResult; stdcall;
  759. var
  760. readcount: Longint;
  761. begin
  762. if m_bReverted then
  763. begin
  764. Result := STG_E_REVERTED;
  765. Exit;
  766. end;
  767. if pv = nil then
  768. begin
  769. Result := STG_E_INVALIDPOINTER;
  770. Exit;
  771. end;
  772. readcount := FStream.Read(pv^, cb);
  773. if pcbRead <> nil then pcbRead^ := readcount;
  774. Result := S_OK;
  775. end;
  776. function TStreamAdapter.Write(pv: Pointer; cb: DWORD; pcbWritten: PDWORD): HResult; stdcall;
  777. var
  778. writecount: Longint;
  779. begin
  780. if m_bReverted then
  781. begin
  782. Result := STG_E_REVERTED;
  783. Exit;
  784. end;
  785. if pv = nil then
  786. begin
  787. Result := STG_E_INVALIDPOINTER;
  788. Exit;
  789. end;
  790. writecount := FStream.Write(pv^, cb);
  791. if pcbWritten <> nil then pcbWritten^ := writecount;
  792. Result := S_OK;
  793. end;
  794. function TStreamAdapter.Seek(dlibMove: Largeint; dwOrigin: Longint; out libNewPosition: Largeint): HResult; stdcall;
  795. var
  796. newpos: Int64;
  797. begin
  798. if m_bReverted then
  799. begin
  800. Result := STG_E_REVERTED;
  801. Exit;
  802. end;
  803. case dwOrigin of
  804. STREAM_SEEK_SET: newpos := FStream.Seek(dlibMove, soBeginning);
  805. STREAM_SEEK_CUR: newpos := FStream.Seek(dlibMove, soCurrent);
  806. STREAM_SEEK_END: newpos := FStream.Seek(dlibMove, soEnd);
  807. else
  808. begin
  809. Result := STG_E_INVALIDFUNCTION;
  810. Exit;
  811. end;
  812. end;
  813. if @libNewPosition <> nil then
  814. libNewPosition := newpos;
  815. Result := S_OK;
  816. end;
  817. function TStreamAdapter.SetSize(libNewSize: Largeint): HResult; stdcall;
  818. begin
  819. if m_bReverted then
  820. begin
  821. Result := STG_E_REVERTED;
  822. Exit;
  823. end;
  824. if libNewSize<0 then
  825. begin
  826. Result := STG_E_INVALIDFUNCTION;
  827. Exit;
  828. end;
  829. try
  830. FStream.Size := libNewSize;
  831. Result := S_OK;
  832. except
  833. // TODO: return different error value according to exception like STG_E_MEDIUMFULL
  834. Result := E_FAIL;
  835. end;
  836. end;
  837. function TStreamAdapter.CopyTo(stm: IStream; cb: Largeint; out cbRead: Largeint; out cbWritten: Largeint): HResult; stdcall;
  838. var
  839. sz: dword;
  840. buffer : array[0..1023] of byte;
  841. begin
  842. if m_bReverted then
  843. begin
  844. Result := STG_E_REVERTED;
  845. Exit;
  846. end;
  847. // the method is similar to TStream.CopyFrom => use CopyFrom implementation
  848. cbWritten := 0;
  849. cbRead := 0;
  850. while cb > 0 do
  851. begin
  852. if (cb > sizeof(buffer)) then
  853. sz := sizeof(Buffer)
  854. else
  855. sz := cb;
  856. sz := FStream.Read(buffer, sz);
  857. inc(cbRead, sz);
  858. stm.Write(@buffer[0], sz, @sz);
  859. inc(cbWritten, sz);
  860. if sz = 0 then
  861. begin
  862. Result := E_FAIL;
  863. Exit;
  864. end;
  865. dec(cb, sz);
  866. end;
  867. Result := S_OK;
  868. end;
  869. function TStreamAdapter.Commit(grfCommitFlags: Longint): HResult; stdcall;
  870. begin
  871. if m_bReverted then
  872. Result := STG_E_REVERTED
  873. else
  874. Result := S_OK;
  875. end;
  876. function TStreamAdapter.Revert: HResult; stdcall;
  877. begin
  878. m_bReverted := True;
  879. Result := S_OK;
  880. end;
  881. function TStreamAdapter.LockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  882. begin
  883. Result := STG_E_INVALIDFUNCTION;
  884. end;
  885. function TStreamAdapter.UnlockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  886. begin
  887. Result := STG_E_INVALIDFUNCTION;
  888. end;
  889. function TStreamAdapter.Stat(out statstg: TStatStg; grfStatFlag: Longint): HResult; stdcall;
  890. begin
  891. if m_bReverted then
  892. begin
  893. Result := STG_E_REVERTED;
  894. Exit;
  895. end;
  896. if grfStatFlag in [STATFLAG_DEFAULT,STATFLAG_NOOPEN,STATFLAG_NONAME] then
  897. begin
  898. if @statstg <> nil then
  899. begin
  900. fillchar(statstg, sizeof(TStatStg),#0);
  901. { //TODO handle pwcsName
  902. if grfStatFlag = STATFLAG_DEFAULT then
  903. runerror(217) //Result :={$ifdef windows} STG_E_INVALIDFLAG{$else}E_INVALID_FLAG{$endif}
  904. }
  905. statstg.dwType := STGTY_STREAM;
  906. statstg.cbSize := FStream.Size;
  907. statstg.grfLocksSupported := LOCK_WRITE;
  908. end;
  909. Result := S_OK;
  910. end else
  911. Result := STG_E_INVALIDFLAG
  912. end;
  913. function TStreamAdapter.Clone(out stm: IStream): HResult; stdcall;
  914. begin
  915. if m_bReverted then
  916. begin
  917. Result := STG_E_REVERTED;
  918. Exit;
  919. end;
  920. // don't raise an exception here return error value that function is not implemented
  921. // to implement this we need a clone method for TStream class
  922. Result := STG_E_UNIMPLEMENTEDFUNCTION;
  923. end;
  924. constructor TProxyStream.Create(const Stream: IStream);
  925. begin
  926. FStream := Stream;
  927. end;
  928. function TProxyStream.Read(var Buffer; Count: Longint): Longint;
  929. begin
  930. Check(FStream.Read(@Buffer, Count, @Result));
  931. end;
  932. function TProxyStream. Seek(const Offset: int64; Origin: TSeekOrigin): int64;
  933. begin
  934. Check(FStream.Seek(Offset, ord(Origin), result));
  935. end;
  936. function TProxyStream.Write(const Buffer; Count: Longint): Longint;
  937. begin
  938. Check(FStream.Write(@Buffer, Count, @Result));
  939. end;
  940. function TProxyStream.GetIStream: IStream;
  941. begin
  942. Result := FStream;
  943. end;
  944. procedure TProxyStream.Check(err:integer);
  945. var e : EInOutError;
  946. begin
  947. e:= EInOutError.Create('Proxystream.Check');
  948. e.Errorcode:=err;
  949. raise e;
  950. end;
  951. {$warnings on}