1 /* Jim - A small embeddable Tcl interpreter
3 * Copyright 2005 Salvatore Sanfilippo <antirez@invece.org>
4 * Copyright 2005 Clemens Hintze <c.hintze@gmx.net>
5 * Copyright 2005 patthoyts - Pat Thoyts <patthoyts@users.sf.net>
6 * Copyright 2008 oharboe - Øyvind Harboe - oyvind.harboe@zylin.com
7 * Copyright 2008 Andrew Lunn <andrew@lunn.ch>
8 * Copyright 2008 Duane Ellis <openocd@duaneellis.com>
9 * Copyright 2008 Uwe Klein <uklein@klein-messgeraete.de>
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
15 * 1. Redistributions of source code must retain the above copyright
16 * notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above
18 * copyright notice, this list of conditions and the following
19 * disclaimer in the documentation and/or other materials
20 * provided with the distribution.
22 * THIS SOFTWARE IS PROVIDED BY THE JIM TCL PROJECT ``AS IS'' AND ANY
23 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
24 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
25 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
26 * JIM TCL PROJECT OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
27 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
28 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
31 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
32 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
33 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35 * The views and conclusions contained in the software and documentation
36 * are those of the authors and should not be interpreted as representing
37 * official policies, either expressed or implied, of the Jim Tcl Project.
40 #include "jimautoconf.h"
52 #if defined(HAVE_SYS_SOCKET_H) && defined(HAVE_SELECT) && defined(HAVE_NETINET_IN_H) && defined(HAVE_NETDB_H) && defined(HAVE_ARPA_INET_H)
53 #include <sys/socket.h>
54 #include <netinet/in.h>
55 #include <arpa/inet.h>
64 #include "jim-eventloop.h"
65 #include "jim-subcmd.h"
67 #define AIO_CMD_LEN 32 /* e.g. aio.handleXXXXXX */
68 #define AIO_BUF_LEN 256 /* Can keep this small and rely on stdio buffering */
77 #define AIO_KEEPOPEN 1
88 #if !defined(JIM_ANSIC) && !defined(JIM_BOOTSTRAP)
91 struct sockaddr_in sin
;
93 struct sockaddr_in6 sin6
;
97 #ifndef HAVE_INET_NTOP
98 const char *inet_ntop(int af
, const void *src
, char *dst
, int size
)
103 snprintf(dst
, size
, "%s", inet_ntoa(((struct sockaddr_in
*)src
)->sin_addr
));
107 #endif /* JIM_BOOTSTRAP */
109 typedef struct AioFile
114 int openFlags
; /* AIO_KEEPOPEN? keep FILE* */
122 static int JimAioSubCmdProc(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
);
123 static int JimMakeChannel(Jim_Interp
*interp
, FILE *fh
, int fd
, Jim_Obj
*filename
,
124 const char *hdlfmt
, int family
, const char *mode
);
126 #if !defined(JIM_ANSIC) && !defined(JIM_BOOTSTRAP)
127 static int JimParseIPv6Address(Jim_Interp
*interp
, const char *hostport
, union sockaddr_any
*sa
, int *salen
)
131 * An IPv6 addr/port looks like:
134 * [fe80::223:6cff:fe95:bdc0%en1]:2000
138 * Note that the "any" address is ::, which is the same as when no address is specified.
146 stport
= strrchr(hostport
, ':');
148 /* No : so, the whole thing is the port */
151 sthost
= Jim_StrDup(hostport
);
157 if (*hostport
== '[') {
158 /* This is a numeric ipv6 address */
159 char *pt
= strchr(++hostport
, ']');
161 sthost
= Jim_StrDupLen(hostport
, pt
- hostport
);
166 sthost
= Jim_StrDupLen(hostport
, stport
- hostport
- 1);
169 memset(&req
, '\0', sizeof(req
));
170 req
.ai_family
= PF_INET6
;
172 if (getaddrinfo(sthost
, NULL
, &req
, &ai
)) {
173 Jim_SetResultFormatted(interp
, "Not a valid address: %s", hostport
);
177 memcpy(&sa
->sin
, ai
->ai_addr
, ai
->ai_addrlen
);
178 *salen
= ai
->ai_addrlen
;
180 sa
->sin
.sin_port
= htons(atoi(stport
));
188 Jim_SetResultString(interp
, "ipv6 not supported", -1);
193 static int JimParseIpAddress(Jim_Interp
*interp
, const char *hostport
, union sockaddr_any
*sa
, int *salen
)
195 /* An IPv4 addr/port looks like:
200 * If the address is missing, INADDR_ANY is used.
201 * If the port is missing, 0 is used (only useful for server sockets).
207 stport
= strrchr(hostport
, ':');
209 /* No : so, the whole thing is the port */
211 sthost
= Jim_StrDup("0.0.0.0");
214 sthost
= Jim_StrDupLen(hostport
, stport
- hostport
);
219 #ifdef HAVE_GETADDRINFO
222 memset(&req
, '\0', sizeof(req
));
223 req
.ai_family
= PF_INET
;
225 if (getaddrinfo(sthost
, NULL
, &req
, &ai
)) {
229 memcpy(&sa
->sin
, ai
->ai_addr
, ai
->ai_addrlen
);
230 *salen
= ai
->ai_addrlen
;
238 if ((he
= gethostbyname(sthost
)) != NULL
) {
239 if (he
->h_length
== sizeof(sa
->sin
.sin_addr
)) {
240 *salen
= sizeof(sa
->sin
);
241 sa
->sin
.sin_family
= he
->h_addrtype
;
242 memcpy(&sa
->sin
.sin_addr
, he
->h_addr
, he
->h_length
); /* set address */
248 sa
->sin
.sin_port
= htons(atoi(stport
));
253 Jim_SetResultFormatted(interp
, "Not a valid address: %s", hostport
);
260 static int JimParseDomainAddress(Jim_Interp
*interp
, const char *path
, struct sockaddr_un
*sa
)
262 sa
->sun_family
= PF_UNIX
;
263 snprintf(sa
->sun_path
, sizeof(sa
->sun_path
), "%s", path
);
268 #endif /* JIM_BOOTSTRAP */
270 static void JimAioSetError(Jim_Interp
*interp
, Jim_Obj
*name
)
273 Jim_SetResultFormatted(interp
, "%#s: %s", name
, strerror(errno
));
276 Jim_SetResultString(interp
, strerror(errno
), -1);
280 static void JimAioDelProc(Jim_Interp
*interp
, void *privData
)
282 AioFile
*af
= privData
;
286 if (!(af
->openFlags
& AIO_KEEPOPEN
)) {
290 Jim_DecrRefCount(interp
, af
->filename
);
292 #ifdef jim_ext_eventloop
293 /* remove all existing EventHandlers */
294 Jim_DeleteFileHandler(interp
, af
->fp
, JIM_EVENT_READABLE
| JIM_EVENT_WRITABLE
| JIM_EVENT_EXCEPTION
);
299 static int JimCheckStreamError(Jim_Interp
*interp
, AioFile
*af
)
301 if (!ferror(af
->fp
)) {
305 /* EAGAIN and similar are not error conditions. Just treat them like eof */
306 if (feof(af
->fp
) || errno
== EAGAIN
|| errno
== EINTR
) {
310 if (errno
== ECONNRESET
) {
315 if (errno
!= ECONNABORTED
) {
319 JimAioSetError(interp
, af
->filename
);
323 static int aio_cmd_read(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
325 AioFile
*af
= Jim_CmdPrivData(interp
);
326 char buf
[AIO_BUF_LEN
];
329 jim_wide neededLen
= -1; /* -1 is "read as much as possible" */
331 if (argc
&& Jim_CompareStringImmediate(interp
, argv
[0], "-nonewline")) {
337 if (Jim_GetWide(interp
, argv
[0], &neededLen
) != JIM_OK
)
340 Jim_SetResultString(interp
, "invalid parameter: negative len", -1);
347 objPtr
= Jim_NewStringObj(interp
, NULL
, 0);
348 while (neededLen
!= 0) {
352 if (neededLen
== -1) {
353 readlen
= AIO_BUF_LEN
;
356 readlen
= (neededLen
> AIO_BUF_LEN
? AIO_BUF_LEN
: neededLen
);
358 retval
= fread(buf
, 1, readlen
, af
->fp
);
360 Jim_AppendString(interp
, objPtr
, buf
, retval
);
361 if (neededLen
!= -1) {
365 if (retval
!= readlen
)
368 /* Check for error conditions */
369 if (JimCheckStreamError(interp
, af
)) {
370 Jim_FreeNewObj(interp
, objPtr
);
375 const char *s
= Jim_GetString(objPtr
, &len
);
377 if (len
> 0 && s
[len
- 1] == '\n') {
379 objPtr
->bytes
[objPtr
->length
] = '\0';
382 Jim_SetResult(interp
, objPtr
);
386 static int aio_cmd_copy(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
388 AioFile
*af
= Jim_CmdPrivData(interp
);
390 jim_wide maxlen
= JIM_WIDE_MAX
;
391 FILE *outfh
= Jim_AioFilehandle(interp
, argv
[0]);
398 if (Jim_GetWide(interp
, argv
[1], &maxlen
) != JIM_OK
) {
403 while (count
< maxlen
) {
404 int ch
= fgetc(af
->fp
);
406 if (ch
== EOF
|| fputc(ch
, outfh
) == EOF
) {
412 if (ferror(af
->fp
)) {
413 Jim_SetResultFormatted(interp
, "error while reading: %s", strerror(errno
));
419 Jim_SetResultFormatted(interp
, "error while writing: %s", strerror(errno
));
424 Jim_SetResultInt(interp
, count
);
429 static int aio_cmd_gets(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
431 AioFile
*af
= Jim_CmdPrivData(interp
);
432 char buf
[AIO_BUF_LEN
];
438 objPtr
= Jim_NewStringObj(interp
, NULL
, 0);
440 buf
[AIO_BUF_LEN
- 1] = '_';
441 if (fgets(buf
, AIO_BUF_LEN
, af
->fp
) == NULL
)
444 if (buf
[AIO_BUF_LEN
- 1] == '\0' && buf
[AIO_BUF_LEN
- 2] != '\n') {
445 Jim_AppendString(interp
, objPtr
, buf
, AIO_BUF_LEN
- 1);
450 if (len
&& (buf
[len
- 1] == '\n')) {
455 Jim_AppendString(interp
, objPtr
, buf
, len
);
459 if (JimCheckStreamError(interp
, af
)) {
461 Jim_FreeNewObj(interp
, objPtr
);
466 if (Jim_SetVariable(interp
, argv
[0], objPtr
) != JIM_OK
) {
467 Jim_FreeNewObj(interp
, objPtr
);
471 len
= Jim_Length(objPtr
);
473 if (len
== 0 && feof(af
->fp
)) {
474 /* On EOF returns -1 if varName was specified */
477 Jim_SetResultInt(interp
, len
);
480 Jim_SetResult(interp
, objPtr
);
485 static int aio_cmd_puts(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
487 AioFile
*af
= Jim_CmdPrivData(interp
);
493 if (!Jim_CompareStringImmediate(interp
, argv
[0], "-nonewline")) {
502 wdata
= Jim_GetString(strObj
, &wlen
);
503 if (fwrite(wdata
, 1, wlen
, af
->fp
) == (unsigned)wlen
) {
504 if (argc
== 2 || putc('\n', af
->fp
) != EOF
) {
508 JimAioSetError(interp
, af
->filename
);
512 static int aio_cmd_isatty(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
515 AioFile
*af
= Jim_CmdPrivData(interp
);
516 Jim_SetResultInt(interp
, isatty(fileno(af
->fp
)));
518 Jim_SetResultInt(interp
, 0);
524 #if !defined(JIM_ANSIC) && !defined(JIM_BOOTSTRAP)
525 static int aio_cmd_recvfrom(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
527 AioFile
*af
= Jim_CmdPrivData(interp
);
529 union sockaddr_any sa
;
531 socklen_t salen
= sizeof(sa
);
534 if (Jim_GetLong(interp
, argv
[0], &len
) != JIM_OK
) {
538 buf
= Jim_Alloc(len
+ 1);
540 rlen
= recvfrom(fileno(af
->fp
), buf
, len
, 0, &sa
.sa
, &salen
);
543 JimAioSetError(interp
, NULL
);
547 Jim_SetResult(interp
, Jim_NewStringObjNoAlloc(interp
, buf
, rlen
));
550 /* INET6_ADDRSTRLEN is 46. Add some for [] and port */
554 if (sa
.sa
.sa_family
== PF_INET6
) {
556 /* Allow 9 for []:65535\0 */
557 inet_ntop(sa
.sa
.sa_family
, &sa
.sin6
.sin6_addr
, addrbuf
+ 1, sizeof(addrbuf
) - 9);
558 snprintf(addrbuf
+ strlen(addrbuf
), 8, "]:%d", ntohs(sa
.sin
.sin_port
));
562 if (sa
.sa
.sa_family
== PF_INET
) {
563 /* Allow 7 for :65535\0 */
564 inet_ntop(sa
.sa
.sa_family
, &sa
.sin
.sin_addr
, addrbuf
, sizeof(addrbuf
) - 7);
565 snprintf(addrbuf
+ strlen(addrbuf
), 7, ":%d", ntohs(sa
.sin
.sin_port
));
568 /* recvfrom still works on unix domain sockets, etc */
572 if (Jim_SetVariable(interp
, argv
[1], Jim_NewStringObj(interp
, addrbuf
, -1)) != JIM_OK
) {
581 static int aio_cmd_sendto(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
583 AioFile
*af
= Jim_CmdPrivData(interp
);
587 union sockaddr_any sa
;
588 const char *addr
= Jim_String(argv
[1]);
591 if (IPV6
&& af
->addr_family
== PF_INET6
) {
592 if (JimParseIPv6Address(interp
, addr
, &sa
, &salen
) != JIM_OK
) {
596 else if (JimParseIpAddress(interp
, addr
, &sa
, &salen
) != JIM_OK
) {
599 wdata
= Jim_GetString(argv
[0], &wlen
);
601 /* Note that we don't validate the socket type. Rely on sendto() failing if appropriate */
602 len
= sendto(fileno(af
->fp
), wdata
, wlen
, 0, &sa
.sa
, salen
);
604 JimAioSetError(interp
, NULL
);
607 Jim_SetResultInt(interp
, len
);
611 static int aio_cmd_accept(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
613 AioFile
*af
= Jim_CmdPrivData(interp
);
615 union sockaddr_any sa
;
616 socklen_t addrlen
= sizeof(sa
);
618 sock
= accept(af
->fd
, &sa
.sa
, &addrlen
);
620 JimAioSetError(interp
, NULL
);
624 /* Create the file command */
625 return JimMakeChannel(interp
, NULL
, sock
, Jim_NewStringObj(interp
, "accept", -1),
626 "aio.sockstream%ld", af
->addr_family
, "r+");
629 static int aio_cmd_listen(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
631 AioFile
*af
= Jim_CmdPrivData(interp
);
634 if (Jim_GetLong(interp
, argv
[0], &backlog
) != JIM_OK
) {
638 if (listen(af
->fd
, backlog
)) {
639 JimAioSetError(interp
, NULL
);
645 #endif /* JIM_BOOTSTRAP */
647 static int aio_cmd_flush(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
649 AioFile
*af
= Jim_CmdPrivData(interp
);
651 if (fflush(af
->fp
) == EOF
) {
652 JimAioSetError(interp
, af
->filename
);
658 static int aio_cmd_eof(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
660 AioFile
*af
= Jim_CmdPrivData(interp
);
662 Jim_SetResultInt(interp
, feof(af
->fp
));
666 static int aio_cmd_close(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
669 #if !defined(JIM_ANSIC) && defined(HAVE_SHUTDOWN)
670 static const char * const options
[] = { "r", "w", NULL
};
671 enum { OPT_R
, OPT_W
, };
673 AioFile
*af
= Jim_CmdPrivData(interp
);
675 if (Jim_GetEnum(interp
, argv
[2], options
, &option
, NULL
, JIM_ERRMSG
) != JIM_OK
) {
678 if (shutdown(af
->fd
, option
== OPT_R
? SHUT_RD
: SHUT_WR
) == 0) {
681 JimAioSetError(interp
, NULL
);
683 Jim_SetResultString(interp
, "async close not supported", -1);
688 return Jim_DeleteCommand(interp
, Jim_String(argv
[0]));
691 static int aio_cmd_seek(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
693 AioFile
*af
= Jim_CmdPrivData(interp
);
698 if (Jim_CompareStringImmediate(interp
, argv
[1], "start"))
700 else if (Jim_CompareStringImmediate(interp
, argv
[1], "current"))
702 else if (Jim_CompareStringImmediate(interp
, argv
[1], "end"))
708 if (Jim_GetWide(interp
, argv
[0], &offset
) != JIM_OK
) {
711 if (fseeko(af
->fp
, offset
, orig
) == -1) {
712 JimAioSetError(interp
, af
->filename
);
718 static int aio_cmd_tell(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
720 AioFile
*af
= Jim_CmdPrivData(interp
);
722 Jim_SetResultInt(interp
, ftello(af
->fp
));
726 static int aio_cmd_filename(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
728 AioFile
*af
= Jim_CmdPrivData(interp
);
730 Jim_SetResult(interp
, af
->filename
);
735 static int aio_cmd_ndelay(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
737 AioFile
*af
= Jim_CmdPrivData(interp
);
739 int fmode
= fcntl(af
->fd
, F_GETFL
);
744 if (Jim_GetLong(interp
, argv
[0], &nb
) != JIM_OK
) {
753 fcntl(af
->fd
, F_SETFL
, fmode
);
755 Jim_SetResultInt(interp
, (fmode
& O_NONBLOCK
) ? 1 : 0);
760 static int aio_cmd_buffering(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
762 AioFile
*af
= Jim_CmdPrivData(interp
);
764 static const char * const options
[] = {
778 if (Jim_GetEnum(interp
, argv
[0], options
, &option
, NULL
, JIM_ERRMSG
) != JIM_OK
) {
783 setvbuf(af
->fp
, NULL
, _IONBF
, 0);
786 setvbuf(af
->fp
, NULL
, _IOLBF
, BUFSIZ
);
789 setvbuf(af
->fp
, NULL
, _IOFBF
, BUFSIZ
);
795 #ifdef jim_ext_eventloop
796 static void JimAioFileEventFinalizer(Jim_Interp
*interp
, void *clientData
)
798 Jim_Obj
**objPtrPtr
= clientData
;
800 Jim_DecrRefCount(interp
, *objPtrPtr
);
804 static int JimAioFileEventHandler(Jim_Interp
*interp
, void *clientData
, int mask
)
806 Jim_Obj
**objPtrPtr
= clientData
;
808 return Jim_EvalObjBackground(interp
, *objPtrPtr
);
811 static int aio_eventinfo(Jim_Interp
*interp
, AioFile
* af
, unsigned mask
, Jim_Obj
**scriptHandlerObj
,
812 int argc
, Jim_Obj
* const *argv
)
815 /* Return current script */
816 if (*scriptHandlerObj
) {
817 Jim_SetResult(interp
, *scriptHandlerObj
);
822 if (*scriptHandlerObj
) {
823 /* Delete old handler */
824 Jim_DeleteFileHandler(interp
, af
->fp
, mask
);
827 /* Now possibly add the new script(s) */
828 if (Jim_Length(argv
[0]) == 0) {
829 /* Empty script, so done */
833 /* A new script to add */
834 Jim_IncrRefCount(argv
[0]);
835 *scriptHandlerObj
= argv
[0];
837 Jim_CreateFileHandler(interp
, af
->fp
, mask
,
838 JimAioFileEventHandler
, scriptHandlerObj
, JimAioFileEventFinalizer
);
843 static int aio_cmd_readable(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
845 AioFile
*af
= Jim_CmdPrivData(interp
);
847 return aio_eventinfo(interp
, af
, JIM_EVENT_READABLE
, &af
->rEvent
, argc
, argv
);
850 static int aio_cmd_writable(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
852 AioFile
*af
= Jim_CmdPrivData(interp
);
854 return aio_eventinfo(interp
, af
, JIM_EVENT_WRITABLE
, &af
->wEvent
, argc
, argv
);
857 static int aio_cmd_onexception(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
859 AioFile
*af
= Jim_CmdPrivData(interp
);
861 return aio_eventinfo(interp
, af
, JIM_EVENT_EXCEPTION
, &af
->wEvent
, argc
, argv
);
865 static const jim_subcmd_type aio_command_table
[] = {
867 "?-nonewline? ?len?",
871 /* Description: Read and return bytes from the stream. To eof if no len. */
878 /* Description: Copy up to 'size' bytes to the given filehandle, or to eof if no size. */
885 /* Description: Read one line and return it or store it in the var */
892 /* Description: Write the string, with newline unless -nonewline */
899 /* Description: Is the file descriptor a tty? */
901 #if !defined(JIM_ANSIC) && !defined(JIM_BOOTSTRAP)
907 /* Description: Receive up to 'len' bytes on the socket. Sets 'addrvar' with receive address, if set */
914 /* Description: Send 'str' to the given address (dgram only) */
921 /* Description: Server socket only: Accept a connection and return stream */
928 /* Description: Set the listen backlog for server socket */
930 #endif /* JIM_BOOTSTRAP */
936 /* Description: Flush the stream */
943 /* Description: Returns 1 if stream is at eof */
950 JIM_MODFLAG_FULLARGV
,
951 /* Description: Closes the stream. */
954 "offset ?start|current|end",
958 /* Description: Seeks in the stream (default 'current') */
965 /* Description: Returns the current seek position */
972 /* Description: Returns the original filename */
980 /* Description: Set O_NDELAY (if arg). Returns current/new setting. */
988 /* Description: Sets buffering */
990 #ifdef jim_ext_eventloop
996 /* Description: Returns script, or invoke readable-script when readable, {} to remove */
1003 /* Description: Returns script, or invoke writable-script when writable, {} to remove */
1006 "?exception-script?",
1007 aio_cmd_onexception
,
1010 /* Description: Returns script, or invoke exception-script when oob data, {} to remove */
1016 static int JimAioSubCmdProc(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
1018 return Jim_CallSubCmd(interp
, Jim_ParseSubCmd(interp
, aio_command_table
, argc
, argv
), argc
, argv
);
1021 static int JimAioOpenCommand(Jim_Interp
*interp
, int argc
,
1022 Jim_Obj
*const *argv
)
1026 if (argc
!= 2 && argc
!= 3) {
1027 Jim_WrongNumArgs(interp
, 1, argv
, "filename ?mode?");
1031 mode
= (argc
== 3) ? Jim_String(argv
[2]) : "r";
1033 #ifdef jim_ext_tclcompat
1035 const char *filename
= Jim_String(argv
[1]);
1037 /* If the filename starts with '|', use popen instead */
1038 if (*filename
== '|') {
1039 Jim_Obj
*evalObj
[3];
1041 evalObj
[0] = Jim_NewStringObj(interp
, "::popen", -1);
1042 evalObj
[1] = Jim_NewStringObj(interp
, filename
+ 1, -1);
1043 evalObj
[2] = Jim_NewStringObj(interp
, mode
, -1);
1045 return Jim_EvalObjVector(interp
, 3, evalObj
);
1049 return JimMakeChannel(interp
, NULL
, -1, argv
[1], "aio.handle%ld", 0, mode
);
1053 * Creates a channel for fh/fd/filename.
1055 * If fh is not NULL, uses that as the channel (and sets AIO_KEEPOPEN).
1056 * Otherwise, if fd is >= 0, uses that as the channel.
1057 * Otherwise opens 'filename' with mode 'mode'.
1059 * hdlfmt is a sprintf format for the filehandle. Anything with %ld at the end will do.
1060 * mode is used for open or fdopen.
1062 * Creates the command and sets the name as the current result.
1064 static int JimMakeChannel(Jim_Interp
*interp
, FILE *fh
, int fd
, Jim_Obj
*filename
,
1065 const char *hdlfmt
, int family
, const char *mode
)
1068 char buf
[AIO_CMD_LEN
];
1072 filename
= Jim_NewStringObj(interp
, hdlfmt
, -1);
1073 openFlags
= AIO_KEEPOPEN
;
1076 Jim_IncrRefCount(filename
);
1079 #if !defined(JIM_ANSIC)
1081 fh
= fdopen(fd
, mode
);
1085 fh
= fopen(Jim_String(filename
), mode
);
1088 JimAioSetError(interp
, filename
);
1089 #if !defined(JIM_ANSIC)
1094 Jim_DecrRefCount(interp
, filename
);
1099 /* Create the file command */
1100 af
= Jim_Alloc(sizeof(*af
));
1101 memset(af
, 0, sizeof(*af
));
1103 af
->fd
= fileno(fh
);
1104 af
->filename
= filename
;
1106 if ((openFlags
& AIO_KEEPOPEN
) == 0) {
1107 fcntl(af
->fd
, F_SETFD
, FD_CLOEXEC
);
1110 af
->openFlags
= openFlags
;
1111 af
->addr_family
= family
;
1112 snprintf(buf
, sizeof(buf
), hdlfmt
, Jim_GetId(interp
));
1113 Jim_CreateCommand(interp
, buf
, JimAioSubCmdProc
, af
, JimAioDelProc
);
1115 /* Note that the command must use the global namespace, even if
1116 * the current namespace is something different
1118 Jim_SetResult(interp
, Jim_MakeGlobalNamespaceName(interp
, Jim_NewStringObj(interp
, buf
, -1)));
1123 static int JimMakeChannelPair(Jim_Interp
*interp
, int p
[2], Jim_Obj
*filename
,
1124 const char *hdlfmt
, int family
, const char *mode
[2])
1126 if (JimMakeChannel(interp
, NULL
, p
[0], filename
, hdlfmt
, family
, mode
[0]) == JIM_OK
) {
1127 Jim_Obj
*objPtr
= Jim_NewListObj(interp
, NULL
, 0);
1128 Jim_ListAppendElement(interp
, objPtr
, Jim_GetResult(interp
));
1130 if (JimMakeChannel(interp
, NULL
, p
[1], filename
, hdlfmt
, family
, mode
[1]) == JIM_OK
) {
1131 Jim_ListAppendElement(interp
, objPtr
, Jim_GetResult(interp
));
1132 Jim_SetResult(interp
, objPtr
);
1137 /* Can only be here if fdopen() failed */
1140 JimAioSetError(interp
, NULL
);
1144 #if !defined(JIM_ANSIC) && !defined(JIM_BOOTSTRAP)
1146 static int JimAioSockCommand(Jim_Interp
*interp
, int argc
, Jim_Obj
*const *argv
)
1148 const char *hdlfmt
= "aio.unknown%ld";
1149 const char *socktypes
[] = {
1169 SOCK_STREAM_SOCKETPAIR
,
1173 const char *hostportarg
= NULL
;
1176 const char *mode
= "r+";
1177 int family
= PF_INET
;
1178 Jim_Obj
*argv0
= argv
[0];
1181 if (argc
> 1 && Jim_CompareStringImmediate(interp
, argv
[1], "-ipv6")) {
1183 Jim_SetResultString(interp
, "ipv6 not supported", -1);
1194 Jim_WrongNumArgs(interp
, 1, &argv0
, "?-ipv6? type ?address?");
1198 if (Jim_GetEnum(interp
, argv
[1], socktypes
, &socktype
, "socket type", JIM_ERRMSG
) != JIM_OK
)
1201 Jim_SetEmptyResult(interp
);
1203 hdlfmt
= "aio.sock%ld";
1206 hostportarg
= Jim_String(argv
[2]);
1210 case SOCK_DGRAM_CLIENT
:
1212 /* No address, so an unconnected dgram socket */
1213 sock
= socket(family
, SOCK_DGRAM
, 0);
1215 JimAioSetError(interp
, NULL
);
1221 case SOCK_STREAM_CLIENT
:
1223 union sockaddr_any sa
;
1231 if (JimParseIPv6Address(interp
, hostportarg
, &sa
, &salen
) != JIM_OK
) {
1235 else if (JimParseIpAddress(interp
, hostportarg
, &sa
, &salen
) != JIM_OK
) {
1238 sock
= socket(family
, (socktype
== SOCK_DGRAM_CLIENT
) ? SOCK_DGRAM
: SOCK_STREAM
, 0);
1240 JimAioSetError(interp
, NULL
);
1243 res
= connect(sock
, &sa
.sa
, salen
);
1245 JimAioSetError(interp
, argv
[2]);
1252 case SOCK_STREAM_SERVER
:
1253 case SOCK_DGRAM_SERVER
:
1255 union sockaddr_any sa
;
1263 if (JimParseIPv6Address(interp
, hostportarg
, &sa
, &salen
) != JIM_OK
) {
1267 else if (JimParseIpAddress(interp
, hostportarg
, &sa
, &salen
) != JIM_OK
) {
1270 sock
= socket(family
, (socktype
== SOCK_DGRAM_SERVER
) ? SOCK_DGRAM
: SOCK_STREAM
, 0);
1272 JimAioSetError(interp
, NULL
);
1276 /* Enable address reuse */
1277 setsockopt(sock
, SOL_SOCKET
, SO_REUSEADDR
, (void *)&on
, sizeof(on
));
1279 res
= bind(sock
, &sa
.sa
, salen
);
1281 JimAioSetError(interp
, argv
[2]);
1285 if (socktype
== SOCK_STREAM_SERVER
) {
1286 res
= listen(sock
, 5);
1288 JimAioSetError(interp
, NULL
);
1293 hdlfmt
= "aio.socksrv%ld";
1297 #ifdef HAVE_SYS_UN_H
1300 struct sockaddr_un sa
;
1303 if (argc
!= 3 || ipv6
) {
1307 if (JimParseDomainAddress(interp
, hostportarg
, &sa
) != JIM_OK
) {
1308 JimAioSetError(interp
, argv
[2]);
1312 sock
= socket(PF_UNIX
, SOCK_STREAM
, 0);
1314 JimAioSetError(interp
, NULL
);
1317 len
= strlen(sa
.sun_path
) + 1 + sizeof(sa
.sun_family
);
1318 res
= connect(sock
, (struct sockaddr
*)&sa
, len
);
1320 JimAioSetError(interp
, argv
[2]);
1324 hdlfmt
= "aio.sockunix%ld";
1328 case SOCK_UNIX_SERVER
:
1330 struct sockaddr_un sa
;
1333 if (argc
!= 3 || ipv6
) {
1337 if (JimParseDomainAddress(interp
, hostportarg
, &sa
) != JIM_OK
) {
1338 JimAioSetError(interp
, argv
[2]);
1342 sock
= socket(PF_UNIX
, SOCK_STREAM
, 0);
1344 JimAioSetError(interp
, NULL
);
1347 len
= strlen(sa
.sun_path
) + 1 + sizeof(sa
.sun_family
);
1348 res
= bind(sock
, (struct sockaddr
*)&sa
, len
);
1350 JimAioSetError(interp
, argv
[2]);
1354 res
= listen(sock
, 5);
1356 JimAioSetError(interp
, NULL
);
1360 hdlfmt
= "aio.sockunixsrv%ld";
1365 #if defined(HAVE_SOCKETPAIR) && defined(HAVE_SYS_UN_H)
1366 case SOCK_STREAM_SOCKETPAIR
:
1369 static const char *mode
[2] = { "r+", "r+" };
1371 if (argc
!= 2 || ipv6
) {
1375 if (socketpair(PF_UNIX
, SOCK_STREAM
, 0, p
) < 0) {
1376 JimAioSetError(interp
, NULL
);
1379 return JimMakeChannelPair(interp
, p
, argv
[1], "aio.sockpair%ld", PF_UNIX
, mode
);
1384 #if defined(HAVE_PIPE)
1385 case SOCK_STREAM_PIPE
:
1388 static const char *mode
[2] = { "r", "w" };
1390 if (argc
!= 2 || ipv6
) {
1395 JimAioSetError(interp
, NULL
);
1399 return JimMakeChannelPair(interp
, p
, argv
[1], "aio.pipe%ld", 0, mode
);
1405 Jim_SetResultString(interp
, "Unsupported socket type", -1);
1409 return JimMakeChannel(interp
, NULL
, sock
, argv
[1], hdlfmt
, family
, mode
);
1411 #endif /* JIM_BOOTSTRAP */
1413 FILE *Jim_AioFilehandle(Jim_Interp
*interp
, Jim_Obj
*command
)
1415 Jim_Cmd
*cmdPtr
= Jim_GetCommand(interp
, command
, JIM_ERRMSG
);
1417 /* XXX: There ought to be a supported API for this */
1418 if (cmdPtr
&& !cmdPtr
->isproc
&& cmdPtr
->u
.native
.cmdProc
== JimAioSubCmdProc
) {
1419 return ((AioFile
*) cmdPtr
->u
.native
.privData
)->fp
;
1421 Jim_SetResultFormatted(interp
, "Not a filehandle: \"%#s\"", command
);
1425 int Jim_aioInit(Jim_Interp
*interp
)
1427 if (Jim_PackageProvide(interp
, "aio", "1.0", JIM_ERRMSG
))
1430 Jim_CreateCommand(interp
, "open", JimAioOpenCommand
, NULL
, NULL
);
1432 Jim_CreateCommand(interp
, "socket", JimAioSockCommand
, NULL
, NULL
);
1435 /* Create filehandles for stdin, stdout and stderr */
1436 JimMakeChannel(interp
, stdin
, -1, NULL
, "stdin", 0, "r");
1437 JimMakeChannel(interp
, stdout
, -1, NULL
, "stdout", 0, "w");
1438 JimMakeChannel(interp
, stderr
, -1, NULL
, "stderr", 0, "w");