Statistics
| Branch: | Revision:

root / prex-0.9.0 / usr / sbin / lock / lock.c @ 03e9c04a

History | View | Annotate | Download (3.61 KB)

1
/*
2
 * Copyright (c) 2009, Kohsuke Ohtani
3
 * All rights reserved.
4
 *
5
 * Redistribution and use in source and binary forms, with or without
6
 * modification, are permitted provided that the following conditions
7
 * are met:
8
 * 1. Redistributions of source code must retain the above copyright
9
 *    notice, this list of conditions and the following disclaimer.
10
 * 2. Redistributions in binary form must reproduce the above copyright
11
 *    notice, this list of conditions and the following disclaimer in the
12
 *    documentation and/or other materials provided with the distribution.
13
 * 3. Neither the name of the author nor the names of any co-contibutors
14
 *    may be used to endorse or promote products derived from this software
15
 *    without specific prior written permission.
16
 *
17
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27
 * SUCH DAMAGE.
28
 */
29

    
30
/*
31
 * lock.c - terminal lock utility.
32
 *
33
 * Required capabilities:
34
 *      CAP_USRFILES
35
 */
36

    
37
#include <sys/stat.h>
38
#include <sys/fcntl.h>
39

    
40
#include <unistd.h>
41
#include <ctype.h>
42
#include <err.h>
43
#include <string.h>
44
#include <stdlib.h>
45
#include <errno.h>
46
#include <stdio.h>
47
#include <paths.h>
48

    
49
static void usage(void);
50
static void setpass(void);
51

    
52
int
53
main(int argc, char *argv[])
54
{
55
        char passcode[PASSWORD_LEN + 1];
56
        FILE *fp;
57
        int update = 0;
58
        char *p;
59

    
60
        if (argc == 2) {
61
                if (!strcmp(argv[1], "-u"))
62
                        update = 1;
63
                else
64
                        usage();
65
        }
66

    
67
        /*
68
         * Load current passcode.
69
         */
70
        if ((fp = fopen(_PATH_PASSWD, "r")) == NULL) {
71
                printf("Passcode is not set.\n");
72
                setpass();
73
        }
74
        if (((p = fgets(passcode, sizeof(passcode), fp)) == NULL) ||
75
            (strlen(passcode) != 4)) {
76
                printf("Invalid passcode is set.\n");
77
                fclose(fp);
78
                setpass();
79
        }
80
        fclose(fp);
81

    
82
        if (update) {
83
                /*
84
                 * Update current passcode.
85
                 */
86
                if (strcmp(getpass("Old passcode:"), passcode)) {
87
                        printf("Mismatch.\n");
88
                        exit(0);
89
                }
90
                setpass();
91
        }
92

    
93
        /*
94
         * Lock keyboard until correct passcode input.
95
         */
96
        signal(SIGQUIT, SIG_IGN);
97
        signal(SIGINT, SIG_IGN);
98
        signal(SIGTSTP, SIG_IGN);
99

    
100
        printf("\33[2J");        /* clear screen */
101
        printf("Device is locked.\n");
102
        for (;;) {
103
                if (!strcmp(getpass("Enter passcode:"), passcode))
104
                        break;
105
        }
106
        exit(0);
107
}
108

    
109
static void
110
usage(void)
111
{
112
        fputs("usage: lock [-u]\n", stderr);
113
        exit(1);
114
        /* NOTREACHED */
115
}
116

    
117
/*
118
 * We don't need to encrypt passcode file because it is stored
119
 * secure area in the file system.
120
 */
121
static void
122
setpass(void)
123
{
124
        FILE *fp;
125
        char *p, *t;
126
        char buf[PASSWORD_LEN + 1];
127

    
128
        for (buf[0] = '\0';;) {
129
                p = getpass("New passcode:");
130
                if (!*p) {
131
                        printf("Passcode unchanged.\n");
132
                        exit(0);
133
                }
134
                for (t = p; *t && isdigit(*t); ++t);
135
                if (strlen(p) != 4 || *t) {
136
                        printf("Please enter 4 digit number for passcode.\n");
137
                        continue;
138
                }
139
                strlcpy(buf, p, sizeof(buf));
140
                if (!strcmp(buf, getpass("Retype new passcode:")))
141
                        break;
142
                printf("Mismatch; try again, EOF to quit.\n");
143
        }
144
        if ((fp = fopen(_PATH_PASSWD, "w+")) == NULL) {
145
                err(1, NULL);
146
                exit(0);
147
        }
148
        fputs(buf, fp);
149
        fclose(fp);
150
        exit(0);
151
}