...

HalleyAssist · Feb 20, 2024 · edf5851 · edf5851
1 parent ab49308
commit edf5851
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -76,6 +76,7 @@ class Socket extends EventEmitter {
 			this.wrap.recv (this.buffer, function (buffer, bytes, source) {
 				const newBuffer = buffer.slice (0, bytes);
 				me.emit ("message", newBuffer, source, buffer);
+				return this.buffer
 			});
 		} catch (error) {
 			me.emit ("error", error);

diff --git a/src/raw.cc b/src/raw.cc
@@ -739,7 +739,11 @@ NAN_METHOD(SocketWrap::Recv) {
 			addr[0] = 0; /* TODO */
 		argv[1] = Nan::New<Number>(rc);
 		argv[2] = Nan::New(addr).ToLocalChecked();
-		cb->Call (context, socket->handle(), argc, argv);
+		MaybeLocal<v8::Value> newBuffer = cb->Call (context, socket->handle(), argc, argv);
+		if (newBuffer.IsEmpty()) {
+			break;
+		}
+		buffer = newBuffer.ToLocalChecked()->ToObject(context).ToLocalChecked();
 	} while(true);
 
 	info.GetReturnValue().Set(info.This());