Merge commit 'be3e8236086165e5e45a5a10783823874b3f3ebd' as 'lib/vscode'

lib/vscode/extensions/git/src/ipc/ipcClient.ts

@@ -0,0 +1,45 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as http from 'http';
+
+export class IPCClient {
+
+	private ipcHandlePath: string;
+
+	constructor(private handlerName: string) {
+		const ipcHandlePath = process.env['VSCODE_GIT_IPC_HANDLE'];
+
+		if (!ipcHandlePath) {
+			throw new Error('Missing VSCODE_GIT_IPC_HANDLE');
+		}
+
+		this.ipcHandlePath = ipcHandlePath;
+	}
+
+	call(request: any): Promise<any> {
+		const opts: http.RequestOptions = {
+			socketPath: this.ipcHandlePath,
+			path: `/${this.handlerName}`,
+			method: 'POST'
+		};
+
+		return new Promise((c, e) => {
+			const req = http.request(opts, res => {
+				if (res.statusCode !== 200) {
+					return e(new Error(`Bad status code: ${res.statusCode}`));
+				}
+
+				const chunks: Buffer[] = [];
+				res.on('data', d => chunks.push(d));
+				res.on('end', () => c(JSON.parse(Buffer.concat(chunks).toString('utf8'))));
+			});
+
+			req.on('error', err => e(err));
+			req.write(JSON.stringify(request));
+			req.end();
+		});
+	}
+}

lib/vscode/extensions/git/src/ipc/ipcServer.ts

@@ -0,0 +1,121 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Disposable } from 'vscode';
+import { toDisposable } from '../util';
+import * as path from 'path';
+import * as http from 'http';
+import * as os from 'os';
+import * as fs from 'fs';
+import * as crypto from 'crypto';
+
+function getIPCHandlePath(id: string): string {
+	if (process.platform === 'win32') {
+		return `\\\\.\\pipe\\vscode-git-${id}-sock`;
+	}
+
+	if (process.env['XDG_RUNTIME_DIR']) {
+		return path.join(process.env['XDG_RUNTIME_DIR'] as string, `vscode-git-${id}.sock`);
+	}
+
+	return path.join(os.tmpdir(), `vscode-git-${id}.sock`);
+}
+
+export interface IIPCHandler {
+	handle(request: any): Promise<any>;
+}
+
+export async function createIPCServer(context?: string): Promise<IIPCServer> {
+	const server = http.createServer();
+	const hash = crypto.createHash('sha1');
+
+	if (!context) {
+		const buffer = await new Promise<Buffer>((c, e) => crypto.randomBytes(20, (err, buf) => err ? e(err) : c(buf)));
+		hash.update(buffer);
+	} else {
+		hash.update(context);
+	}
+
+	const ipcHandlePath = getIPCHandlePath(hash.digest('hex').substr(0, 10));
+
+	if (process.platform !== 'win32') {
+		try {
+			await fs.promises.unlink(ipcHandlePath);
+		} catch {
+			// noop
+		}
+	}
+
+	return new Promise((c, e) => {
+		try {
+			server.on('error', err => e(err));
+			server.listen(ipcHandlePath);
+			c(new IPCServer(server, ipcHandlePath));
+		} catch (err) {
+			e(err);
+		}
+	});
+}
+
+export interface IIPCServer extends Disposable {
+	readonly ipcHandlePath: string | undefined;
+	getEnv(): { [key: string]: string; };
+	registerHandler(name: string, handler: IIPCHandler): Disposable;
+}
+
+class IPCServer implements IIPCServer, Disposable {
+
+	private handlers = new Map<string, IIPCHandler>();
+	get ipcHandlePath(): string { return this._ipcHandlePath; }
+
+	constructor(private server: http.Server, private _ipcHandlePath: string) {
+		this.server.on('request', this.onRequest.bind(this));
+	}
+
+	registerHandler(name: string, handler: IIPCHandler): Disposable {
+		this.handlers.set(`/${name}`, handler);
+		return toDisposable(() => this.handlers.delete(name));
+	}
+
+	private onRequest(req: http.IncomingMessage, res: http.ServerResponse): void {
+		if (!req.url) {
+			console.warn(`Request lacks url`);
+			return;
+		}
+
+		const handler = this.handlers.get(req.url);
+
+		if (!handler) {
+			console.warn(`IPC handler for ${req.url} not found`);
+			return;
+		}
+
+		const chunks: Buffer[] = [];
+		req.on('data', d => chunks.push(d));
+		req.on('end', () => {
+			const request = JSON.parse(Buffer.concat(chunks).toString('utf8'));
+			handler.handle(request).then(result => {
+				res.writeHead(200);
+				res.end(JSON.stringify(result));
+			}, () => {
+				res.writeHead(500);
+				res.end();
+			});
+		});
+	}
+
+	getEnv(): { [key: string]: string; } {
+		return { VSCODE_GIT_IPC_HANDLE: this.ipcHandlePath };
+	}
+
+	dispose(): void {
+		this.handlers.clear();
+		this.server.close();
+
+		if (this._ipcHandlePath && process.platform !== 'win32') {
+			fs.unlinkSync(this._ipcHandlePath);
+		}
+	}
+}