streams.inc 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315
  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: UnicodeString;
  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. {* TRawByteStringStream *}
  722. {****************************************************************************}
  723. constructor TRawByteStringStream.Create(const aData: RawByteString);
  724. begin
  725. Inherited Create;
  726. If Length(aData)>0 then
  727. begin
  728. WriteBuffer(aData[1],Length(aData));
  729. Position:=0;
  730. end;
  731. end;
  732. function TRawByteStringStream.DataString: RawByteString;
  733. begin
  734. Result:='';
  735. SetLength(Result,Size);
  736. if Size>0 then
  737. Move(Memory^, Result[1], Size);
  738. end;
  739. function TRawByteStringStream.ReadString(Count: Longint): RawByteString;
  740. Var
  741. NewLen : Longint;
  742. begin
  743. NewLen:=Size-FPosition;
  744. If NewLen>Count then NewLen:=Count;
  745. Result:='';
  746. if NewLen>0 then
  747. begin
  748. SetLength(Result, NewLen);
  749. Move(FBytes[FPosition],Result[1],NewLen);
  750. end;
  751. end;
  752. procedure TRawByteStringStream.WriteString(const AString: RawByteString);
  753. begin
  754. if Length(AString)>0 then
  755. WriteBuffer(AString[1],Length(AString));
  756. end;
  757. {****************************************************************************}
  758. {* TResourceStream *}
  759. {****************************************************************************}
  760. {$ifdef FPC_OS_UNICODE}
  761. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PWideChar; NameIsID: Boolean);
  762. begin
  763. Res:=FindResource(Instance, Name, ResType);
  764. if Res=0 then
  765. if NameIsID then
  766. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  767. else
  768. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  769. Handle:=LoadResource(Instance,Res);
  770. if Handle=0 then
  771. if NameIsID then
  772. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  773. else
  774. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  775. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  776. end;
  777. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: WideString; ResType: PWideChar);
  778. begin
  779. inherited create;
  780. Initialize(Instance,PWideChar(ResName),ResType,False);
  781. end;
  782. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PWideChar);
  783. begin
  784. inherited create;
  785. Initialize(Instance,PWideChar(ResID),ResType,True);
  786. end;
  787. {$else FPC_OS_UNICODE}
  788. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PChar; NameIsID: Boolean);
  789. begin
  790. Res:=FindResource(Instance, Name, ResType);
  791. if Res=0 then
  792. if NameIsID then
  793. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  794. else
  795. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  796. Handle:=LoadResource(Instance,Res);
  797. if Handle=0 then
  798. if NameIsID then
  799. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  800. else
  801. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  802. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  803. end;
  804. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: string; ResType: PChar);
  805. begin
  806. inherited create;
  807. Initialize(Instance,pchar(ResName),ResType,False);
  808. end;
  809. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PChar);
  810. begin
  811. inherited create;
  812. Initialize(Instance,pchar(PtrInt(ResID)),ResType,True);
  813. end;
  814. {$endif FPC_OS_UNICODE}
  815. destructor TResourceStream.Destroy;
  816. begin
  817. UnlockResource(Handle);
  818. FreeResource(Handle);
  819. inherited destroy;
  820. end;
  821. {****************************************************************************}
  822. {* TOwnerStream *}
  823. {****************************************************************************}
  824. constructor TOwnerStream.Create(ASource: TStream);
  825. begin
  826. FSource:=ASource;
  827. end;
  828. destructor TOwnerStream.Destroy;
  829. begin
  830. If FOwner then
  831. FreeAndNil(FSource);
  832. inherited Destroy;
  833. end;
  834. {****************************************************************************}
  835. {* TStreamAdapter *}
  836. {****************************************************************************}
  837. constructor TStreamAdapter.Create(Stream: TStream; Ownership: TStreamOwnership = soReference);
  838. begin
  839. inherited Create;
  840. FStream:=Stream;
  841. FOwnership:=Ownership;
  842. m_bReverted:=false; // mantis 15003
  843. // http://www.tech-archive.net/Archive/German/microsoft.public.de.vc/2005-08/msg00791.html
  844. // http://code.google.com/p/ddab-lib/wiki/TPJIStreamWrapper
  845. end;
  846. destructor TStreamAdapter.Destroy;
  847. begin
  848. if StreamOwnership=soOwned then
  849. FreeAndNil(FStream);
  850. inherited Destroy;
  851. end;
  852. {$warnings off}
  853. function TStreamAdapter.Read(pv: Pointer; cb: DWORD; pcbRead: PDWORD): HResult; stdcall;
  854. var
  855. readcount: Longint;
  856. begin
  857. if m_bReverted then
  858. begin
  859. Result := STG_E_REVERTED;
  860. Exit;
  861. end;
  862. if pv = nil then
  863. begin
  864. Result := STG_E_INVALIDPOINTER;
  865. Exit;
  866. end;
  867. readcount := FStream.Read(pv^, cb);
  868. if pcbRead <> nil then pcbRead^ := readcount;
  869. Result := S_OK;
  870. end;
  871. function TStreamAdapter.Write(pv: Pointer; cb: DWORD; pcbWritten: PDWORD): HResult; stdcall;
  872. var
  873. writecount: Longint;
  874. begin
  875. if m_bReverted then
  876. begin
  877. Result := STG_E_REVERTED;
  878. Exit;
  879. end;
  880. if pv = nil then
  881. begin
  882. Result := STG_E_INVALIDPOINTER;
  883. Exit;
  884. end;
  885. writecount := FStream.Write(pv^, cb);
  886. if pcbWritten <> nil then pcbWritten^ := writecount;
  887. Result := S_OK;
  888. end;
  889. function TStreamAdapter.Seek(dlibMove: LargeInt; dwOrigin: DWORD; out libNewPosition: LargeUint): HResult; stdcall;
  890. var
  891. newpos: QWord;
  892. begin
  893. if m_bReverted then
  894. begin
  895. Result := STG_E_REVERTED;
  896. Exit;
  897. end;
  898. case dwOrigin of
  899. STREAM_SEEK_SET: newpos := FStream.Seek(dlibMove, soBeginning);
  900. STREAM_SEEK_CUR: newpos := FStream.Seek(dlibMove, soCurrent);
  901. STREAM_SEEK_END: newpos := FStream.Seek(dlibMove, soEnd);
  902. else
  903. begin
  904. Result := STG_E_INVALIDFUNCTION;
  905. Exit;
  906. end;
  907. end;
  908. if @libNewPosition <> nil then
  909. libNewPosition := newpos;
  910. Result := S_OK;
  911. end;
  912. function TStreamAdapter.SetSize(libNewSize: LargeUint): HResult; stdcall;
  913. begin
  914. if m_bReverted then
  915. begin
  916. Result := STG_E_REVERTED;
  917. Exit;
  918. end;
  919. if libNewSize<0 then
  920. begin
  921. Result := STG_E_INVALIDFUNCTION;
  922. Exit;
  923. end;
  924. try
  925. FStream.Size := libNewSize;
  926. Result := S_OK;
  927. except
  928. // TODO: return different error value according to exception like STG_E_MEDIUMFULL
  929. Result := E_FAIL;
  930. end;
  931. end;
  932. function TStreamAdapter.CopyTo(stm: IStream; cb: LargeUint; out cbRead: LargeUint; out cbWritten: Largeuint): HResult; stdcall;
  933. var
  934. sz: dword;
  935. buffer : array[0..1023] of byte;
  936. begin
  937. if m_bReverted then
  938. begin
  939. Result := STG_E_REVERTED;
  940. Exit;
  941. end;
  942. // the method is similar to TStream.CopyFrom => use CopyFrom implementation
  943. cbWritten := 0;
  944. cbRead := 0;
  945. while cb > 0 do
  946. begin
  947. if (cb > sizeof(buffer)) then
  948. sz := sizeof(Buffer)
  949. else
  950. sz := cb;
  951. sz := FStream.Read(buffer, sz);
  952. inc(cbRead, sz);
  953. stm.Write(@buffer[0], sz, @sz);
  954. inc(cbWritten, sz);
  955. if sz = 0 then
  956. begin
  957. Result := E_FAIL;
  958. Exit;
  959. end;
  960. dec(cb, sz);
  961. end;
  962. Result := S_OK;
  963. end;
  964. function TStreamAdapter.Commit(grfCommitFlags: DWORD): HResult; stdcall;
  965. begin
  966. if m_bReverted then
  967. Result := STG_E_REVERTED
  968. else
  969. Result := S_OK;
  970. end;
  971. function TStreamAdapter.Revert: HResult; stdcall;
  972. begin
  973. m_bReverted := True;
  974. Result := S_OK;
  975. end;
  976. function TStreamAdapter.LockRegion(libOffset: LargeUint; cb: LargeUint; dwLockType: DWORD): HResult; stdcall;
  977. begin
  978. Result := STG_E_INVALIDFUNCTION;
  979. end;
  980. function TStreamAdapter.UnlockRegion(libOffset: LargeUint; cb: LargeUint; dwLockType: DWORD): HResult; stdcall;
  981. begin
  982. Result := STG_E_INVALIDFUNCTION;
  983. end;
  984. function TStreamAdapter.Stat(out statstg: TStatStg; grfStatFlag: DWORD): HResult; stdcall;
  985. begin
  986. if m_bReverted then
  987. begin
  988. Result := STG_E_REVERTED;
  989. Exit;
  990. end;
  991. if grfStatFlag in [STATFLAG_DEFAULT,STATFLAG_NOOPEN,STATFLAG_NONAME] then
  992. begin
  993. if @statstg <> nil then
  994. begin
  995. fillchar(statstg, sizeof(TStatStg),#0);
  996. { //TODO handle pwcsName
  997. if grfStatFlag = STATFLAG_DEFAULT then
  998. runerror(217) //Result :={$ifdef windows} STG_E_INVALIDFLAG{$else}E_INVALID_FLAG{$endif}
  999. }
  1000. statstg.dwType := STGTY_STREAM;
  1001. statstg.cbSize := FStream.Size;
  1002. statstg.grfLocksSupported := LOCK_WRITE;
  1003. end;
  1004. Result := S_OK;
  1005. end else
  1006. Result := STG_E_INVALIDFLAG
  1007. end;
  1008. function TStreamAdapter.Clone(out stm: IStream): HResult; stdcall;
  1009. begin
  1010. if m_bReverted then
  1011. begin
  1012. Result := STG_E_REVERTED;
  1013. Exit;
  1014. end;
  1015. // don't raise an exception here return error value that function is not implemented
  1016. // to implement this we need a clone method for TStream class
  1017. Result := STG_E_UNIMPLEMENTEDFUNCTION;
  1018. end;
  1019. constructor TProxyStream.Create(const Stream: IStream);
  1020. begin
  1021. FStream := Stream;
  1022. end;
  1023. function TProxyStream.Read(var Buffer; Count: Longint): Longint;
  1024. begin
  1025. Check(FStream.Read(@Buffer, Count, @Result));
  1026. end;
  1027. function TProxyStream.Seek(const Offset: int64; Origin: TSeekOrigin): int64;
  1028. begin
  1029. Check(FStream.Seek(Offset, ord(Origin), QWord(result)));
  1030. end;
  1031. function TProxyStream.Write(const Buffer; Count: Longint): Longint;
  1032. begin
  1033. Check(FStream.Write(@Buffer, Count, @Result));
  1034. end;
  1035. function TProxyStream.GetIStream: IStream;
  1036. begin
  1037. Result := FStream;
  1038. end;
  1039. procedure TProxyStream.Check(err:integer);
  1040. var e : EInOutError;
  1041. begin
  1042. e:= EInOutError.Create('Proxystream.Check');
  1043. e.Errorcode:=err;
  1044. raise e;
  1045. end;
  1046. {$warnings on}