streams.inc 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200
  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. end;
  619. function TStringStream.Read(var Buffer; Count: Longint): Longint;
  620. begin
  621. Result:=Length(FDataString)-FPosition;
  622. If Result>Count then Result:=Count;
  623. // This supposes FDataString to be of type AnsiString !
  624. Move (Pchar(FDataString)[FPosition],Buffer,Result);
  625. FPosition:=FPosition+Result;
  626. end;
  627. function TStringStream.ReadString(Count: Longint): string;
  628. Var NewLen : Longint;
  629. begin
  630. NewLen:=Length(FDataString)-FPosition;
  631. If NewLen>Count then NewLen:=Count;
  632. SetLength(Result,NewLen);
  633. Read (Pointer(Result)^,NewLen);
  634. end;
  635. function TStringStream.Seek(Offset: Longint; Origin: Word): Longint;
  636. begin
  637. Case Origin of
  638. soFromBeginning : FPosition:=Offset;
  639. soFromEnd : FPosition:=Length(FDataString)+Offset;
  640. soFromCurrent : FPosition:=FPosition+Offset;
  641. end;
  642. If FPosition>Length(FDataString) then
  643. FPosition:=Length(FDataString)
  644. else If FPosition<0 then
  645. FPosition:=0;
  646. Result:=FPosition;
  647. end;
  648. function TStringStream.Write(const Buffer; Count: Longint): Longint;
  649. begin
  650. Result:=Count;
  651. if FPosition+Count>Size then
  652. SetSize(FPosition+Count);
  653. // This supposes that FDataString is of type AnsiString)
  654. Move (Buffer,PChar(FDataString)[Fposition],Count);
  655. FPosition:=FPosition+Count;
  656. end;
  657. procedure TStringStream.WriteString(const AString: string);
  658. begin
  659. Write (PChar(Astring)[0],Length(AString));
  660. end;
  661. {****************************************************************************}
  662. {* TResourceStream *}
  663. {****************************************************************************}
  664. {$ifdef FPC_OS_UNICODE}
  665. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PWideChar; NameIsID: Boolean);
  666. begin
  667. Res:=FindResource(Instance, Name, ResType);
  668. if Res=0 then
  669. if NameIsID then
  670. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  671. else
  672. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  673. Handle:=LoadResource(Instance,Res);
  674. if Handle=0 then
  675. if NameIsID then
  676. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  677. else
  678. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  679. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  680. end;
  681. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: WideString; ResType: PWideChar);
  682. begin
  683. inherited create;
  684. Initialize(Instance,PWideChar(ResName),ResType,False);
  685. end;
  686. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PWideChar);
  687. begin
  688. inherited create;
  689. Initialize(Instance,PWideChar(ResID),ResType,True);
  690. end;
  691. {$else FPC_OS_UNICODE}
  692. procedure TResourceStream.Initialize(Instance: TFPResourceHMODULE; Name, ResType: PChar; NameIsID: Boolean);
  693. begin
  694. Res:=FindResource(Instance, Name, ResType);
  695. if Res=0 then
  696. if NameIsID then
  697. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  698. else
  699. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  700. Handle:=LoadResource(Instance,Res);
  701. if Handle=0 then
  702. if NameIsID then
  703. raise EResNotFound.CreateFmt(SResNotFound,[IntToStr(PtrInt(Name))])
  704. else
  705. raise EResNotFound.CreateFmt(SResNotFound,[Name]);
  706. SetPointer(LockResource(Handle),SizeOfResource(Instance,Res));
  707. end;
  708. constructor TResourceStream.Create(Instance: TFPResourceHMODULE; const ResName: string; ResType: PChar);
  709. begin
  710. inherited create;
  711. Initialize(Instance,pchar(ResName),ResType,False);
  712. end;
  713. constructor TResourceStream.CreateFromID(Instance: TFPResourceHMODULE; ResID: Integer; ResType: PChar);
  714. begin
  715. inherited create;
  716. Initialize(Instance,pchar(PtrInt(ResID)),ResType,True);
  717. end;
  718. {$endif FPC_OS_UNICODE}
  719. destructor TResourceStream.Destroy;
  720. begin
  721. UnlockResource(Handle);
  722. FreeResource(Handle);
  723. inherited destroy;
  724. end;
  725. {****************************************************************************}
  726. {* TOwnerStream *}
  727. {****************************************************************************}
  728. constructor TOwnerStream.Create(ASource: TStream);
  729. begin
  730. FSource:=ASource;
  731. end;
  732. destructor TOwnerStream.Destroy;
  733. begin
  734. If FOwner then
  735. FreeAndNil(FSource);
  736. inherited Destroy;
  737. end;
  738. {****************************************************************************}
  739. {* TStreamAdapter *}
  740. {****************************************************************************}
  741. constructor TStreamAdapter.Create(Stream: TStream; Ownership: TStreamOwnership = soReference);
  742. begin
  743. inherited Create;
  744. FStream:=Stream;
  745. FOwnership:=Ownership;
  746. m_bReverted:=false; // mantis 15003
  747. // http://www.tech-archive.net/Archive/German/microsoft.public.de.vc/2005-08/msg00791.html
  748. // http://code.google.com/p/ddab-lib/wiki/TPJIStreamWrapper
  749. end;
  750. destructor TStreamAdapter.Destroy;
  751. begin
  752. if StreamOwnership=soOwned then
  753. FreeAndNil(FStream);
  754. inherited Destroy;
  755. end;
  756. {$warnings off}
  757. function TStreamAdapter.Read(pv: Pointer; cb: DWORD; pcbRead: PDWORD): HResult; stdcall;
  758. var
  759. readcount: Longint;
  760. begin
  761. if m_bReverted then
  762. begin
  763. Result := STG_E_REVERTED;
  764. Exit;
  765. end;
  766. if pv = nil then
  767. begin
  768. Result := STG_E_INVALIDPOINTER;
  769. Exit;
  770. end;
  771. readcount := FStream.Read(pv^, cb);
  772. if pcbRead <> nil then pcbRead^ := readcount;
  773. Result := S_OK;
  774. end;
  775. function TStreamAdapter.Write(pv: Pointer; cb: DWORD; pcbWritten: PDWORD): HResult; stdcall;
  776. var
  777. writecount: Longint;
  778. begin
  779. if m_bReverted then
  780. begin
  781. Result := STG_E_REVERTED;
  782. Exit;
  783. end;
  784. if pv = nil then
  785. begin
  786. Result := STG_E_INVALIDPOINTER;
  787. Exit;
  788. end;
  789. writecount := FStream.Write(pv^, cb);
  790. if pcbWritten <> nil then pcbWritten^ := writecount;
  791. Result := S_OK;
  792. end;
  793. function TStreamAdapter.Seek(dlibMove: Largeint; dwOrigin: Longint; out libNewPosition: Largeint): HResult; stdcall;
  794. var
  795. newpos: Int64;
  796. begin
  797. if m_bReverted then
  798. begin
  799. Result := STG_E_REVERTED;
  800. Exit;
  801. end;
  802. case dwOrigin of
  803. STREAM_SEEK_SET: newpos := FStream.Seek(dlibMove, soBeginning);
  804. STREAM_SEEK_CUR: newpos := FStream.Seek(dlibMove, soCurrent);
  805. STREAM_SEEK_END: newpos := FStream.Seek(dlibMove, soEnd);
  806. else
  807. begin
  808. Result := STG_E_INVALIDFUNCTION;
  809. Exit;
  810. end;
  811. end;
  812. if @libNewPosition <> nil then
  813. libNewPosition := newpos;
  814. Result := S_OK;
  815. end;
  816. function TStreamAdapter.SetSize(libNewSize: Largeint): HResult; stdcall;
  817. begin
  818. if m_bReverted then
  819. begin
  820. Result := STG_E_REVERTED;
  821. Exit;
  822. end;
  823. if libNewSize<0 then
  824. begin
  825. Result := STG_E_INVALIDFUNCTION;
  826. Exit;
  827. end;
  828. try
  829. FStream.Size := libNewSize;
  830. Result := S_OK;
  831. except
  832. // TODO: return different error value according to exception like STG_E_MEDIUMFULL
  833. Result := E_FAIL;
  834. end;
  835. end;
  836. function TStreamAdapter.CopyTo(stm: IStream; cb: Largeint; out cbRead: Largeint; out cbWritten: Largeint): HResult; stdcall;
  837. var
  838. sz: dword;
  839. buffer : array[0..1023] of byte;
  840. begin
  841. if m_bReverted then
  842. begin
  843. Result := STG_E_REVERTED;
  844. Exit;
  845. end;
  846. // the method is similar to TStream.CopyFrom => use CopyFrom implementation
  847. cbWritten := 0;
  848. cbRead := 0;
  849. while cb > 0 do
  850. begin
  851. if (cb > sizeof(buffer)) then
  852. sz := sizeof(Buffer)
  853. else
  854. sz := cb;
  855. sz := FStream.Read(buffer, sz);
  856. inc(cbRead, sz);
  857. stm.Write(@buffer[0], sz, @sz);
  858. inc(cbWritten, sz);
  859. if sz = 0 then
  860. begin
  861. Result := E_FAIL;
  862. Exit;
  863. end;
  864. dec(cb, sz);
  865. end;
  866. Result := S_OK;
  867. end;
  868. function TStreamAdapter.Commit(grfCommitFlags: Longint): HResult; stdcall;
  869. begin
  870. if m_bReverted then
  871. Result := STG_E_REVERTED
  872. else
  873. Result := S_OK;
  874. end;
  875. function TStreamAdapter.Revert: HResult; stdcall;
  876. begin
  877. m_bReverted := True;
  878. Result := S_OK;
  879. end;
  880. function TStreamAdapter.LockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  881. begin
  882. Result := STG_E_INVALIDFUNCTION;
  883. end;
  884. function TStreamAdapter.UnlockRegion(libOffset: Largeint; cb: Largeint; dwLockType: Longint): HResult; stdcall;
  885. begin
  886. Result := STG_E_INVALIDFUNCTION;
  887. end;
  888. function TStreamAdapter.Stat(out statstg: TStatStg; grfStatFlag: Longint): HResult; stdcall;
  889. begin
  890. if m_bReverted then
  891. begin
  892. Result := STG_E_REVERTED;
  893. Exit;
  894. end;
  895. if grfStatFlag in [STATFLAG_DEFAULT,STATFLAG_NOOPEN,STATFLAG_NONAME] then
  896. begin
  897. if @statstg <> nil then
  898. begin
  899. fillchar(statstg, sizeof(TStatStg),#0);
  900. { //TODO handle pwcsName
  901. if grfStatFlag = STATFLAG_DEFAULT then
  902. runerror(217) //Result :={$ifdef windows} STG_E_INVALIDFLAG{$else}E_INVALID_FLAG{$endif}
  903. }
  904. statstg.dwType := STGTY_STREAM;
  905. statstg.cbSize := FStream.Size;
  906. statstg.grfLocksSupported := LOCK_WRITE;
  907. end;
  908. Result := S_OK;
  909. end else
  910. Result := STG_E_INVALIDFLAG
  911. end;
  912. function TStreamAdapter.Clone(out stm: IStream): HResult; stdcall;
  913. begin
  914. if m_bReverted then
  915. begin
  916. Result := STG_E_REVERTED;
  917. Exit;
  918. end;
  919. // don't raise an exception here return error value that function is not implemented
  920. // to implement this we need a clone method for TStream class
  921. Result := STG_E_UNIMPLEMENTEDFUNCTION;
  922. end;
  923. constructor TProxyStream.Create(const Stream: IStream);
  924. begin
  925. FStream := Stream;
  926. end;
  927. function TProxyStream.Read(var Buffer; Count: Longint): Longint;
  928. begin
  929. Check(FStream.Read(@Buffer, Count, @Result));
  930. end;
  931. function TProxyStream. Seek(const Offset: int64; Origin: TSeekOrigin): int64;
  932. begin
  933. Check(FStream.Seek(Offset, ord(Origin), result));
  934. end;
  935. function TProxyStream.Write(const Buffer; Count: Longint): Longint;
  936. begin
  937. Check(FStream.Write(@Buffer, Count, @Result));
  938. end;
  939. function TProxyStream.GetIStream: IStream;
  940. begin
  941. Result := FStream;
  942. end;
  943. procedure TProxyStream.Check(err:integer);
  944. var e : EInOutError;
  945. begin
  946. e:= EInOutError.Create('Proxystream.Check');
  947. e.Errorcode:=err;
  948. raise e;
  949. end;
  950. {$warnings on}