OneScript/src/VSCode.DebugAdapter/Protocol.cs at develop · dmpas/OneScript · GitHub

History

270 lines (218 loc) · 6.67 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

/*----------------------------------------------------------

This Source Code Form is subject to the terms of the

Mozilla Public License, v.2.0. If a copy of the MPL

was not distributed with this file, You can obtain one

at http://mozilla.org/MPL/2.0/.

----------------------------------------------------------*/

/*---------------------------------------------------------------------------------------------

* Licensed under the MIT License. See License.txt in the project root for license information.

*--------------------------------------------------------------------------------------------*/

using System;

using System.Text;

using System.IO;

using System.Text.RegularExpressions;

using Newtonsoft.Json;

using Serilog;

namespace VSCodeDebug

{

public class ProtocolMessage

{

public int seq;

public string type { get; }

public ProtocolMessage(string typ) {

type = typ;

}

public ProtocolMessage(string typ, int sq) {

type = typ;

seq = sq;

}

public class Request : ProtocolMessage

{

public string command;

public dynamic arguments;

public Request(int id, string cmd, dynamic arg) : base("request", id) {

command = cmd;

arguments = arg;

}

* subclasses of ResponseBody are serialized as the body of a response.

* Don't change their instance variables since that will break the debug protocol.

public class ResponseBody {

// empty

}

public class Response : ProtocolMessage

{

public bool success { get; private set; }

public string message { get; private set; }

public int request_seq { get; }

public string command { get; }

public ResponseBody body { get; private set; }

public Response(Request req) : base("response") {

success = true;

request_seq = req.seq;

command = req.command;

}

public void SetBody(ResponseBody bdy) {

success = true;

body = bdy;

}

public void SetErrorBody(string msg, ResponseBody bdy = null) {

success = false;

message = msg;

body = bdy;

}

public class Event : ProtocolMessage

{

[JsonProperty(PropertyName = "event")]

public string eventType { get; }

public dynamic body { get; }

public Event(string type, dynamic bdy = null) : base("event") {

eventType = type;

body = bdy;

}

* The ProtocolServer can be used to implement a server that uses the VSCode debug protocol.

public abstract class ProtocolServer

{

protected const int BUFFER_SIZE = 4096;

protected const string TWO_CRLF = "\r\n\r\n";

protected static readonly Regex CONTENT_LENGTH_MATCHER = new Regex(@"Content-Length: (\d+)");

protected static readonly Encoding Encoding = System.Text.Encoding.UTF8;

private int _sequenceNumber = 1;

private Stream _outputStream;

private ByteBuffer _rawData = new ByteBuffer();

private int _bodyLength = -1;

private bool _stopRequested;

public void Start(Stream inputStream, Stream outputStream)

{

_outputStream = outputStream;

byte[] buffer = new byte[BUFFER_SIZE];

_stopRequested = false;

while (!_stopRequested) {

var read = inputStream.Read(buffer, 0, buffer.Length);

if (read == 0) {

// end of stream

break;

}

if (read > 0) {

_rawData.Append(buffer, read);

ProcessData();

}

public void Stop()

{

_stopRequested = true;

}

public void SendEvent(Event e)

{

SendMessage(e);

}

protected abstract void DispatchRequest(string command, dynamic args, Response response);

// ---- private ------------------------------------------------------------------------

private void ProcessData()

{

while (true) {

if (_bodyLength >= 0) {

if (_rawData.Length >= _bodyLength) {

var buf = _rawData.RemoveFirst(_bodyLength);

_bodyLength = -1;

Dispatch(Encoding.GetString(buf));

continue; // there may be more complete messages to process

}

else {

string s = _rawData.GetString(Encoding);

var idx = s.IndexOf(TWO_CRLF);

if (idx != -1) {

Match m = CONTENT_LENGTH_MATCHER.Match(s);

if (m.Success && m.Groups.Count == 2) {

_bodyLength = Convert.ToInt32(m.Groups[1].ToString());

_rawData.RemoveFirst(idx + TWO_CRLF.Length);

continue; // try to handle a complete message

}

break;

}

private void Dispatch(string req)

{

var request = JsonConvert.DeserializeObject<Request>(req);

if (request != null && request.type == "request") {

Log.Verbose("Got {Command} with args {Args}", request.command, JsonConvert.SerializeObject(request.arguments));

var response = new Response(request);

DispatchRequest(request.command, request.arguments, response);

SendMessage(response);

}

protected void SendMessage(ProtocolMessage message)

{

message.seq = _sequenceNumber++;

if (message.type == "response") {

Log.Verbose("Response {Response}", JsonConvert.SerializeObject(message));

}

if (message.type == "event") {

Event e = (Event)message;

Log.Verbose("Event {EventType} with args {Args}", e.eventType, JsonConvert.SerializeObject(e.body));

}

var data = ConvertToBytes(message);

try {

_outputStream.Write(data, 0, data.Length);

_outputStream.Flush();

}

catch (Exception) {

// ignore

}

private static byte[] ConvertToBytes(ProtocolMessage request)

{

var asJson = JsonConvert.SerializeObject(request);

byte[] jsonBytes = Encoding.GetBytes(asJson);

string header = string.Format("Content-Length: {0}{1}", jsonBytes.Length, TWO_CRLF);

byte[] headerBytes = Encoding.GetBytes(header);

byte[] data = new byte[headerBytes.Length + jsonBytes.Length];

System.Buffer.BlockCopy(headerBytes, 0, data, 0, headerBytes.Length);

System.Buffer.BlockCopy(jsonBytes, 0, data, headerBytes.Length, jsonBytes.Length);

return data;

}

//--------------------------------------------------------------------------------------

class ByteBuffer

{

private byte[] _buffer;

public ByteBuffer() {

_buffer = new byte[0];

}

public int Length {

get { return _buffer.Length; }

}

public string GetString(Encoding enc)

{

return enc.GetString(_buffer);

}

public void Append(byte[] b, int length)

{

byte[] newBuffer = new byte[_buffer.Length + length];

System.Buffer.BlockCopy(_buffer, 0, newBuffer, 0, _buffer.Length);

System.Buffer.BlockCopy(b, 0, newBuffer, _buffer.Length, length);

_buffer = newBuffer;

}

public byte[] RemoveFirst(int n)

{

byte[] b = new byte[n];

System.Buffer.BlockCopy(_buffer, 0, b, 0, n);

byte[] newBuffer = new byte[_buffer.Length - n];

System.Buffer.BlockCopy(_buffer, n, newBuffer, 0, _buffer.Length - n);

_buffer = newBuffer;

return b;

}

OneScript/src/VSCode.DebugAdapter/Protocol.cs at develop · dmpas/OneScript · GitHub

Expand file tree

Breadcrumbs

Protocol.cs

Latest commit

History

Breadcrumbs

Protocol.cs

File metadata and controls