projects/pwm: pwm.c annotate

annotate pwm.c @ 19:5c6155c8e9b6

Handle signals Handled signals are generally blocked and only unblocked when doing blocking I/O, i.e. either when reading commands or printing results. A (possibly queued) signal will then interrupt I/O and can be dealt with in the main loop.

author	Guido Berhoerster <guido+pwm@berhoerster.name>
date	Fri, 01 Sep 2017 22:33:41 +0200
parents	1e39a251cbe9
children	efef93e54c5f

rev	line source
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	1 /*
13 cf81eb0c2d5a Warn before quitting if there are unsaved changes Guido Berhoerster <guido+pwm@berhoerster.name> parents: 5 diff changeset	2 * Copyright (C) 2017 Guido Berhoerster <guido+pwm@berhoerster.name>
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	3 *
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	4 * Permission is hereby granted, free of charge, to any person obtaining
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	5 * a copy of this software and associated documentation files (the
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	6 * "Software"), to deal in the Software without restriction, including
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	7 * without limitation the rights to use, copy, modify, merge, publish,
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	8 * distribute, sublicense, and/or sell copies of the Software, and to
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	9 * permit persons to whom the Software is furnished to do so, subject to
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	10 * the following conditions:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	11 *
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	12 * The above copyright notice and this permission notice shall be included
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	13 * in all copies or substantial portions of the Software.
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	14 *
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	16 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	17 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	18 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	19 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	20 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	21 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	22 */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	23
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	24 #include "compat.h"
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	25
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	26 #ifdef HAVE_ERR_H
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	27 #include <err.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	28 #endif /* HAVE_ERR_H */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	29 #include <errno.h>
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	30 #include <fcntl.h>
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	31 #include <langinfo.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	32 #include <locale.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	33 #include <pwd.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	34 #include <stdlib.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	35 #include <stdio.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	36 #include <string.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	37 #include <pwd.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	38 #include <sys/stat.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	39 #include <time.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	40 #include <unistd.h>
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	41
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	42 #include "pwm.h"
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	43 #include "cmd.h"
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	44 #include "io.h"
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	45 #include "pwfile.h"
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	46 #include "tok.h"
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	47 #include "util.h"
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	48
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	49 static void
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	50 usage(void)
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	51 {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	52 fprintf(stderr, "usage: %s [-P file] [filename]\n", getprogname());
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	53 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	54
16 a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	55 void
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	56 pwm_err(struct pwm_ctx ctx, char fmt, ...)
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	57 {
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	58 va_list args;
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	59
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	60 free(ctx->errmsg);
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	61
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	62 if (fmt != NULL) {
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	63 va_start(args, fmt);
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	64 xvasprintf(&ctx->errmsg, fmt, args);
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	65 va_end(args);
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	66
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	67 fprintf(stderr, "%s\n", ctx->errmsg);
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	68 } else {
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	69 ctx->errmsg = NULL;
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	70 }
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	71 }
a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	72
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	73 void
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	74 pwm_block_signals(void)
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	75 {
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	76 sigset_t set;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	77
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	78 sigemptyset(&set);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	79 sigaddset(&set, SIGINT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	80 sigaddset(&set, SIGTERM);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	81 sigaddset(&set, SIGHUP);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	82 sigaddset(&set, SIGQUIT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	83 if (sigprocmask(SIG_BLOCK, &set, NULL) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	84 err(1, "sigprocmask");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	85 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	86 }
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	87
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	88 void
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	89 pwm_unblock_signals(void)
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	90 {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	91 sigset_t set;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	92
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	93 sigemptyset(&set);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	94 sigaddset(&set, SIGINT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	95 sigaddset(&set, SIGTERM);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	96 sigaddset(&set, SIGHUP);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	97 sigaddset(&set, SIGQUIT);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	98 if (sigprocmask(SIG_UNBLOCK, &set, NULL) != 0) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	99 err(1, "sigprocmask");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	100 }
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	101 }
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	102
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	103 static int
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	104 run_input_loop(struct pwm_ctx *ctx, int is_interactive)
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	105 {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	106 int retval = -1;
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	107 char prompt[8 + 2 + 1];
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	108 GetLine *gl = NULL;
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	109 char buf[PWM_LINE_MAX + 1];
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	110 int io_retval;
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	111 int argc = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	112 char **argv = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	113 struct cmd *cmd;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	114 int i;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	115
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	116 snprintf(prompt, sizeof (prompt), "%.*s> ", 8, getprogname());
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	117
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	118 pwm_block_signals();
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	119
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	120 /* initialize libtecla */
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	121 gl = new_GetLine(PWM_LINE_MAX, PWM_HISTORY_MAX);
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	122 if (gl == NULL) {
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	123 err(1, "new_GetLine");
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	124 }
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	125 gl_catch_blocked(gl);
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	126 gl_limit_history(gl, PWM_HISTORY_LINES_MAX);
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	127 /* disable default filename completion */
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	128 gl_customize_completion(gl, NULL, io_gl_complete_nothing);
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	129
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	130 for (;;) {
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	131 /* read next line */
13 cf81eb0c2d5a Warn before quitting if there are unsaved changes Guido Berhoerster <guido+pwm@berhoerster.name> parents: 5 diff changeset	132 cmd = NULL;
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	133 buf[0] = '\0';
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	134 io_retval = io_get_line(gl, prompt, 1, NULL, 0,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	135 sizeof (buf), buf);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	136 switch (io_retval) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	137 case IO_OK:
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	138 break;
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	139 case IO_TRUNCATED:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	140 /* line was truncated in non-interactive mode */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	141 fprintf(stderr, "line too long\n");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	142 goto out;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	143 case IO_EOF: /* FALLTHROUGH */
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	144 case IO_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	145 goto quit;
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	146 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	147 fprintf(stderr, "unknown error\n");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	148 goto quit;
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	149 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	150
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	151 /* tokenize line */
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	152 switch (tok_tokenize(buf, &argc, &argv)) {
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	153 case TOK_ERR_SYSTEM_ERROR:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	154 err(1, "tok_tokenize");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	155 case TOK_ERR_UNTERMINATED_QUOTE:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	156 fprintf(stderr, "unterminated quote\n");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	157 if (!is_interactive) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	158 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	159 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	160 goto next;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	161 case TOK_ERR_TRAILING_BACKSLASH:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	162 fprintf(stderr, "trailing backslash\n");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	163 if (!is_interactive) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	164 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	165 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	166 goto next;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	167 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	168
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	169 if (argc == 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	170 /* empty line */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	171 goto next;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	172 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	173
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	174 /* find and execute the command */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	175 cmd = cmd_match(argv[0]);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	176 if (cmd == NULL) {
16 a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	177 pwm_err(ctx, "unknown command: %s", argv[0]);
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	178 if (is_interactive) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	179 goto next;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	180 } else {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	181 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	182 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	183 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	184 switch (cmd->cmd_func(ctx, argc, argv)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	185 case CMD_OK:
16 a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	186 pwm_err(ctx, NULL); /* clear error */
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	187 break;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	188 case CMD_USAGE:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	189 fprintf(stderr, "usage: %s\n", cmd->usage);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	190 case CMD_ERR: /* FALLTHROUGH */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	191 if (!is_interactive) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	192 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	193 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	194 break;
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	195 case CMD_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	196 fprintf(stderr, "received signal, quitting\n");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	197 case CMD_QUIT: /* FALLTHROUGH */
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	198 goto quit;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	199 }
16 a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	200 ctx->prev_cmd = cmd->full_cmd;
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	201
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	202 next:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	203 for (i = 0; i < argc; i++) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	204 free(argv[i]);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	205 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	206 free(argv);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	207 argc = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	208 argv = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	209 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	210
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	211 quit:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	212 retval = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	213
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	214 out:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	215 for (i = 0; i < argc; i++) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	216 free(argv[i]);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	217 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	218 free(argv);
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	219 del_GetLine(gl);
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	220
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	221 return (retval);
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	222 }
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	223
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	224 int
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	225 pwm_read_password(struct pwm_ctx *ctx, int is_new_password)
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	226 {
19 5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	227 switch (io_get_password(is_new_password ? "New Password:" :
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	228 "Password:", is_new_password ? "Confirm Password:" : NULL,
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	229 sizeof (ctx->password), ctx->password)) {
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	230 case IO_OK:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	231 return (0);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	232 case IO_SIGNAL:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	233 return (-2);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	234 case IO_PASSWORD_EMPTY:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	235 pwm_err(ctx, "password must not be empty");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	236 return (-1);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	237 case IO_PASSWORD_MISMATCH:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	238 pwm_err(ctx, "passwords do not match");
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	239 return (-1);
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	240 default:
5c6155c8e9b6 Handle signals Guido Berhoerster <guido+pwm@berhoerster.name> parents: 18 diff changeset	241 return (-1);
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	242 }
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	243 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	244
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	245 static int
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	246 read_password_from_file(const char filename, char password,
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	247 size_t password_size)
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	248 {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	249 int retval = -1;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	250 char *buf = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	251 FILE *fp = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	252 size_t password_len = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	253
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	254 buf = xmalloc(password_size);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	255
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	256 fp = fopen(filename, "r");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	257 if (fp == NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	258 warn("failed to open master password file \"%s\"", filename);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	259 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	260 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	261 if (fgets(buf, password_size, fp) == NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	262 /* read error or empty file */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	263 if (ferror(fp)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	264 /* read error */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	265 warn("failed to read master password from \"%s\"",
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	266 filename);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	267 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	268 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	269 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	270 password_len = strlen(buf);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	271 if (buf[password_len - 1] == '\n') {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	272 /* strip trailing newline */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	273 password_len--;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	274 if (password_len == 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	275 /* first line is empty */
5 0b1bce8db371 Prevent misleading double error message Guido Berhoerster <guido+pwm@berhoerster.name> parents: 0 diff changeset	276 fprintf(stderr, "malformed password file\n");
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	277 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	278 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	279 } else if (!feof(fp)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	280 /* the first line was truncated, password is too long */
5 0b1bce8db371 Prevent misleading double error message Guido Berhoerster <guido+pwm@berhoerster.name> parents: 0 diff changeset	281 fprintf(stderr, "malformed password file\n");
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	282 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	283 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	284 memcpy(password, buf, password_size);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	285 retval = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	286
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	287 out:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	288 password[password_len] = '\0';
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	289
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	290 if (fp != NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	291 fclose(fp);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	292 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	293 free(buf);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	294
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	295 return (retval);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	296 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	297
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	298 int
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	299 main(int argc, char *argv[])
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	300 {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	301 int status = EXIT_FAILURE;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	302 char *locale;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	303 int is_interactive;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	304 int errflag = 0;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	305 int c;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	306 const char *master_password_filename = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	307 struct pwm_ctx ctx = { 0 };
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	308 struct passwd *passwd;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	309 char *pwm_dirname = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	310 FILE *fp = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	311
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	312 setprogname(argv[0]);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	313
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	314 /* set up locale and check for UTF-8 codeset */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	315 locale = setlocale(LC_ALL, "");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	316 if (locale == NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	317 errx(1, "invalid locale");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	318 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	319 if ((strcasecmp(nl_langinfo(CODESET), "UTF-8") != 0) &&
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	320 (strcasecmp(nl_langinfo(CODESET), "UTF8") != 0)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	321 fprintf(stderr, "pwm requires a locale with UTF-8 character "
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	322 "encoding.\n");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	323 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	324 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	325
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	326 /* timestamps are processed as UTC */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	327 if (setenv("TZ", "", 1) != 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	328 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	329 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	330 tzset();
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	331
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	332 /* initialize libpws */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	333 if (pws_init() != 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	334 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	335 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	336
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	337 is_interactive = isatty(STDIN_FILENO);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	338
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	339 while (!errflag && (c = getopt(argc, argv, "P:h")) != -1) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	340 switch (c) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	341 case 'P':
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	342 master_password_filename = optarg;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	343 break;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	344 case 'h':
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	345 usage();
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	346 status = EXIT_SUCCESS;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	347 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	348 default:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	349 errflag = 1;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	350 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	351 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	352 if (errflag \|\| (optind + 1 < argc)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	353 usage();
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	354 status = EXIT_USAGE;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	355 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	356 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	357
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	358 if (optind == argc) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	359 passwd = getpwuid(getuid());
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	360 if (passwd == NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	361 err(1, "getpwuid");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	362 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	363 xasprintf(&pwm_dirname, "%s/.pwm", passwd->pw_dir);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	364 xasprintf(&ctx.filename, "%s/pwm.psafe3", pwm_dirname);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	365
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	366 /* create ~/.pwm directory if necessary */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	367 if ((mkdir(pwm_dirname, S_IRWXU) != 0) && (errno != EEXIST)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	368 warn("failed to create directory \"%s\"", pwm_dirname);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	369 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	370 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	371 } else if (optind + 1 == argc) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	372 ctx.filename = xstrdup(argv[optind++]);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	373 } else {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	374 usage();
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	375 status = EXIT_USAGE;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	376 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	377 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	378
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	379 if (is_interactive) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	380 printf("pwm version %s\n", VERSION);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	381 } else if (master_password_filename == NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	382 fprintf(stderr, "master password file must be specified when "
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	383 "running non-interactively\n");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	384 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	385 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	386
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	387 pwfile_init(&ctx);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	388
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	389 fp = fopen(ctx.filename, "r");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	390 if ((fp == NULL) && (errno != ENOENT)) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	391 warn("failed to open password database");
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	392 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	393 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	394 /* obtain master password */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	395 if (master_password_filename != NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	396 if (read_password_from_file(master_password_filename,
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	397 ctx.password, sizeof (ctx.password)) != 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	398 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	399 }
18 1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	400 } else if (pwm_read_password(&ctx, (fp == NULL)) != 0) {
1e39a251cbe9 Use libtecla for interactive input Guido Berhoerster <guido+pwm@berhoerster.name> parents: 16 diff changeset	401 goto out;
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	402 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	403 if (fp != NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	404 if (pwfile_read_file(&ctx, fp) != 0) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	405 goto out;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	406 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	407 fclose(fp);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	408 fp = NULL;
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	409 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	410
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	411 /* run main input loop */
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	412 status = (run_input_loop(&ctx, is_interactive) != 0);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	413
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	414 out:
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	415 pwfile_destroy(&ctx);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	416 if (fp != NULL) {
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	417 fclose(fp);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	418 }
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	419 free(ctx.filename);
16 a07665727c19 Add status command Guido Berhoerster <guido+pwm@berhoerster.name> parents: 13 diff changeset	420 free(ctx.errmsg);
0 a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	421 free(pwm_dirname);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	422
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	423 exit(status);
a7e41e1a79c8 Initial revision Guido Berhoerster <guido+pwm@berhoerster.name> parents: diff changeset	424 }

Mercurial > projects > pwm

annotate pwm.c @ 19:5c6155c8e9b6