aboutsummaryrefslogtreecommitdiffstats
path: root/lib/erl_interface/src/connect/eirecv.c
diff options
context:
space:
mode:
authorErlang/OTP <[email protected]>2009-11-20 14:54:40 +0000
committerErlang/OTP <[email protected]>2009-11-20 14:54:40 +0000
commit84adefa331c4159d432d22840663c38f155cd4c1 (patch)
treebff9a9c66adda4df2106dfd0e5c053ab182a12bd /lib/erl_interface/src/connect/eirecv.c
downloadotp-84adefa331c4159d432d22840663c38f155cd4c1.tar.gz
otp-84adefa331c4159d432d22840663c38f155cd4c1.tar.bz2
otp-84adefa331c4159d432d22840663c38f155cd4c1.zip
The R13B03 release.OTP_R13B03
Diffstat (limited to 'lib/erl_interface/src/connect/eirecv.c')
-rw-r--r--lib/erl_interface/src/connect/eirecv.c280
1 files changed, 280 insertions, 0 deletions
diff --git a/lib/erl_interface/src/connect/eirecv.c b/lib/erl_interface/src/connect/eirecv.c
new file mode 100644
index 0000000000..51fc32d65c
--- /dev/null
+++ b/lib/erl_interface/src/connect/eirecv.c
@@ -0,0 +1,280 @@
+/*
+ * %CopyrightBegin%
+ *
+ * Copyright Ericsson AB 1998-2009. All Rights Reserved.
+ *
+ * The contents of this file are subject to the Erlang Public License,
+ * Version 1.1, (the "License"); you may not use this file except in
+ * compliance with the License. You should have received a copy of the
+ * Erlang Public License along with this software. If not, it can be
+ * retrieved online at http://www.erlang.org/.
+ *
+ * Software distributed under the License is distributed on an "AS IS"
+ * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
+ * the License for the specific language governing rights and limitations
+ * under the License.
+ *
+ * %CopyrightEnd%
+ */
+#ifdef __WIN32__
+# include <winsock2.h>
+# include <windows.h>
+# include <winbase.h>
+#else /* Unix/VxWorks */
+# include <unistd.h>
+#endif
+
+/* common */
+#include <string.h>
+#include <stdlib.h>
+#include <errno.h>
+
+#include "eidef.h"
+#include "eiext.h"
+#include "eirecv.h"
+#include "ei_portio.h"
+#include "ei_internal.h"
+#include "putget.h"
+#include "ei_trace.h"
+#include "show_msg.h"
+
+#include <errno.h>
+
+#define EIRECVBUF 2048 /* largest possible header is approx 1300 bytes */
+
+/* length (4), PASS_THOUGH (1), header, message */
+int
+ei_recv_internal (int fd,
+ char **mbufp, int *bufsz,
+ erlang_msg *msg, int *msglenp,
+ int staticbufp, unsigned ms)
+{
+ char header[EIRECVBUF];
+ char *s=header;
+ char *mbuf=*mbufp;
+ int len = 0;
+ int msglen = 0;
+ int bytesread = 0;
+ int remain;
+ int arity;
+ int version;
+ int index = 0;
+ int i = 0;
+ int res;
+ int show_this_msg = 0;
+
+ /* get length field */
+ if ((res = ei_read_fill_t(fd, header, 4, ms)) != 4)
+ {
+ erl_errno = (res == -2) ? ETIMEDOUT : EIO;
+ return -1;
+ }
+ len = get32be(s);
+
+ /* got tick - respond and return */
+ if (!len) {
+ char tock[] = {0,0,0,0};
+ ei_write_fill_t(fd, tock, sizeof(tock), ms); /* Failure no problem */
+ *msglenp = 0;
+ return 0; /* maybe flag ERL_EAGAIN [sverkerw] */
+ }
+
+ /* turn off tracing on each receive. it will be turned back on if
+ * we receive a trace token.
+ */
+ ei_trace(-1,NULL);
+
+ /* read enough to get at least entire header */
+ bytesread = (len > EIRECVBUF ? EIRECVBUF : len);
+ if ((i = ei_read_fill_t(fd,header,bytesread,ms)) != bytesread) {
+ erl_errno = (i == -2) ? ETIMEDOUT : EIO;
+ return -1;
+ }
+
+ /* now decode header */
+ /* pass-through, version, control tuple header, control message type */
+ s = header;
+ index = 1;
+ if ((get8(s) != ERL_PASS_THROUGH)
+ || ei_decode_version(header,&index,&version)
+ || (version != ERL_VERSION_MAGIC)
+ || ei_decode_tuple_header(header,&index,&arity)
+ || ei_decode_long(header,&index,&msg->msgtype))
+ {
+ erl_errno = EIO; /* Maybe another code for decoding errors */
+ return -1;
+ }
+
+ switch (msg->msgtype) {
+ case ERL_SEND: /* { SEND, Cookie, ToPid } */
+ if (ei_tracelevel > 0) show_this_msg = 1;
+ if (ei_decode_atom(header,&index,msg->cookie)
+ || ei_decode_pid(header,&index,&msg->to))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ break;
+
+ case ERL_REG_SEND: /* { REG_SEND, From, Cookie, ToName } */
+ if (ei_tracelevel > 0) show_this_msg = 1;
+ if (ei_decode_pid(header,&index,&msg->from)
+ || ei_decode_atom(header,&index,msg->cookie)
+ || ei_decode_atom(header,&index,msg->toname))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ /* actual message is remaining part of headerbuf, plus any unread bytes */
+ break;
+
+ case ERL_LINK: /* { LINK, From, To } */
+ case ERL_UNLINK: /* { UNLINK, From, To } */
+ case ERL_GROUP_LEADER: /* { GROUP_LEADER, From, To } */
+ if (ei_tracelevel > 1) show_this_msg = 1;
+ if (ei_decode_pid(header,&index,&msg->from)
+ || ei_decode_pid(header,&index,&msg->to))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ break;
+
+ case ERL_EXIT: /* { EXIT, From, To, Reason } */
+ case ERL_EXIT2: /* { EXIT2, From, To, Reason } */
+ if (ei_tracelevel > 1) show_this_msg = 1;
+ if (ei_decode_pid(header,&index,&msg->from)
+ || ei_decode_pid(header,&index,&msg->to))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ break;
+
+ case ERL_SEND_TT: /* { SEND_TT, Cookie, ToPid, TraceToken } */
+ if (ei_tracelevel > 0) show_this_msg = 1;
+ if (ei_decode_atom(header,&index,msg->cookie)
+ || ei_decode_pid(header,&index,&msg->to)
+ || ei_decode_trace(header,&index,&msg->token))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ ei_trace(1,&msg->token); /* turn on tracing */
+ break;
+
+ case ERL_REG_SEND_TT: /* { REG_SEND_TT, From, Cookie, ToName, TraceToken } */
+ if (ei_tracelevel > 0) show_this_msg = 1;
+ if (ei_decode_pid(header,&index,&msg->from)
+ || ei_decode_atom(header,&index,msg->cookie)
+ || ei_decode_atom(header,&index,msg->toname)
+ || ei_decode_trace(header,&index,&msg->token))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ ei_trace(1,&msg->token); /* turn on tracing */
+ break;
+
+ case ERL_EXIT_TT: /* { EXIT_TT, From, To, TraceToken, Reason } */
+ case ERL_EXIT2_TT: /* { EXIT2_TT, From, To, TraceToken, Reason } */
+ if (ei_tracelevel > 1) show_this_msg = 1;
+ if (ei_decode_pid(header,&index,&msg->from)
+ || ei_decode_pid(header,&index,&msg->to)
+ || ei_decode_trace(header,&index,&msg->token))
+ {
+ erl_errno = EIO;
+ return -1;
+ }
+
+ ei_trace(1,&msg->token); /* turn on tracing */
+ break;
+
+ case ERL_NODE_LINK: /* { NODE_LINK } */
+ if (ei_tracelevel > 1) show_this_msg = 1;
+ break;
+
+ default:
+ /* unknown type, just put any remaining bytes into buffer */
+ break;
+ }
+
+ /* actual message is remaining part of headerbuf, plus any unread bytes */
+ msglen = len - index; /* message size (payload) */
+ remain = len - bytesread; /* bytes left to read */
+
+ /* if callers buffer is too small, we flush in the rest of the
+ * message and discard it, unless we know that we can reallocate
+ * the buffer in which case we do that and read the message.
+ */
+ if (msglen > *bufsz) {
+ if (staticbufp) {
+ int sz = EIRECVBUF;
+ /* flush in rest of packet */
+ while (remain > 0) {
+ if (remain < sz) sz = remain;
+ if ((i=ei_read_fill_t(fd,header,sz,ms)) <= 0) break;
+ remain -= i;
+ }
+ erl_errno = EMSGSIZE;
+ return -1;
+ }
+ else {
+ /* Dynamic buffer --- grow it. */
+#ifdef DEBUG
+ fprintf(stderr, "Growing buffer from %d bytes to %d bytes\n",
+ *bufsz, msglen);
+#endif
+ if ((mbuf = realloc(*mbufp, msglen)) == NULL)
+ {
+ erl_errno = ENOMEM;
+ return -1;
+ }
+
+ *mbufp = mbuf;
+ *bufsz = msglen;
+ }
+ }
+
+ /* move remaining bytes to callers buffer */
+ memmove(mbuf,header+index,bytesread-index);
+
+ /* let the caller know how big the message is in his buffer */
+ *msglenp = msglen;
+
+ /* read the rest of the message into callers buffer */
+ if (remain > 0) {
+ if ((i = ei_read_fill_t(fd,mbuf+bytesread-index,remain,ms)) != remain) {
+ *msglenp = bytesread-index+1; /* actual bytes in users buffer */
+ erl_errno = (i == -2) ? ETIMEDOUT : EIO;
+ return -1;
+ }
+ }
+
+ if (show_this_msg)
+ ei_show_recmsg(stderr,msg,mbuf);
+
+ /* the caller only sees "untraced" message types */
+ /* the trace token is buried in the message struct */
+ if (msg->msgtype > 10) msg->msgtype -= 10;
+
+ return msg->msgtype;
+}
+
+int ei_receive_encoded(int fd, char **mbufp, int *bufsz,
+ erlang_msg *msg, int *msglen)
+{
+ return ei_recv_internal(fd, mbufp, bufsz, msg, msglen, 0, 0);
+}
+
+int ei_receive_encoded_tmo(int fd, char **mbufp, int *bufsz, erlang_msg *msg, int *msglen, unsigned ms)
+{
+ return ei_recv_internal(fd, mbufp, bufsz, msg, msglen, 0, ms);
+}
+