xref: /dflybsd-src/libexec/ftpd/site_md5.c (revision bf2127bebce06a2326892fc41a4017f52fd3a0c8)
1*bf2127beSzrj /*
2*bf2127beSzrj  * Copyright (c) 2019 The DragonFly Project.  All rights reserved.
3*bf2127beSzrj  *
4*bf2127beSzrj  * Redistribution and use in source and binary forms, with or without
5*bf2127beSzrj  * modification, are permitted provided that the following conditions
6*bf2127beSzrj  * are met:
7*bf2127beSzrj  * 1. Redistributions of source code must retain the above copyright
8*bf2127beSzrj  *    notice, this list of conditions and the following disclaimer.
9*bf2127beSzrj  * 2. Redistributions in binary form must reproduce the above copyright
10*bf2127beSzrj  *    notice, this list of conditions and the following disclaimer in the
11*bf2127beSzrj  *    documentation and/or other materials provided with the distribution.
12*bf2127beSzrj  *
13*bf2127beSzrj  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
14*bf2127beSzrj  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
15*bf2127beSzrj  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
16*bf2127beSzrj  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
17*bf2127beSzrj  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
18*bf2127beSzrj  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
19*bf2127beSzrj  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
20*bf2127beSzrj  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
21*bf2127beSzrj  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
22*bf2127beSzrj  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
23*bf2127beSzrj  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*bf2127beSzrj  * SUCH DAMAGE.
25*bf2127beSzrj  */
26*bf2127beSzrj 
27*bf2127beSzrj #include <sys/types.h>
28*bf2127beSzrj #include <sys/stat.h>
29*bf2127beSzrj #include <errno.h>
30*bf2127beSzrj #include <fcntl.h>
31*bf2127beSzrj #include <unistd.h>
32*bf2127beSzrj #include <openssl/md5.h>
33*bf2127beSzrj 
34*bf2127beSzrj #include <stdio.h>		/* for FILE in extern.h */
35*bf2127beSzrj #include <netinet/in.h>		/* for struct sockaddr_in */
36*bf2127beSzrj #include "extern.h"
37*bf2127beSzrj 
38*bf2127beSzrj char *
sitemd5(const char * filename,char * const buf)39*bf2127beSzrj sitemd5(const char *filename, char * const buf)
40*bf2127beSzrj {
41*bf2127beSzrj 	unsigned char digest[MD5_DIGEST_LENGTH];
42*bf2127beSzrj 	static const char hex[]="0123456789abcdef";
43*bf2127beSzrj 	MD5_CTX ctx;
44*bf2127beSzrj 	unsigned char buffer[4096];
45*bf2127beSzrj 	struct stat st;
46*bf2127beSzrj 	off_t size;
47*bf2127beSzrj 	int fd, bytes, i, saved_errno;
48*bf2127beSzrj 
49*bf2127beSzrj 	if (!buf)
50*bf2127beSzrj 		return NULL;
51*bf2127beSzrj 
52*bf2127beSzrj 	fd = open(filename, O_RDONLY);
53*bf2127beSzrj 	if (fd < 0)
54*bf2127beSzrj 		return NULL;
55*bf2127beSzrj 	if (fstat(fd, &st) < 0) {
56*bf2127beSzrj 		bytes = -1;
57*bf2127beSzrj 		goto err;
58*bf2127beSzrj 	}
59*bf2127beSzrj 
60*bf2127beSzrj 	MD5_Init(&ctx);
61*bf2127beSzrj 	size = st.st_size;
62*bf2127beSzrj 	bytes = 0;
63*bf2127beSzrj 	while (size > 0) {
64*bf2127beSzrj 		if ((size_t)size > sizeof(buffer))
65*bf2127beSzrj 			bytes = read(fd, buffer, sizeof(buffer));
66*bf2127beSzrj 		else
67*bf2127beSzrj 			bytes = read(fd, buffer, size);
68*bf2127beSzrj 		if (bytes < 0)
69*bf2127beSzrj 			break;
70*bf2127beSzrj 		MD5_Update(&ctx, buffer, bytes);
71*bf2127beSzrj 		size -= bytes;
72*bf2127beSzrj 	}
73*bf2127beSzrj 
74*bf2127beSzrj err:
75*bf2127beSzrj 	saved_errno = errno;
76*bf2127beSzrj 	close(fd);
77*bf2127beSzrj 	errno = saved_errno;
78*bf2127beSzrj 
79*bf2127beSzrj 	if (bytes < 0)
80*bf2127beSzrj 		return NULL;
81*bf2127beSzrj 
82*bf2127beSzrj 	MD5_Final(digest, &ctx);
83*bf2127beSzrj 	for (i = 0; i < MD5_DIGEST_LENGTH; i++) {
84*bf2127beSzrj 		buf[2*i] = hex[digest[i] >> 4];
85*bf2127beSzrj 		buf[2*i+1] = hex[digest[i] & 0x0f];
86*bf2127beSzrj 	}
87*bf2127beSzrj 	buf[MD5_DIGEST_LENGTH * 2] = '\0';
88*bf2127beSzrj 
89*bf2127beSzrj 	return buf;
90*bf2127beSzrj }
91