streams.inc 30 KB

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