[Asterisk-code-review] main: Slight refactor of main. Improve color situation. (asterisk[master])
Walter Doekes
asteriskteam at digium.com
Tue Nov 24 16:29:48 CST 2015
Walter Doekes has uploaded a new change for review.
https://gerrit.asterisk.org/1714
Change subject: main: Slight refactor of main. Improve color situation.
......................................................................
main: Slight refactor of main. Improve color situation.
Several issues are addressed here:
- main() is large, and half of it is only used if we're not
rasterisk; fixed by spliting up the daemon part into a separate
function.
- Call ast_term_init from rasterisk as well.
- Remove duplicate code reading/writing asterisk history file.
- Attempt to tackle background color issues and color changes that
occur. Tested by starting asterisk -c until the colors stopped
changing at odd locations.
- Remove unused term_prep() and term_prompt() functions.
ASTERISK-25585 #close
Change-Id: Ib641a0964c59ef9fe6f59efa8ccb481a9580c52f
---
M include/asterisk/term.h
M main/asterisk.c
M main/term.c
3 files changed, 81 insertions(+), 112 deletions(-)
git pull ssh://gerrit.asterisk.org:29418/asterisk refs/changes/14/1714/1
diff --git a/include/asterisk/term.h b/include/asterisk/term.h
index 18d743b..f91b047 100644
--- a/include/asterisk/term.h
+++ b/include/asterisk/term.h
@@ -67,8 +67,8 @@
#define COLORIZE_FMT "%s%s%s"
#define COLORIZE(fg, bg, str) ast_term_color(fg,bg),str,ast_term_reset()
/*! \brief Maximum number of characters needed for a color escape sequence,
- * plus a null char */
-#define AST_TERM_MAX_ESCAPE_CHARS 12
+ * and another one for a trailing reset, plus a null char */
+#define AST_TERM_MAX_ESCAPE_CHARS 23
#define AST_TERM_MAX_ROTATING_BUFFERS 15
/*! \brief Colorize a specified string by adding terminal color codes
@@ -136,10 +136,6 @@
char *term_strip(char *outbuf, const char *inbuf, int maxout);
void term_filter_escapes(char *line);
-
-char *term_prompt(char *outbuf, const char *inbuf, int maxout);
-
-const char *term_prep(void);
const char *term_end(void);
diff --git a/main/asterisk.c b/main/asterisk.c
index e0a87d6..6728e73 100644
--- a/main/asterisk.c
+++ b/main/asterisk.c
@@ -373,9 +373,14 @@
char ast_defaultlanguage[MAX_LANGUAGE] = DEFAULT_LANGUAGE;
-static int ast_el_add_history(char *);
-static int ast_el_read_history(char *);
-static int ast_el_write_history(char *);
+static int ast_el_add_history(const char *);
+static int ast_el_read_history(const char *);
+static int ast_el_write_history(const char *);
+
+static void ast_el_read_default_histfile(void);
+static void ast_el_write_default_histfile(void);
+
+static void asterisk_daemon(int isroot, const char *runuser, const char *rungroup);
struct _cfg_paths {
char config_dir[PATH_MAX];
@@ -1965,13 +1970,7 @@
}
if (ast_opt_console || (ast_opt_remote && !ast_opt_exec)) {
- char filename[80] = "";
- if (getenv("HOME")) {
- snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
- }
- if (!ast_strlen_zero(filename)) {
- ast_el_write_history(filename);
- }
+ ast_el_write_default_histfile();
if (consolethread == AST_PTHREADT_NULL || consolethread == pthread_self()) {
/* Only end if we are the consolethread, otherwise there's a race with that thread. */
if (el != NULL) {
@@ -2202,7 +2201,7 @@
}
}
-static int ast_all_zeros(char *s)
+static int ast_all_zeros(const char *s)
{
while (*s) {
if (*s > 32)
@@ -2213,7 +2212,7 @@
}
/* This is the main console CLI command handler. Run by the main() thread. */
-static void consolehandler(char *s)
+static void consolehandler(const char *s)
{
printf("%s", term_end());
fflush(stdout);
@@ -2231,7 +2230,7 @@
ast_cli_command(STDOUT_FILENO, s);
}
-static int remoteconsolehandler(char *s)
+static int remoteconsolehandler(const char *s)
{
int ret = 0;
@@ -3147,7 +3146,7 @@
#define MAX_HISTORY_COMMAND_LENGTH 256
-static int ast_el_add_history(char *buf)
+static int ast_el_add_history(const char *buf)
{
HistEvent ev;
char *stripped_buf;
@@ -3169,7 +3168,7 @@
return history(el_hist, &ev, H_ENTER, stripped_buf);
}
-static int ast_el_write_history(char *filename)
+static int ast_el_write_history(const char *filename)
{
HistEvent ev;
@@ -3179,7 +3178,7 @@
return (history(el_hist, &ev, H_SAVE, filename));
}
-static int ast_el_read_history(char *filename)
+static int ast_el_read_history(const char *filename)
{
HistEvent ev;
@@ -3190,11 +3189,30 @@
return history(el_hist, &ev, H_LOAD, filename);
}
+static void ast_el_read_default_histfile(void)
+{
+ char histfile[80] = "";
+ const char *home = getenv("HOME");
+ if (!ast_strlen_zero(home)) {
+ snprintf(histfile, sizeof(histfile), "%s/.asterisk_history", home);
+ ast_el_read_history(histfile);
+ }
+}
+
+static void ast_el_write_default_histfile(void)
+{
+ char histfile[80] = "";
+ const char *home = getenv("HOME");
+ if (!ast_strlen_zero(home)) {
+ snprintf(histfile, sizeof(histfile), "%s/.asterisk_history", home);
+ ast_el_write_history(histfile);
+ }
+}
+
static void ast_remotecontrol(char *data)
{
char buf[256] = "";
int res;
- char filename[80] = "";
char *hostname;
char *cpid;
char *version;
@@ -3203,6 +3221,10 @@
char *ebuf;
int num = 0;
+
+ ast_term_init();
+ printf("%s", term_end());
+ fflush(stdout);
memset(&sig_flags, 0, sizeof(sig_flags));
signal(SIGINT, __remote_quit_handler);
@@ -3302,15 +3324,11 @@
ast_verbose("Connected to Asterisk %s currently running on %s (pid = %d)\n", version, hostname, pid);
remotehostname = hostname;
- if (getenv("HOME"))
- snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
if (el_hist == NULL || el == NULL)
ast_el_initialize();
+ ast_el_read_default_histfile();
el_set(el, EL_GETCFN, ast_el_read_char);
-
- if (!ast_strlen_zero(filename))
- ast_el_read_history(filename);
for (;;) {
ebuf = (char *)el_gets(el, &num);
@@ -3780,17 +3798,10 @@
int main(int argc, char *argv[])
{
int c;
- char filename[80] = "";
- char hostname[MAXHOSTNAMELEN] = "";
char * xarg = NULL;
int x;
- FILE *f;
- sigset_t sigs;
- int num;
int isroot = 1, rundir_exists = 0;
- char *buf;
const char *runuser = NULL, *rungroup = NULL;
- int moduleresult; /*!< Result from the module load subsystem */
struct rlimit l;
static const char *getopt_settings = "BC:cde:FfG:ghIiL:M:mnpqRrs:TtU:VvWXx:";
@@ -3810,12 +3821,7 @@
if (argv[0] && (strstr(argv[0], "rasterisk")) != NULL) {
ast_set_flag(&ast_options, AST_OPT_FLAG_NO_FORK | AST_OPT_FLAG_REMOTE);
}
- if (gethostname(hostname, sizeof(hostname)-1))
- ast_copy_string(hostname, "<Unknown>", sizeof(hostname));
ast_mainpid = getpid();
-
- if (getenv("HOME"))
- snprintf(filename, sizeof(filename), "%s/.asterisk_history", getenv("HOME"));
/* Set config file to default before checking arguments for override. */
ast_copy_string(cfg_paths.config_file, DEFAULT_CONFIG_FILE, sizeof(cfg_paths.config_file));
@@ -4197,6 +4203,10 @@
quit_handler(0, SHUTDOWN_FAST, 0);
exit(0);
}
+ ast_term_init();
+ printf("%s", term_end());
+ fflush(stdout);
+
print_intro_message(runuser, rungroup);
printf("%s", term_quit());
ast_remotecontrol(NULL);
@@ -4212,6 +4222,19 @@
printf("%s", term_quit());
exit(1);
}
+
+ /* Not a remote console? Start the daemon. */
+ asterisk_daemon(isroot, runuser, rungroup);
+ return 0;
+}
+
+static void asterisk_daemon(int isroot, const char *runuser, const char *rungroup)
+{
+ FILE *f;
+ sigset_t sigs;
+ int num;
+ char *buf;
+ int moduleresult; /*!< Result from the module load subsystem */
/* This needs to remain as high up in the initial start up as possible.
* daemon causes a fork to occur, which has all sorts of unintended
@@ -4309,9 +4332,7 @@
if (ast_opt_console) {
if (el_hist == NULL || el == NULL)
ast_el_initialize();
-
- if (!ast_strlen_zero(filename))
- ast_el_read_history(filename);
+ ast_el_read_default_histfile();
}
ast_json_init();
@@ -4476,7 +4497,7 @@
/* initialize the data retrieval API */
if (ast_data_init()) {
- printf ("Failed: ast_data_init\n%s", term_quit());
+ printf("Failed: ast_data_init\n%s", term_quit());
exit(1);
}
@@ -4658,6 +4679,11 @@
/* Console stuff now... */
/* Register our quit function */
char title[256];
+ char hostname[MAXHOSTNAMELEN] = "";
+
+ if (gethostname(hostname, sizeof(hostname) - 1)) {
+ ast_copy_string(hostname, "<Unknown>", sizeof(hostname));
+ }
ast_pthread_create_detached(&mon_sig_flags, NULL, monitor_sig_flags, NULL);
@@ -4675,30 +4701,17 @@
buf = (char *) el_gets(el, &num);
if (!buf && write(1, "", 1) < 0)
- goto lostterm;
+ return; /* quit */
if (buf) {
if (buf[strlen(buf)-1] == '\n')
buf[strlen(buf)-1] = '\0';
- consolehandler((char *)buf);
- } else if (ast_opt_remote && (write(STDOUT_FILENO, "\nUse EXIT or QUIT to exit the asterisk console\n",
- strlen("\nUse EXIT or QUIT to exit the asterisk console\n")) < 0)) {
- /* Whoa, stdout disappeared from under us... Make /dev/null's */
- int fd;
- fd = open("/dev/null", O_RDWR);
- if (fd > -1) {
- dup2(fd, STDOUT_FILENO);
- dup2(fd, STDIN_FILENO);
- } else
- ast_log(LOG_WARNING, "Failed to open /dev/null to recover from dead console. Bad things will happen!\n");
- break;
+ consolehandler(buf);
}
}
}
+ /* Stall until a quit signal is given */
monitor_sig_flags(NULL);
-
-lostterm:
- return 0;
}
diff --git a/main/term.c b/main/term.c
index cf21719..11fbe2d 100644
--- a/main/term.c
+++ b/main/term.c
@@ -44,7 +44,6 @@
static int vt100compat;
-static char prepdata[80] = "";
static char enddata[80] = "";
static char quitdata[80] = "";
@@ -173,18 +172,13 @@
if (vt100compat) {
/* Make commands show up in nice colors */
if (ast_opt_light_background) {
- snprintf(prepdata, sizeof(prepdata), "%c[%dm", ESC, COLOR_BROWN);
snprintf(enddata, sizeof(enddata), "%c[%dm", ESC, COLOR_BLACK);
- snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
} else if (ast_opt_force_black_background) {
- snprintf(prepdata, sizeof(prepdata), "%c[%d;%d;%dm", ESC, ATTR_BRIGHT, COLOR_BROWN, COLOR_BLACK + 10);
snprintf(enddata, sizeof(enddata), "%c[%d;%d;%dm", ESC, ATTR_RESET, COLOR_WHITE, COLOR_BLACK + 10);
- snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
} else {
- snprintf(prepdata, sizeof(prepdata), "%c[%d;%dm", ESC, ATTR_BRIGHT, COLOR_BROWN);
- snprintf(enddata, sizeof(enddata), "%c[%d;%dm", ESC, ATTR_RESET, COLOR_WHITE);
- snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
+ snprintf(enddata, sizeof(enddata), "%c[%dm", ESC, ATTR_RESET);
}
+ snprintf(quitdata, sizeof(quitdata), "%c[%dm", ESC, ATTR_RESET);
}
return 0;
}
@@ -216,9 +210,12 @@
}
if (ast_opt_force_black_background) {
- snprintf(outbuf, maxout, "%c[%d;%d;%dm%s%c[%d;%dm", ESC, attr, fgcolor, bgcolor + 10, inbuf, ESC, COLOR_WHITE, COLOR_BLACK + 10);
+ if (!bgcolor) {
+ bgcolor = COLOR_BLACK;
+ }
+ snprintf(outbuf, maxout, "%c[%d;%d;%dm%s%s", ESC, attr, fgcolor, bgcolor + 10, inbuf, term_end());
} else {
- snprintf(outbuf, maxout, "%c[%d;%dm%s%c[0m", ESC, attr, fgcolor, inbuf, ESC);
+ snprintf(outbuf, maxout, "%c[%d;%dm%s%s", ESC, attr, fgcolor, inbuf, term_end());
}
return outbuf;
}
@@ -242,16 +239,16 @@
}
}
-static int check_colors_allowed(int fgcolor)
+static int check_colors_allowed(void)
{
- return (!vt100compat || !fgcolor) ? 0 : 1;
+ return vt100compat;
}
int ast_term_color_code(struct ast_str **str, int fgcolor, int bgcolor)
{
int attr = 0;
- if (!check_colors_allowed(fgcolor)) {
+ if (!check_colors_allowed()) {
return -1;
}
@@ -273,7 +270,7 @@
{
int attr = 0;
- if (!check_colors_allowed(fgcolor)) {
+ if (!check_colors_allowed()) {
*outbuf = '\0';
return outbuf;
}
@@ -310,11 +307,7 @@
const char *ast_term_reset(void)
{
- if (ast_opt_force_black_background) {
- return enddata;
- } else {
- return quitdata;
- }
+ return term_end();
}
char *term_strip(char *outbuf, const char *inbuf, int maxout)
@@ -335,34 +328,6 @@
if (! *inbuf_ptr)
break;
inbuf_ptr++;
- }
- return outbuf;
-}
-
-char *term_prompt(char *outbuf, const char *inbuf, int maxout)
-{
- if (!vt100compat) {
- ast_copy_string(outbuf, inbuf, maxout);
- return outbuf;
- }
- if (ast_opt_force_black_background) {
- snprintf(outbuf, maxout, "%c[%d;%d;%dm%c%c[%d;%dm%s",
- ESC, ATTR_BRIGHT, COLOR_BLUE, COLOR_BLACK + 10,
- inbuf[0],
- ESC, COLOR_WHITE, COLOR_BLACK + 10,
- inbuf + 1);
- } else if (ast_opt_light_background) {
- snprintf(outbuf, maxout, "%c[%d;0m%c%c[0m%s",
- ESC, COLOR_BLUE,
- inbuf[0],
- ESC,
- inbuf + 1);
- } else {
- snprintf(outbuf, maxout, "%c[%d;%d;0m%c%c[0m%s",
- ESC, ATTR_BRIGHT, COLOR_BLUE,
- inbuf[0],
- ESC,
- inbuf + 1);
}
return outbuf;
}
@@ -388,11 +353,6 @@
/* replace ESC with a space */
line[i] = ' ';
}
-}
-
-const char *term_prep(void)
-{
- return prepdata;
}
const char *term_end(void)
--
To view, visit https://gerrit.asterisk.org/1714
To unsubscribe, visit https://gerrit.asterisk.org/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: Ib641a0964c59ef9fe6f59efa8ccb481a9580c52f
Gerrit-PatchSet: 1
Gerrit-Project: asterisk
Gerrit-Branch: master
Gerrit-Owner: Walter Doekes <walter+asterisk at wjd.nu>
More information about the asterisk-code-review
mailing list