Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 1 | (* |
| 2 | * Licensed to the Apache Software Foundation (ASF) under one |
| 3 | * or more contributor license agreements. See the NOTICE file |
| 4 | * distributed with this work for additional information |
| 5 | * regarding copyright ownership. The ASF licenses this file |
| 6 | * to you under the Apache License, Version 2.0 (the |
| 7 | * "License"); you may not use this file except in compliance |
| 8 | * with the License. You may obtain a copy of the License at |
| 9 | * |
| 10 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 11 | * |
| 12 | * Unless required by applicable law or agreed to in writing, |
| 13 | * software distributed under the License is distributed on an |
| 14 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| 15 | * KIND, either express or implied. See the License for the |
| 16 | * specific language governing permissions and limitations |
| 17 | * under the License. |
| 18 | *) |
| 19 | unit Thrift.Transport.Pipes; |
| 20 | |
| 21 | {$WARN SYMBOL_PLATFORM OFF} |
| 22 | |
| 23 | interface |
| 24 | |
| 25 | uses |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 26 | Windows, SysUtils, Math, AccCtrl, AclAPI, SyncObjs, |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 27 | Thrift.Transport, |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 28 | Thrift.Stream; |
| 29 | |
| 30 | const |
| 31 | DEFAULT_THRIFT_PIPE_TIMEOUT = 5 * 1000; // ms |
| 32 | |
| 33 | |
| 34 | type |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 35 | //--- Pipe Streams --- |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 36 | |
| 37 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 38 | TPipeStreamBase = class( TThriftStreamImpl) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 39 | strict protected |
| 40 | FPipe : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 41 | FTimeout : DWORD; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 42 | |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 43 | procedure Write( const buffer: TBytes; offset: Integer; count: Integer); override; |
| 44 | function Read( var buffer: TBytes; offset: Integer; count: Integer): Integer; override; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 45 | //procedure Open; override; - see derived classes |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 46 | procedure Close; override; |
| 47 | procedure Flush; override; |
| 48 | |
| 49 | function IsOpen: Boolean; override; |
| 50 | function ToArray: TBytes; override; |
| 51 | public |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 52 | constructor Create( const aTimeOut : DWORD = DEFAULT_THRIFT_PIPE_TIMEOUT); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 53 | destructor Destroy; override; |
| 54 | end; |
| 55 | |
| 56 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 57 | TNamedPipeStreamImpl = class sealed( TPipeStreamBase) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 58 | private |
| 59 | FPipeName : string; |
| 60 | FShareMode : DWORD; |
| 61 | FSecurityAttribs : PSecurityAttributes; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 62 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 63 | protected |
| 64 | procedure Open; override; |
| 65 | |
| 66 | public |
| 67 | constructor Create( const aPipeName : string; |
| 68 | const aShareMode: DWORD = 0; |
| 69 | const aSecurityAttributes: PSecurityAttributes = nil; |
| 70 | const aTimeOut : DWORD = DEFAULT_THRIFT_PIPE_TIMEOUT); overload; |
| 71 | end; |
| 72 | |
| 73 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 74 | THandlePipeStreamImpl = class sealed( TPipeStreamBase) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 75 | private |
| 76 | FSrcHandle : THandle; |
| 77 | |
| 78 | protected |
| 79 | procedure Open; override; |
| 80 | |
| 81 | public |
| 82 | constructor Create( const aPipeHandle : THandle; aOwnsHandle : Boolean); overload; |
| 83 | destructor Destroy; override; |
| 84 | end; |
| 85 | |
| 86 | |
| 87 | //--- Pipe Transports --- |
| 88 | |
| 89 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 90 | IPipeTransport = interface( IStreamTransport) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 91 | ['{5E05CC85-434F-428F-BFB2-856A168B5558}'] |
| 92 | end; |
| 93 | |
| 94 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 95 | TPipeTransportBase = class( TStreamTransportImpl, IPipeTransport) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 96 | public |
| 97 | // ITransport |
| 98 | function GetIsOpen: Boolean; override; |
| 99 | procedure Open; override; |
| 100 | procedure Close; override; |
| 101 | end; |
| 102 | |
| 103 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 104 | TNamedPipeTransportClientEndImpl = class( TPipeTransportBase) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 105 | public |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 106 | // Named pipe constructors |
| 107 | constructor Create( aPipe : THandle; aOwnsHandle : Boolean); overload; |
| 108 | constructor Create( const aPipeName : string; |
| 109 | const aShareMode: DWORD = 0; |
| 110 | const aSecurityAttributes: PSecurityAttributes = nil; |
| 111 | const aTimeOut : DWORD = DEFAULT_THRIFT_PIPE_TIMEOUT); overload; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 112 | end; |
| 113 | |
| 114 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 115 | TNamedPipeTransportServerEndImpl = class( TNamedPipeTransportClientEndImpl) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 116 | strict private |
| 117 | FHandle : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 118 | public |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 119 | // ITransport |
| 120 | procedure Close; override; |
| 121 | constructor Create( aPipe : THandle; aOwnsHandle : Boolean); reintroduce; |
| 122 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 123 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 124 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 125 | TAnonymousPipeTransportImpl = class( TPipeTransportBase) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 126 | public |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 127 | // Anonymous pipe constructor |
| 128 | constructor Create( const aPipeRead, aPipeWrite : THandle; aOwnsHandles : Boolean); overload; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 129 | end; |
| 130 | |
| 131 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 132 | //--- Server Transports --- |
| 133 | |
| 134 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 135 | IAnonymousPipeServerTransport = interface( IServerTransport) |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 136 | ['{7AEE6793-47B9-4E49-981A-C39E9108E9AD}'] |
| 137 | // Server side anonymous pipe ends |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 138 | function ReadHandle : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 139 | function WriteHandle : THandle; |
| 140 | // Client side anonymous pipe ends |
| 141 | function ClientAnonRead : THandle; |
| 142 | function ClientAnonWrite : THandle; |
| 143 | end; |
| 144 | |
| 145 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 146 | INamedPipeServerTransport = interface( IServerTransport) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 147 | ['{9DF9EE48-D065-40AF-8F67-D33037D3D960}'] |
| 148 | function Handle : THandle; |
| 149 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 150 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 151 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 152 | TPipeServerTransportBase = class( TServerTransportImpl) |
| 153 | protected |
| 154 | FStopServer : Boolean; |
| 155 | procedure InternalClose; virtual; abstract; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 156 | public |
| 157 | procedure Listen; override; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 158 | procedure Close; override; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 159 | end; |
| 160 | |
| 161 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 162 | TAnonymousPipeServerTransportImpl = class( TPipeServerTransportBase, IAnonymousPipeServerTransport) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 163 | private |
| 164 | FBufSize : DWORD; |
| 165 | |
| 166 | // Server side anonymous pipe handles |
| 167 | FReadHandle, |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 168 | FWriteHandle : THandle; |
| 169 | |
| 170 | //Client side anonymous pipe handles |
| 171 | FClientAnonRead, |
| 172 | FClientAnonWrite : THandle; |
| 173 | |
| 174 | protected |
| 175 | function AcceptImpl: ITransport; override; |
| 176 | |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 177 | function CreateAnonPipe : Boolean; |
| 178 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 179 | // IAnonymousPipeServerTransport |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 180 | function ReadHandle : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 181 | function WriteHandle : THandle; |
| 182 | function ClientAnonRead : THandle; |
| 183 | function ClientAnonWrite : THandle; |
| 184 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 185 | procedure InternalClose; override; |
| 186 | |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 187 | public |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 188 | constructor Create( aBufsize : Cardinal = 4096); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 189 | end; |
| 190 | |
| 191 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 192 | TNamedPipeServerTransportImpl = class( TPipeServerTransportBase, INamedPipeServerTransport) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 193 | private |
| 194 | FPipeName : string; |
| 195 | FMaxConns : DWORD; |
| 196 | FBufSize : DWORD; |
Jens Geyer | 0b20cc8 | 2013-03-07 20:47:01 +0100 | [diff] [blame] | 197 | FTimeout : DWORD; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 198 | FHandle : THandle; |
| 199 | FConnected : Boolean; |
| 200 |
|
| 201 |
protected |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 202 | function AcceptImpl: ITransport; override; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 203 | function CreateNamedPipe : THandle; |
| 204 | function CreateTransportInstance : ITransport; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 205 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 206 | // INamedPipeServerTransport |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 207 | function Handle : THandle; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 208 | procedure InternalClose; override; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 209 | |
| 210 | public |
| 211 | constructor Create( aPipename : string; aBufsize : Cardinal = 4096; |
Jens Geyer | 0b20cc8 | 2013-03-07 20:47:01 +0100 | [diff] [blame] | 212 | aMaxConns : Cardinal = PIPE_UNLIMITED_INSTANCES; |
| 213 | aTimeOut : Cardinal = 0); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 214 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 215 | |
| 216 | |
| 217 | implementation |
| 218 | |
| 219 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 220 | procedure ClosePipeHandle( var hPipe : THandle); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 221 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 222 | if hPipe <> INVALID_HANDLE_VALUE |
| 223 | then try |
| 224 | CloseHandle( hPipe); |
| 225 | finally |
| 226 | hPipe := INVALID_HANDLE_VALUE; |
| 227 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 228 | end; |
| 229 | |
| 230 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 231 | function DuplicatePipeHandle( const hSource : THandle) : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 232 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 233 | if not DuplicateHandle( GetCurrentProcess, hSource, |
| 234 | GetCurrentProcess, @result, |
| 235 | 0, FALSE, DUPLICATE_SAME_ACCESS) |
| 236 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 237 | 'DuplicateHandle: '+SysErrorMessage(GetLastError)); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 238 | end; |
| 239 | |
| 240 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 241 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 242 | { TPipeStreamBase } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 243 | |
| 244 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 245 | constructor TPipeStreamBase.Create( const aTimeOut : DWORD = DEFAULT_THRIFT_PIPE_TIMEOUT); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 246 | begin |
| 247 | inherited Create; |
| 248 | FPipe := INVALID_HANDLE_VALUE; |
| 249 | FTimeout := aTimeOut; |
| 250 | end; |
| 251 | |
| 252 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 253 | destructor TPipeStreamBase.Destroy; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 254 | begin |
| 255 | try |
| 256 | Close; |
| 257 | finally |
| 258 | inherited Destroy; |
| 259 | end; |
| 260 | end; |
| 261 | |
| 262 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 263 | procedure TPipeStreamBase.Close; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 264 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 265 | ClosePipeHandle( FPipe); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 266 | end; |
| 267 | |
| 268 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 269 | procedure TPipeStreamBase.Flush; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 270 | begin |
| 271 | // nothing to do |
| 272 | end; |
| 273 | |
| 274 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 275 | function TPipeStreamBase.IsOpen: Boolean; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 276 | begin |
| 277 | result := (FPipe <> INVALID_HANDLE_VALUE); |
| 278 | end; |
| 279 | |
| 280 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 281 | procedure TPipeStreamBase.Write(const buffer: TBytes; offset, count: Integer); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 282 | var cbWritten : DWORD; |
| 283 | begin |
| 284 | if not IsOpen |
| 285 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 286 | 'Called write on non-open pipe'); |
| 287 | |
| 288 | if not WriteFile( FPipe, buffer[offset], count, cbWritten, nil) |
| 289 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 290 | 'Write to pipe failed'); |
| 291 | end; |
| 292 | |
| 293 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 294 | function TPipeStreamBase.Read( var buffer: TBytes; offset, count: Integer): Integer; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 295 | var cbRead, dwErr : DWORD; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 296 | bytes, retries : LongInt; |
| 297 | bOk : Boolean; |
| 298 | const INTERVAL = 10; // ms |
| 299 | begin |
| 300 | if not IsOpen |
| 301 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 302 | 'Called read on non-open pipe'); |
| 303 | |
| 304 | // MSDN: Handle can be a handle to a named pipe instance, |
| 305 | // or it can be a handle to the read end of an anonymous pipe, |
| 306 | // The handle must have GENERIC_READ access to the pipe. |
| 307 | if FTimeOut <> INFINITE then begin |
| 308 | retries := Max( 1, Round( 1.0 * FTimeOut / INTERVAL)); |
| 309 | while TRUE do begin |
| 310 | if IsOpen |
| 311 | and PeekNamedPipe( FPipe, nil, 0, nil, @bytes, nil) |
| 312 | and (bytes > 0) |
| 313 | then Break; // there are data |
| 314 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 315 | dwErr := GetLastError; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 316 | if (dwErr = ERROR_INVALID_HANDLE) |
| 317 | or (dwErr = ERROR_BROKEN_PIPE) |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 318 | or (dwErr = ERROR_PIPE_NOT_CONNECTED) |
| 319 | then begin |
| 320 | result := 0; // other side closed the pipe |
| 321 | Exit; |
| 322 | end; |
| 323 | |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 324 | Dec( retries); |
| 325 | if retries > 0 |
| 326 | then Sleep( INTERVAL) |
| 327 | else raise TTransportException.Create( TTransportException.TExceptionType.TimedOut, |
| 328 | 'Pipe read timed out'); |
| 329 | end; |
| 330 | end; |
| 331 | |
| 332 | // read the data (or block INFINITE-ly) |
| 333 | bOk := ReadFile( FPipe, buffer[offset], count, cbRead, nil); |
| 334 | if (not bOk) and (GetLastError() <> ERROR_MORE_DATA) |
| 335 | then result := 0 // No more data, possibly because client disconnected. |
| 336 | else result := cbRead; |
| 337 | end; |
| 338 | |
| 339 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 340 | function TPipeStreamBase.ToArray: TBytes; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 341 | var bytes : LongInt; |
| 342 | begin |
| 343 | SetLength( result, 0); |
| 344 | bytes := 0; |
| 345 | |
| 346 | if IsOpen |
| 347 | and PeekNamedPipe( FPipe, nil, 0, nil, @bytes, nil) |
| 348 | and (bytes > 0) |
| 349 | then begin |
| 350 | SetLength( result, bytes); |
| 351 | Read( result, 0, bytes); |
| 352 | end; |
| 353 | end; |
| 354 | |
| 355 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 356 | { TNamedPipeStreamImpl } |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 357 | |
| 358 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 359 | constructor TNamedPipeStreamImpl.Create( const aPipeName : string; const aShareMode: DWORD; |
| 360 | const aSecurityAttributes: PSecurityAttributes; |
| 361 | const aTimeOut : DWORD); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 362 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 363 | inherited Create( aTimeout); |
| 364 | |
| 365 | FPipeName := aPipeName; |
| 366 | FShareMode := aShareMode; |
| 367 | FSecurityAttribs := aSecurityAttributes; |
| 368 | |
| 369 | if Copy(FPipeName,1,2) <> '\\' |
| 370 | then FPipeName := '\\.\pipe\' + FPipeName; // assume localhost |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 371 | end; |
| 372 | |
| 373 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 374 | procedure TNamedPipeStreamImpl.Open; |
| 375 | var hPipe : THandle; |
| 376 | dwMode : DWORD; |
| 377 | begin |
| 378 | if IsOpen then Exit; |
| 379 | |
| 380 | // open that thingy |
| 381 | |
| 382 | if not WaitNamedPipe( PChar(FPipeName), FTimeout) |
| 383 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 384 | 'Unable to open pipe, '+SysErrorMessage(GetLastError)); |
| 385 | |
| 386 | hPipe := CreateFile( PChar( FPipeName), |
| 387 | GENERIC_READ or GENERIC_WRITE, |
| 388 | FShareMode, // sharing |
| 389 | FSecurityAttribs, // security attributes |
| 390 | OPEN_EXISTING, // opens existing pipe |
| 391 | 0, // default attributes |
| 392 | 0); // no template file |
| 393 | |
| 394 | if hPipe = INVALID_HANDLE_VALUE |
| 395 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 396 | 'Unable to open pipe, '+SysErrorMessage(GetLastError)); |
| 397 | |
| 398 | // pipe connected; change to message-read mode. |
| 399 | dwMode := PIPE_READMODE_MESSAGE; |
| 400 | if not SetNamedPipeHandleState( hPipe, dwMode, nil, nil) then begin |
| 401 | Close; |
| 402 | raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 403 | 'SetNamedPipeHandleState failed'); |
| 404 | end; |
| 405 | |
| 406 | // everything fine |
| 407 | FPipe := hPipe; |
| 408 | end; |
| 409 | |
| 410 | |
| 411 | { THandlePipeStreamImpl } |
| 412 | |
| 413 | |
| 414 | constructor THandlePipeStreamImpl.Create( const aPipeHandle : THandle; aOwnsHandle : Boolean); |
| 415 | begin |
| 416 | inherited Create( DEFAULT_THRIFT_PIPE_TIMEOUT); |
| 417 | |
| 418 | if aOwnsHandle |
| 419 | then FSrcHandle := aPipeHandle |
| 420 | else FSrcHandle := DuplicatePipeHandle( aPipeHandle); |
| 421 | |
| 422 | Open; |
| 423 | end; |
| 424 | |
| 425 | |
| 426 | destructor THandlePipeStreamImpl.Destroy; |
| 427 | begin |
| 428 | try |
| 429 | ClosePipeHandle( FSrcHandle); |
| 430 | finally |
| 431 | inherited Destroy; |
| 432 | end; |
| 433 | end; |
| 434 | |
| 435 | |
| 436 | procedure THandlePipeStreamImpl.Open; |
| 437 | begin |
| 438 | if not IsOpen |
| 439 | then FPipe := DuplicatePipeHandle( FSrcHandle); |
| 440 | end; |
| 441 | |
| 442 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 443 | { TPipeTransportBase } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 444 | |
| 445 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 446 | function TPipeTransportBase.GetIsOpen: Boolean; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 447 | begin |
Jens Geyer | 0b20cc8 | 2013-03-07 20:47:01 +0100 | [diff] [blame] | 448 | result := (FInputStream <> nil) and (FInputStream.IsOpen) |
| 449 | and (FOutputStream <> nil) and (FOutputStream.IsOpen); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 450 | end; |
| 451 | |
| 452 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 453 | procedure TPipeTransportBase.Open; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 454 | begin |
| 455 | FInputStream.Open; |
| 456 | FOutputStream.Open; |
| 457 | end; |
| 458 | |
| 459 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 460 | procedure TPipeTransportBase.Close; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 461 | begin |
| 462 | FInputStream.Close; |
| 463 | FOutputStream.Close; |
| 464 | end; |
| 465 | |
| 466 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 467 | { TNamedPipeTransportClientEndImpl } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 468 | |
| 469 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 470 | constructor TNamedPipeTransportClientEndImpl.Create( const aPipeName : string; const aShareMode: DWORD; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 471 | const aSecurityAttributes: PSecurityAttributes; |
| 472 | const aTimeOut : DWORD); |
| 473 | // Named pipe constructor |
| 474 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 475 | inherited Create( nil, nil); |
| 476 | FInputStream := TNamedPipeStreamImpl.Create( aPipeName, aShareMode, aSecurityAttributes, aTimeOut); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 477 | FOutputStream := FInputStream; // true for named pipes |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 478 | end; |
| 479 | |
| 480 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 481 | constructor TNamedPipeTransportClientEndImpl.Create( aPipe : THandle; aOwnsHandle : Boolean); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 482 | // Named pipe constructor |
| 483 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 484 | inherited Create( nil, nil); |
| 485 | FInputStream := THandlePipeStreamImpl.Create( aPipe, aOwnsHandle); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 486 | FOutputStream := FInputStream; // true for named pipes |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 487 | end; |
| 488 | |
| 489 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 490 | { TNamedPipeTransportServerEndImpl } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 491 | |
| 492 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 493 | constructor TNamedPipeTransportServerEndImpl.Create( aPipe : THandle; aOwnsHandle : Boolean); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 494 | // Named pipe constructor |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 495 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 496 | FHandle := DuplicatePipeHandle( aPipe); |
| 497 | inherited Create( aPipe, aOwnsHandle); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 498 | end; |
| 499 | |
| 500 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 501 | procedure TNamedPipeTransportServerEndImpl.Close; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 502 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 503 | FlushFileBuffers( FHandle); |
| 504 | DisconnectNamedPipe( FHandle); // force client off the pipe |
| 505 | ClosePipeHandle( FHandle); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 506 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 507 | inherited Close; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 508 | end; |
| 509 | |
| 510 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 511 | { TAnonymousPipeTransportImpl } |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 512 | |
| 513 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 514 | constructor TAnonymousPipeTransportImpl.Create( const aPipeRead, aPipeWrite : THandle; aOwnsHandles : Boolean); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 515 | // Anonymous pipe constructor |
| 516 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 517 | inherited Create( nil, nil); |
| 518 | FInputStream := THandlePipeStreamImpl.Create( aPipeRead, aOwnsHandles); |
| 519 | FOutputStream := THandlePipeStreamImpl.Create( aPipeWrite, aOwnsHandles); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 520 | end; |
| 521 | |
| 522 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 523 | { TPipeServerTransportBase } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 524 | |
| 525 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 526 | procedure TPipeServerTransportBase.Listen; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 527 | begin |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 528 | FStopServer := FALSE; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 529 | end; |
| 530 | |
| 531 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 532 | procedure TPipeServerTransportBase.Close; |
| 533 | begin |
| 534 | FStopServer := TRUE; |
| 535 | InternalClose; |
| 536 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 537 | |
| 538 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 539 | { TAnonymousPipeServerTransportImpl } |
| 540 | |
| 541 | |
| 542 | constructor TAnonymousPipeServerTransportImpl.Create( aBufsize : Cardinal); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 543 | // Anonymous pipe CTOR |
| 544 | begin |
| 545 | inherited Create; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 546 | FBufsize := aBufSize; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 547 | FReadHandle := INVALID_HANDLE_VALUE; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 548 | FWriteHandle := INVALID_HANDLE_VALUE; |
| 549 | FClientAnonRead := INVALID_HANDLE_VALUE; |
| 550 | FClientAnonWrite := INVALID_HANDLE_VALUE; |
| 551 | |
| 552 | // The anonymous pipe needs to be created first so that the server can |
| 553 | // pass the handles on to the client before the serve (acceptImpl) |
| 554 | // blocking call. |
| 555 | if not CreateAnonPipe |
| 556 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 557 | ClassName+'.Create() failed'); |
| 558 | end; |
| 559 | |
| 560 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 561 | function TAnonymousPipeServerTransportImpl.AcceptImpl: ITransport; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 562 | var buf : Byte; |
| 563 | br : DWORD; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 564 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 565 | // This 0-byte read serves merely as a blocking call. |
| 566 | if not ReadFile( FReadHandle, buf, 0, br, nil) |
| 567 | and (GetLastError() <> ERROR_MORE_DATA) |
| 568 | then raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 569 | 'TServerPipe unable to initiate pipe communication'); |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 570 | |
| 571 | // create the transport impl |
| 572 | result := TAnonymousPipeTransportImpl.Create( FReadHandle, FWriteHandle, FALSE); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 573 | end; |
| 574 | |
| 575 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 576 | procedure TAnonymousPipeServerTransportImpl.InternalClose; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 577 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 578 | ClosePipeHandle( FReadHandle); |
| 579 | ClosePipeHandle( FWriteHandle); |
| 580 | ClosePipeHandle( FClientAnonRead); |
| 581 | ClosePipeHandle( FClientAnonWrite); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 582 | end; |
| 583 | |
| 584 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 585 | function TAnonymousPipeServerTransportImpl.ReadHandle : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 586 | begin |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 587 | result := FReadHandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 588 | end; |
| 589 | |
| 590 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 591 | function TAnonymousPipeServerTransportImpl.WriteHandle : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 592 | begin |
| 593 | result := FWriteHandle; |
| 594 | end; |
| 595 | |
| 596 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 597 | function TAnonymousPipeServerTransportImpl.ClientAnonRead : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 598 | begin |
| 599 | result := FClientAnonRead; |
| 600 | end; |
| 601 | |
| 602 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 603 | function TAnonymousPipeServerTransportImpl.ClientAnonWrite : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 604 | begin |
| 605 | result := FClientAnonWrite; |
| 606 | end; |
| 607 | |
| 608 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 609 | function TAnonymousPipeServerTransportImpl.CreateAnonPipe : Boolean; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 610 | var sd : PSECURITY_DESCRIPTOR; |
| 611 | sa : SECURITY_ATTRIBUTES; //TSecurityAttributes; |
| 612 | hCAR, hPipeW, hCAW, hPipe : THandle; |
| 613 | begin |
| 614 | result := FALSE; |
| 615 | |
| 616 | sd := PSECURITY_DESCRIPTOR( LocalAlloc( LPTR,SECURITY_DESCRIPTOR_MIN_LENGTH)); |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 617 | try |
| 618 | Win32Check( InitializeSecurityDescriptor( sd, SECURITY_DESCRIPTOR_REVISION)); |
| 619 | Win32Check( SetSecurityDescriptorDacl( sd, TRUE, nil, FALSE)); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 620 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 621 | sa.nLength := sizeof( sa); |
| 622 | sa.lpSecurityDescriptor := sd; |
| 623 | sa.bInheritHandle := TRUE; //allow passing handle to child |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 624 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 625 | if not CreatePipe( hCAR, hPipeW, @sa, FBufSize) then begin //create stdin pipe |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 626 | raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 627 | 'TServerPipe CreatePipe (anon) failed, '+SysErrorMessage(GetLastError)); |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 628 | Exit; |
| 629 | end; |
| 630 | |
| 631 | if not CreatePipe( hPipe, hCAW, @sa, FBufSize) then begin //create stdout pipe |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 632 | CloseHandle( hCAR); |
| 633 | CloseHandle( hPipeW); |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 634 | raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
| 635 | 'TServerPipe CreatePipe (anon) failed, '+SysErrorMessage(GetLastError)); |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 636 | Exit; |
| 637 | end; |
| 638 | |
| 639 | FClientAnonRead := hCAR; |
| 640 | FClientAnonWrite := hCAW; |
| 641 | FReadHandle := hPipe; |
| 642 | FWriteHandle := hPipeW; |
| 643 | |
| 644 | result := TRUE; |
| 645 | |
| 646 | finally |
| 647 | if sd <> nil then LocalFree( Cardinal(sd)); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 648 | end; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 649 | end; |
| 650 | |
| 651 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 652 | { TNamedPipeServerTransportImpl } |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 653 | |
| 654 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 655 | constructor TNamedPipeServerTransportImpl.Create( aPipename : string; aBufsize, aMaxConns, aTimeOut : Cardinal); |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 656 | // Named Pipe CTOR |
| 657 | begin |
| 658 | inherited Create; |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 659 | FPipeName := aPipename; |
| 660 | FBufsize := aBufSize; |
| 661 | FMaxConns := Max( 1, Min( PIPE_UNLIMITED_INSTANCES, aMaxConns)); |
| 662 | FHandle := INVALID_HANDLE_VALUE; |
| 663 | FTimeout := aTimeOut; |
| 664 | FConnected := FALSE; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 665 | |
| 666 | if Copy(FPipeName,1,2) <> '\\' |
| 667 | then FPipeName := '\\.\pipe\' + FPipeName; // assume localhost |
| 668 | end; |
| 669 | |
| 670 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 671 | function TNamedPipeServerTransportImpl.AcceptImpl: ITransport; |
| 672 | var dwError, dwWait, dwDummy : DWORD; |
| 673 | overlapped : TOverlapped; |
| 674 | event : TEvent;
|
| 675 | begin
|
| 676 | FillChar( overlapped, SizeOf(overlapped), 0); |
| 677 | event := TEvent.Create( nil, TRUE, FALSE, ''); // always ManualReset, see MSDN |
| 678 | try |
| 679 | overlapped.hEvent := event.Handle;
|
| 680 |
|
| 681 | ASSERT( not FConnected);
|
| 682 | while not FConnected do begin
|
| 683 | InternalClose; |
| 684 | if FStopServer then Abort; |
| 685 | CreateNamedPipe; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 686 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 687 | // Wait for the client to connect; if it succeeds, the |
| 688 | // function returns a nonzero value. If the function returns |
| 689 | // zero, GetLastError should return ERROR_PIPE_CONNECTED. |
| 690 | if ConnectNamedPipe( Handle, @overlapped) |
| 691 | then FConnected := TRUE |
| 692 | else begin |
| 693 | // ConnectNamedPipe() returns FALSE for OverlappedIO, even if connected. |
| 694 | // We have to check GetLastError() explicitly to find out
|
| 695 | dwError := GetLastError;
|
| 696 | case dwError of
|
| 697 | ERROR_PIPE_CONNECTED : begin
|
| 698 | FConnected := TRUE; // special case: pipe immediately connected
|
| 699 | end;
|
| 700 |
|
| 701 | ERROR_IO_PENDING : begin
|
| 702 | dwWait := WaitForSingleObject( overlapped.hEvent, DEFAULT_THRIFT_PIPE_TIMEOUT);
|
| 703 | FConnected := (dwWait = WAIT_OBJECT_0)
|
| 704 | and GetOverlappedResult( Handle, overlapped, dwDummy, TRUE);
|
| 705 | end;
|
| 706 |
|
| 707 | else
|
| 708 | InternalClose;
|
| 709 | raise TTransportException.Create( TTransportException.TExceptionType.NotOpen,
|
| 710 | 'Client connection failed'); |
| 711 | end;
|
| 712 | end;
|
| 713 | end;
|
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 714 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 715 | // create the transport impl |
| 716 | result := CreateTransportInstance; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 717 | |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 718 | finally |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 719 | event.Free;
|
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 720 | end; |
| 721 | end; |
| 722 | |
| 723 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 724 | function TNamedPipeServerTransportImpl.CreateTransportInstance : ITransport; |
| 725 | // create the transport impl |
| 726 | var hPipe : THandle; |
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 727 | begin |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 728 | hPipe := THandle( InterlockedExchangePointer( Pointer(FHandle), Pointer(INVALID_HANDLE_VALUE))); |
| 729 | try |
| 730 | FConnected := FALSE; |
| 731 | result := TNamedPipeTransportServerEndImpl.Create( hPipe, TRUE); |
| 732 | except |
| 733 | ClosePipeHandle(hPipe);
|
| 734 | raise;
|
| 735 | end;
|
Roger Meier | 79655fb | 2012-10-20 20:59:41 +0000 | [diff] [blame] | 736 | end; |
| 737 | |
| 738 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 739 | procedure TNamedPipeServerTransportImpl.InternalClose; |
| 740 | var hPipe : THandle; |
| 741 | begin |
| 742 | hPipe := THandle( InterlockedExchangePointer( Pointer(FHandle), Pointer(INVALID_HANDLE_VALUE))); |
| 743 | if hPipe = INVALID_HANDLE_VALUE then Exit; |
| 744 | |
| 745 | try |
| 746 | if FConnected |
| 747 | then FlushFileBuffers( hPipe) |
| 748 | else CancelIo( hPipe); |
| 749 | DisconnectNamedPipe( hPipe); |
| 750 | finally |
| 751 | ClosePipeHandle( hPipe); |
| 752 | FConnected := FALSE; |
| 753 | end; |
| 754 | end; |
| 755 | |
| 756 | |
| 757 | function TNamedPipeServerTransportImpl.Handle : THandle; |
| 758 | begin |
| 759 | {$IFDEF WIN64} |
| 760 | result := THandle( InterlockedExchangeAdd64( Integer(FHandle), 0)); |
| 761 | {$ELSE} |
| 762 | result := THandle( InterlockedExchangeAdd( Integer(FHandle), 0)); |
| 763 | {$ENDIF} |
| 764 | end; |
| 765 | |
| 766 | |
| 767 | function TNamedPipeServerTransportImpl.CreateNamedPipe : THandle; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 768 | var SIDAuthWorld : SID_IDENTIFIER_AUTHORITY ; |
| 769 | everyone_sid : PSID; |
| 770 | ea : EXPLICIT_ACCESS; |
| 771 | acl : PACL; |
| 772 | sd : PSECURITY_DESCRIPTOR; |
| 773 | sa : SECURITY_ATTRIBUTES; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 774 | const |
| 775 | SECURITY_WORLD_SID_AUTHORITY : TSIDIdentifierAuthority = (Value : (0,0,0,0,0,1)); |
| 776 | SECURITY_WORLD_RID = $00000000; |
| 777 | begin |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 778 | sd := nil; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 779 | everyone_sid := nil; |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 780 | try |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 781 | ASSERT( (FHandle = INVALID_HANDLE_VALUE) and not FConnected); |
| 782 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 783 | // Windows - set security to allow non-elevated apps |
| 784 | // to access pipes created by elevated apps. |
| 785 | SIDAuthWorld := SECURITY_WORLD_SID_AUTHORITY; |
| 786 | AllocateAndInitializeSid( SIDAuthWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, everyone_sid); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 787 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 788 | ZeroMemory( @ea, SizeOf(ea)); |
| 789 | ea.grfAccessPermissions := GENERIC_ALL; //SPECIFIC_RIGHTS_ALL or STANDARD_RIGHTS_ALL; |
| 790 | ea.grfAccessMode := SET_ACCESS; |
| 791 | ea.grfInheritance := NO_INHERITANCE; |
| 792 | ea.Trustee.TrusteeForm := TRUSTEE_IS_SID; |
| 793 | ea.Trustee.TrusteeType := TRUSTEE_IS_WELL_KNOWN_GROUP; |
| 794 | ea.Trustee.ptstrName := PChar(everyone_sid); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 795 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 796 | acl := nil; |
| 797 | SetEntriesInAcl( 1, @ea, nil, acl); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 798 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 799 | sd := PSECURITY_DESCRIPTOR( LocalAlloc( LPTR,SECURITY_DESCRIPTOR_MIN_LENGTH)); |
| 800 | Win32Check( InitializeSecurityDescriptor( sd, SECURITY_DESCRIPTOR_REVISION)); |
| 801 | Win32Check( SetSecurityDescriptorDacl( sd, TRUE, acl, FALSE)); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 802 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 803 | sa.nLength := SizeOf(sa); |
| 804 | sa.lpSecurityDescriptor := sd; |
| 805 | sa.bInheritHandle := FALSE; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 806 | |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 807 | // Create an instance of the named pipe |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 808 | result := Windows.CreateNamedPipe( PChar( FPipeName), // pipe name |
| 809 | PIPE_ACCESS_DUPLEX or // read/write access |
| 810 | FILE_FLAG_OVERLAPPED, // async mode |
| 811 | PIPE_TYPE_MESSAGE or // message type pipe |
| 812 | PIPE_READMODE_MESSAGE, // message-read mode |
| 813 | FMaxConns, // max. instances |
| 814 | FBufSize, // output buffer size |
| 815 | FBufSize, // input buffer size |
| 816 | FTimeout, // time-out, see MSDN |
| 817 | @sa); // default security attribute |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 818 | |
Jens Geyer | 06045cf | 2013-03-27 20:26:25 +0200 | [diff] [blame] | 819 | if( result <> INVALID_HANDLE_VALUE) |
| 820 | then InterlockedExchangePointer( Pointer(FHandle), Pointer(result)) |
| 821 | else raise TTransportException.Create( TTransportException.TExceptionType.NotOpen, |
Jens Geyer | b64a774 | 2013-01-23 20:58:47 +0100 | [diff] [blame] | 822 | 'CreateNamedPipe() failed ' + IntToStr(GetLastError)); |
| 823 | |
| 824 | finally |
| 825 | if sd <> nil then LocalFree( Cardinal( sd)); |
| 826 | if acl <> nil then LocalFree( Cardinal( acl)); |
| 827 | if everyone_sid <> nil then FreeSid(everyone_sid); |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 828 | end; |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 829 | end; |
| 830 | |
| 831 | |
Roger Meier | 3bef8c2 | 2012-10-06 06:58:00 +0000 | [diff] [blame] | 832 | |
| 833 | end. |
| 834 | |
| 835 | |
| 836 | |