streams.inc 31 KB

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