streams.inc 28 KB

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