/*
* Simple serial terminal
*
- * Copyright 2014, 2015 Michal Sojka <sojkam1@fel.cvut.cz>
+ * Copyright 2014, 2015, 2016, 2017, 2019 Michal Sojka <sojkam1@fel.cvut.cz>
*
* This program is free software: you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
*/
#define _BSD_SOURCE
+#define _DEFAULT_SOURCE
+#define _GNU_SOURCE
#include <sys/ioctl.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdbool.h>
#include <string.h>
#include <signal.h>
+#ifdef HAVE_LOCKDEV
#include <lockdev.h>
+#endif
+#include <sys/file.h>
+#include <errno.h>
#define STRINGIFY(val) #val
#define TOSTRING(val) STRINGIFY(val)
#define CHECK(cmd) ({ int ret = (cmd); if (ret == -1) { perror(#cmd " line " TOSTRING(__LINE__)); exit(1); }; ret; })
#define CHECKPTR(cmd) ({ void *ptr = (cmd); if (ptr == (void*)-1) { perror(#cmd " line " TOSTRING(__LINE__)); exit(1); }; ptr; })
+#define CHECKNULL(cmd) ({ void *ptr = (cmd); if (ptr == NULL) { perror(#cmd " line " TOSTRING(__LINE__)); exit(1); }; ptr; })
#define VERBOSE(format, ...) do { if (verbose) fprintf(stderr, "sterm: " format, ##__VA_ARGS__); } while (0)
tcsetattr(0, TCSANOW, &stdin_tio_backup);
}
+#ifdef HAVE_LOCKDEV
void unlock()
{
dev_unlock(dev, getpid());
}
+#endif
void sighandler(int arg)
{
exit(0); /* Invoke exit handlers */
}
-int dtr_rts_arg(const char option)
+int dtr_rts_arg(const char option, const char *optarg)
{
int val = -1;
if (optarg) {
- switch (optarg[0]) {
- case '+': val = +1; break;
- case '-': val = -1; break;
- default:
- fprintf(stderr, "Unknown -%c argument: %s", option, optarg);
- exit(1);
+ char *end;
+ val = strtol(optarg, &end, 10);
+ if (end == optarg) {
+ /* Not a number */
+ switch (optarg[0]) {
+ case '+': val = +1; break;
+ case '-': val = -1; break;
+ default:
+ fprintf(stderr, "Unknown -%c argument: %s\n", option, optarg);
+ exit(1);
+ }
}
}
return val;
void usage(const char* argv0)
{
fprintf(stderr, "Usage: %s [options] <device>\n", argv0);
- fprintf(stderr, "Options:\n");
- fprintf(stderr, " -d [+|-] create short positive/negative pulse on DTR\n");
- fprintf(stderr, " -e ignore '~.' escape sequence\n");
- fprintf(stderr, " -n do not switch the device to raw mode\n");
- fprintf(stderr, " -r [+|-] create short positive/negative pulse on RTS\n");
- fprintf(stderr, " -s <baudrate>\n");
- fprintf(stderr, " -v verbose\n");
+ fprintf(stderr,
+ "Options:\n"
+ " -b <duration> send break signal\n"
+ " -c enter command mode\n"
+ " -d[PULSE] make pulse on DTR\n"
+ " -e ignore '~.' escape sequence\n"
+ " -n do not switch the device to raw mode\n"
+ " -r[PULSE] make pulse on RTS\n"
+ " -s <baudrate>\n"
+ " -v verbose mode\n"
+ "\n"
+ "PULSE is a number specifying the pulse. Absolute value defines the\n"
+ "length of the pulse in milliseconds, sign determines the polarity of\n"
+ "the pulse. Alternatively, PULSE can be either '+' or '-', which\n"
+ "corresponds to +1 or -1.\n"
+ );
+}
+
+void pulse(int fd, int dtr, int rts)
+{
+ int status, ms = 0;
+ /* tio.c_cflag &= ~HUPCL; */ /* Don't lower DTR/RTS on close */
+
+ CHECK(ioctl(fd, TIOCMGET, &status));
+ if (dtr > 0) { status &= ~TIOCM_DTR; ms = +dtr; }
+ if (dtr < 0) { status |= TIOCM_DTR; ms = -dtr; }
+ if (rts > 0) { status &= ~TIOCM_RTS; ms = +rts; }
+ if (rts < 0) { status |= TIOCM_RTS; ms = -rts; }
+ CHECK(ioctl(fd, TIOCMSET, &status));
+
+ usleep(ms*1000);
+
+ if (dtr < 0) { status &= ~TIOCM_DTR; }
+ if (dtr > 0) { status |= TIOCM_DTR; }
+ if (rts < 0) { status &= ~TIOCM_RTS; }
+ if (rts > 0) { status |= TIOCM_RTS; }
+ CHECK(ioctl(fd, TIOCMSET, &status));
+}
+
+void handle_commands(int fd)
+{
+ char command[100];
+ bool go = false;
+
+ while (!go) {
+ char *p1 = NULL;
+ int num;
+ if (fgets(command, sizeof(command), stdin) == NULL) {
+ if (!feof(stdin))
+ perror("Command read");
+ exit(1);
+ }
+ if (sscanf(command, "dtr %ms", &p1) == 1)
+ pulse(fd, dtr_rts_arg('d', p1), 0);
+ else if (sscanf(command, "rts %ms", &p1) == 1)
+ pulse(fd, 0, dtr_rts_arg('r', p1));
+ else if (sscanf(command, "break %d", &num) == 1)
+ CHECK(tcsendbreak(fd, num));
+ else if (strcmp(command, "go\n") == 0)
+ break;
+ else if (strcmp(command, "exit\n") == 0)
+ exit(0);
+ else {
+ fprintf(stderr, "Unknown command: %s\n", command);
+ exit(1);
+ }
+
+ free(p1);
+ }
}
int main(int argc, char *argv[])
struct termios tio;
bool stdin_tty;
bool raw = true;
+ bool cmd = false;
+ int break_dur = -1;
if ((stdin_tty = isatty(0))) {
CHECK(tcgetattr(0, &stdin_tio_backup));
atexit(restore_stdin_term);
}
- while ((opt = getopt(argc, argv, "nd::er::s:v")) != -1) {
+ while ((opt = getopt(argc, argv, "b:cnd::er::s:v")) != -1) {
switch (opt) {
- case 'd': dtr = dtr_rts_arg(opt); break;
+ case 'b': break_dur = atoi(optarg); break;
+ case 'c': cmd = true; break;
+ case 'd': dtr = dtr_rts_arg(opt, optarg); break;
case 'e': exit_on_escape = false; break;
case 'n': raw = false; break;
- case 'r': rts = dtr_rts_arg(opt); break;
+ case 'r': rts = dtr_rts_arg(opt, optarg); break;
case 's': {
int s = atoi(optarg);
switch (s) {
signal(SIGTERM, sighandler);
signal(SIGHUP, sighandler);
+#ifdef HAVE_LOCKDEV
pid_t pid = dev_lock(dev);
if (pid > 0) {
fprintf(stderr, "%s is used by PID %d\n", dev, pid);
exit(1);
} else if (pid < 0) {
- perror("dev_lock()");
+ char *msg;
+ asprintf(&msg, "dev_lock('%s')", dev); /* No free() because we exit() immediately */
+ if (errno)
+ perror(msg);
+ else
+ fprintf(stderr, "%s: Error\n", msg);
exit(1);
}
atexit(unlock);
+#endif
/* O_NONBLOCK is needed to not wait for the CDC signal. See tty_ioctl(4). */
if ((fd = open(dev, O_RDWR|O_NOCTTY|O_NONBLOCK)) < 0) {
int n = fcntl(fd, F_GETFL, 0);
fcntl(fd, F_SETFL, n & ~O_NDELAY);
+ flock(fd, LOCK_EX);
+
if (isatty(fd)) {
CHECK(ioctl(fd, TIOCEXCL, NULL));
CHECK(cfsetispeed(&tio, speed));
}
- if (dtr || rts) {
- int status;
- /* tio.c_cflag &= ~HUPCL; */ /* Don't lower DTR/RTS on close */
+ if (dtr || rts)
+ pulse(fd, dtr, rts);
- CHECK(ioctl(fd, TIOCMGET, &status));
- if (dtr == -1) status &= ~TIOCM_DTR;
- if (dtr == +1) status |= TIOCM_DTR;
- if (rts == -1) status &= ~TIOCM_RTS;
- if (rts == +1) status |= TIOCM_RTS;
- CHECK(ioctl(fd, TIOCMSET, &status));
- }
+ if (break_dur != -1)
+ CHECK(tcsendbreak(fd, break_dur));
+
+ /* Disable flow control */
+ tio.c_cflag &= ~(CRTSCTS);
+ tio.c_iflag &= ~(IXON|IXOFF);
CHECK(tcsetattr(fd, TCSANOW, &tio));
} else if (speed || dtr || rts) {
exit(1);
}
+ VERBOSE("Connected.\r\n");
+
+ if (cmd)
+ handle_commands(fd);
+
struct pollfd fds[2] = {
{ .fd = 0, .events = POLLIN },
{ .fd = fd, .events = POLLIN },
CHECK(tcsetattr(0, TCSANOW, &tio));
}
- VERBOSE("Connected.\r\n");
if (exit_on_escape)
VERBOSE("Use '<Enter>~.' sequence to exit.\r\n");
exit(1);
}
}
+ if (fds[0].revents & POLLHUP) {
+ VERBOSE("HUP on stdin\r\n");
+ break;
+ }
if (fds[1].revents & POLLIN) {
r1 = CHECK(read(fd, buf, sizeof(buf)));
if (r1 == 0) {
projects / sojka / sterm.git / blobdiff