xref: /freebsd/usr.bin/factor/factor.c (revision eb69d1f144a6fcc765d1b9d44a5ae8082353e70b)
1 /*
2  * Copyright (c) 1989, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Landon Curt Noll.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the name of the University nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  */
32 
33 #ifndef lint
34 #include <sys/cdefs.h>
35 #ifdef __COPYRIGHT
36 __COPYRIGHT("@(#) Copyright (c) 1989, 1993\
37 	The Regents of the University of California.  All rights reserved.");
38 #endif
39 #ifdef __SCCSID
40 __SCCSID("@(#)factor.c	8.4 (Berkeley) 5/4/95");
41 #endif
42 #ifdef __RCSID
43 __RCSID("$NetBSD: factor.c,v 1.19 2009/08/12 05:54:31 dholland Exp $");
44 #endif
45 #ifdef __FBSDID
46 __FBSDID("$FreeBSD$");
47 #endif
48 #endif /* not lint */
49 
50 /*
51  * factor - factor a number into primes
52  *
53  * By: Landon Curt Noll   chongo@toad.com,   ...!{sun,tolsoft}!hoptoad!chongo
54  *
55  *   chongo <for a good prime call: 391581 * 2^216193 - 1> /\oo/\
56  *
57  * usage:
58  *	factor [-h] [number] ...
59  *
60  * The form of the output is:
61  *
62  *	number: factor1 factor1 factor2 factor3 factor3 factor3 ...
63  *
64  * where factor1 <= factor2 <= factor3 <= ...
65  *
66  * If no args are given, the list of numbers are read from stdin.
67  */
68 
69 #include <ctype.h>
70 #include <err.h>
71 #include <errno.h>
72 #include <inttypes.h>
73 #include <limits.h>
74 #include <stdio.h>
75 #include <stdlib.h>
76 #include <unistd.h>
77 
78 #include "primes.h"
79 
80 #ifdef HAVE_OPENSSL
81 
82 #include <openssl/bn.h>
83 
84 #define	PRIME_CHECKS	5
85 
86 static void	pollard_pminus1(BIGNUM *); /* print factors for big numbers */
87 
88 #else
89 
90 typedef ubig	BIGNUM;
91 typedef u_long	BN_ULONG;
92 
93 #define BN_CTX			int
94 #define BN_CTX_new()		NULL
95 #define BN_new()		((BIGNUM *)calloc(sizeof(BIGNUM), 1))
96 #define BN_is_zero(v)		(*(v) == 0)
97 #define BN_is_one(v)		(*(v) == 1)
98 #define BN_mod_word(a, b)	(*(a) % (b))
99 
100 static int	BN_dec2bn(BIGNUM **a, const char *str);
101 static int	BN_hex2bn(BIGNUM **a, const char *str);
102 static BN_ULONG BN_div_word(BIGNUM *, BN_ULONG);
103 static void	BN_print_fp(FILE *, const BIGNUM *);
104 
105 #endif
106 
107 static void	BN_print_dec_fp(FILE *, const BIGNUM *);
108 
109 static void	pr_fact(BIGNUM *);	/* print factors of a value */
110 static void	pr_print(BIGNUM *);	/* print a prime */
111 static void	usage(void);
112 
113 static BN_CTX	*ctx;			/* just use a global context */
114 static int	hflag;
115 
116 int
117 main(int argc, char *argv[])
118 {
119 	BIGNUM *val;
120 	int ch;
121 	char *p, buf[LINE_MAX];		/* > max number of digits. */
122 
123 	ctx = BN_CTX_new();
124 	val = BN_new();
125 	if (val == NULL)
126 		errx(1, "can't initialise bignum");
127 
128 	while ((ch = getopt(argc, argv, "h")) != -1)
129 		switch (ch) {
130 		case 'h':
131 			hflag++;
132 			break;
133 		case '?':
134 		default:
135 			usage();
136 		}
137 	argc -= optind;
138 	argv += optind;
139 
140 	/* No args supplied, read numbers from stdin. */
141 	if (argc == 0)
142 		for (;;) {
143 			if (fgets(buf, sizeof(buf), stdin) == NULL) {
144 				if (ferror(stdin))
145 					err(1, "stdin");
146 				exit (0);
147 			}
148 			for (p = buf; isblank(*p); ++p);
149 			if (*p == '\n' || *p == '\0')
150 				continue;
151 			if (*p == '-')
152 				errx(1, "negative numbers aren't permitted.");
153 			if (BN_dec2bn(&val, buf) == 0 &&
154 			    BN_hex2bn(&val, buf) == 0)
155 				errx(1, "%s: illegal numeric format.", buf);
156 			pr_fact(val);
157 		}
158 	/* Factor the arguments. */
159 	else
160 		for (; *argv != NULL; ++argv) {
161 			if (argv[0][0] == '-')
162 				errx(1, "negative numbers aren't permitted.");
163 			if (BN_dec2bn(&val, argv[0]) == 0 &&
164 			    BN_hex2bn(&val, argv[0]) == 0)
165 				errx(1, "%s: illegal numeric format.", argv[0]);
166 			pr_fact(val);
167 		}
168 	exit(0);
169 }
170 
171 /*
172  * pr_fact - print the factors of a number
173  *
174  * Print the factors of the number, from the lowest to the highest.
175  * A factor will be printed multiple times if it divides the value
176  * multiple times.
177  *
178  * Factors are printed with leading tabs.
179  */
180 static void
181 pr_fact(BIGNUM *val)
182 {
183 	const ubig *fact;	/* The factor found. */
184 
185 	/* Firewall - catch 0 and 1. */
186 	if (BN_is_zero(val))	/* Historical practice; 0 just exits. */
187 		exit(0);
188 	if (BN_is_one(val)) {
189 		printf("1: 1\n");
190 		return;
191 	}
192 
193 	/* Factor value. */
194 
195 	if (hflag) {
196 		fputs("0x", stdout);
197 		BN_print_fp(stdout, val);
198 	} else
199 		BN_print_dec_fp(stdout, val);
200 	putchar(':');
201 	for (fact = &prime[0]; !BN_is_one(val); ++fact) {
202 		/* Look for the smallest factor. */
203 		do {
204 			if (BN_mod_word(val, (BN_ULONG)*fact) == 0)
205 				break;
206 		} while (++fact <= pr_limit);
207 
208 		/* Watch for primes larger than the table. */
209 		if (fact > pr_limit) {
210 #ifdef HAVE_OPENSSL
211 			BIGNUM *bnfact;
212 
213 			bnfact = BN_new();
214 			BN_set_word(bnfact, *(fact - 1));
215 			if (!BN_sqr(bnfact, bnfact, ctx))
216 				errx(1, "error in BN_sqr()");
217 			if (BN_cmp(bnfact, val) > 0 ||
218 			    BN_is_prime(val, PRIME_CHECKS,
219 					NULL, NULL, NULL) == 1)
220 				pr_print(val);
221 			else
222 				pollard_pminus1(val);
223 #else
224 			pr_print(val);
225 #endif
226 			break;
227 		}
228 
229 		/* Divide factor out until none are left. */
230 		do {
231 			printf(hflag ? " 0x%" PRIx64 "" : " %" PRIu64 "", *fact);
232 			BN_div_word(val, (BN_ULONG)*fact);
233 		} while (BN_mod_word(val, (BN_ULONG)*fact) == 0);
234 
235 		/* Let the user know we're doing something. */
236 		fflush(stdout);
237 	}
238 	putchar('\n');
239 }
240 
241 static void
242 pr_print(BIGNUM *val)
243 {
244 	if (hflag) {
245 		fputs(" 0x", stdout);
246 		BN_print_fp(stdout, val);
247 	} else {
248 		putchar(' ');
249 		BN_print_dec_fp(stdout, val);
250 	}
251 }
252 
253 static void
254 usage(void)
255 {
256 	fprintf(stderr, "usage: factor [-h] [value ...]\n");
257 	exit(1);
258 }
259 
260 #ifdef HAVE_OPENSSL
261 
262 /* pollard p-1, algorithm from Jim Gillogly, May 2000 */
263 static void
264 pollard_pminus1(BIGNUM *val)
265 {
266 	BIGNUM *base, *rbase, *num, *i, *x;
267 
268 	base = BN_new();
269 	rbase = BN_new();
270 	num = BN_new();
271 	i = BN_new();
272 	x = BN_new();
273 
274 	BN_set_word(rbase, 1);
275 newbase:
276 	if (!BN_add_word(rbase, 1))
277 		errx(1, "error in BN_add_word()");
278 	BN_set_word(i, 2);
279 	BN_copy(base, rbase);
280 
281 	for (;;) {
282 		BN_mod_exp(base, base, i, val, ctx);
283 		if (BN_is_one(base))
284 			goto newbase;
285 
286 		BN_copy(x, base);
287 		BN_sub_word(x, 1);
288 		if (!BN_gcd(x, x, val, ctx))
289 			errx(1, "error in BN_gcd()");
290 
291 		if (!BN_is_one(x)) {
292 			if (BN_is_prime(x, PRIME_CHECKS, NULL, NULL,
293 			    NULL) == 1)
294 				pr_print(x);
295 			else
296 				pollard_pminus1(x);
297 			fflush(stdout);
298 
299 			BN_div(num, NULL, val, x, ctx);
300 			if (BN_is_one(num))
301 				return;
302 			if (BN_is_prime(num, PRIME_CHECKS, NULL, NULL,
303 			    NULL) == 1) {
304 				pr_print(num);
305 				fflush(stdout);
306 				return;
307 			}
308 			BN_copy(val, num);
309 		}
310 		if (!BN_add_word(i, 1))
311 			errx(1, "error in BN_add_word()");
312 	}
313 }
314 
315 /*
316  * Sigh..  No _decimal_ output to file functions in BN.
317  */
318 static void
319 BN_print_dec_fp(FILE *fp, const BIGNUM *num)
320 {
321 	char *buf;
322 
323 	buf = BN_bn2dec(num);
324 	if (buf == NULL)
325 		return;	/* XXX do anything here? */
326 	fprintf(fp, "%s", buf);
327 	free(buf);
328 }
329 
330 #else
331 
332 static void
333 BN_print_fp(FILE *fp, const BIGNUM *num)
334 {
335 	fprintf(fp, "%lx", (unsigned long)*num);
336 }
337 
338 static void
339 BN_print_dec_fp(FILE *fp, const BIGNUM *num)
340 {
341 	fprintf(fp, "%lu", (unsigned long)*num);
342 }
343 
344 static int
345 BN_dec2bn(BIGNUM **a, const char *str)
346 {
347 	char *p;
348 
349 	errno = 0;
350 	**a = strtoul(str, &p, 10);
351 	return (errno == 0 && (*p == '\n' || *p == '\0'));
352 }
353 
354 static int
355 BN_hex2bn(BIGNUM **a, const char *str)
356 {
357 	char *p;
358 
359 	errno = 0;
360 	**a = strtoul(str, &p, 16);
361 	return (errno == 0 && (*p == '\n' || *p == '\0'));
362 }
363 
364 static BN_ULONG
365 BN_div_word(BIGNUM *a, BN_ULONG b)
366 {
367 	BN_ULONG mod;
368 
369 	mod = *a % b;
370 	*a /= b;
371 	return mod;
372 }
373 
374 #endif
375