xref: /netbsd-src/sys/kern/cnmagic.c (revision df97a425935fb635a7c65b50a215459ca89e39db)
1 /*	$NetBSD: cnmagic.c,v 1.14 2017/05/04 11:01:16 kamil Exp $	*/
2 
3 /*
4  * Copyright (c) 2000 Eduardo Horvath
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #include <sys/cdefs.h>
29 __KERNEL_RCSID(0, "$NetBSD: cnmagic.c,v 1.14 2017/05/04 11:01:16 kamil Exp $");
30 
31 #include <sys/param.h>
32 #include <sys/systm.h>
33 #include <sys/kernel.h>
34 
35 #define ENCODE_STATE(c, n) (short)(((c)&0x1ff)|(((n)&0x7f)<<9))
36 
37 static unsigned short cn_magic[CNS_LEN];
38 
39 /*
40  * Initialize a cnm_state_t.
41  */
42 void
cn_init_magic(cnm_state_t * cnm)43 cn_init_magic(cnm_state_t *cnm)
44 {
45 	cnm->cnm_state = 0;
46 	cnm->cnm_magic = cn_magic;
47 }
48 
49 /*
50  * Destroy a cnm_state_t.
51  */
52 void
cn_destroy_magic(cnm_state_t * cnm)53 cn_destroy_magic(cnm_state_t *cnm)
54 {
55 	cnm->cnm_state = 0;
56 	cnm->cnm_magic = NULL;
57 }
58 
59 /*
60  * Translate a magic string to a state
61  * machine table.
62  */
63 int
cn_set_magic(const char * smagic)64 cn_set_magic(const char *smagic)
65 {
66 	const unsigned char *magic = (const unsigned char *)smagic;
67 	unsigned short i, c, n;
68 	unsigned short m[CNS_LEN];
69 
70 	for (i = 0; i < CNS_LEN; i++) {
71 		c = *magic++;
72 		switch (c) {
73 		case 0:
74 			/* End of string */
75 			if (i == 0) {
76 				/* empty string? */
77 #ifdef DEBUG
78 				printf("cn_set_magic(): empty!\n");
79 #endif
80 			}
81 			cn_magic[i] = 0;
82 			while (i--)
83 				cn_magic[i] = m[i];
84 			return 0;
85 		case 0x27:
86 			/* Escape sequence */
87 			c = *magic++;
88 			switch (c) {
89 			case 0x27:
90 				break;
91 			case 0x01:
92 				/* BREAK */
93 				c = CNC_BREAK;
94 				break;
95 			case 0x02:
96 				/* NUL */
97 				c = 0;
98 				break;
99 			}
100 			/* FALLTHROUGH */
101 		default:
102 			/* Transition to the next state. */
103 			n = *magic ? i + 1 : CNS_TERM;
104 #ifdef DEBUG
105 			if (!cold)
106 				aprint_normal("mag %d %x:%x\n", i, c, n);
107 #endif
108 			m[i] = ENCODE_STATE(c, n);
109 			break;
110 		}
111 	}
112 	return EINVAL;
113 }
114 
115 /*
116  * Translate a state machine table back to
117  * a magic string.
118  */
119 int
cn_get_magic(char * magic,size_t maglen)120 cn_get_magic(char *magic, size_t maglen)
121 {
122 	size_t i, n = 0;
123 
124 #define ADD_CHAR(x) \
125 do \
126 	if (n < maglen) \
127 		magic[n++] = (x); \
128 	else \
129 		goto error; \
130 while (/*CONSTCOND*/0)
131 
132 	for (i = 0; i < CNS_LEN; /* empty */) {
133 		unsigned short c = cn_magic[i];
134 		i = CNS_MAGIC_NEXT(c);
135 		if (i == 0)
136 			goto finish;
137 
138 		/* Translate a character */
139 		switch (CNS_MAGIC_VAL(c)) {
140 		case CNC_BREAK:
141 			ADD_CHAR(0x27);
142 			ADD_CHAR(0x01);
143 			break;
144 		case 0:
145 			ADD_CHAR(0x27);
146 			ADD_CHAR(0x02);
147 			break;
148 		case 0x27:
149 			ADD_CHAR(0x27);
150 			ADD_CHAR(0x27);
151 			break;
152 		default:
153 			ADD_CHAR(c);
154 			break;
155 		}
156 		/* Now go to the next state */
157 		if (i == CNS_TERM)
158 			goto finish;
159 	}
160 
161 error:
162 	return EINVAL;
163 
164 finish:
165 	/* Either termination state or empty machine */
166 	ADD_CHAR('\0');
167 	return 0;
168 }
169