streams.inc 29 KB

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