Orange Pi5 kernel

Deprecated Linux kernel 5.10.110 for OrangePi 5/5B/5+ boards

3 Commits   0 Branches   0 Tags
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   1) /* mpicoder.c  -  Coder for the external representation of MPIs
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   2)  * Copyright (C) 1998, 1999 Free Software Foundation, Inc.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   3)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   4)  * This file is part of GnuPG.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   5)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   6)  * GnuPG is free software; you can redistribute it and/or modify
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   7)  * it under the terms of the GNU General Public License as published by
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   8)  * the Free Software Foundation; either version 2 of the License, or
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   9)  * (at your option) any later version.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  10)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  11)  * GnuPG is distributed in the hope that it will be useful,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  12)  * but WITHOUT ANY WARRANTY; without even the implied warranty of
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  13)  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  14)  * GNU General Public License for more details.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  15)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  16)  * You should have received a copy of the GNU General Public License
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  17)  * along with this program; if not, write to the Free Software
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  18)  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  19)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  20) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  21) #include <linux/bitops.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  22) #include <linux/count_zeros.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  23) #include <linux/byteorder/generic.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  24) #include <linux/scatterlist.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  25) #include <linux/string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  26) #include "mpi-internal.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  27) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  28) #define MAX_EXTERN_SCAN_BYTES (16*1024*1024)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  29) #define MAX_EXTERN_MPI_BITS 16384
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  30) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  31) /**
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  32)  * mpi_read_raw_data - Read a raw byte stream as a positive integer
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  33)  * @xbuffer: The data to read
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  34)  * @nbytes: The amount of data to read
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  35)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  36) MPI mpi_read_raw_data(const void *xbuffer, size_t nbytes)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  37) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  38) 	const uint8_t *buffer = xbuffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  39) 	int i, j;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  40) 	unsigned nbits, nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  41) 	mpi_limb_t a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  42) 	MPI val = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  43) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  44) 	while (nbytes > 0 && buffer[0] == 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  45) 		buffer++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  46) 		nbytes--;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  47) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  48) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  49) 	nbits = nbytes * 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  50) 	if (nbits > MAX_EXTERN_MPI_BITS) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  51) 		pr_info("MPI: mpi too large (%u bits)\n", nbits);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  52) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  53) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  54) 	if (nbytes > 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  55) 		nbits -= count_leading_zeros(buffer[0]) - (BITS_PER_LONG - 8);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  56) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  57) 	nlimbs = DIV_ROUND_UP(nbytes, BYTES_PER_MPI_LIMB);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  58) 	val = mpi_alloc(nlimbs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  59) 	if (!val)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  60) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  61) 	val->nbits = nbits;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  62) 	val->sign = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  63) 	val->nlimbs = nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  64) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  65) 	if (nbytes > 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  66) 		i = BYTES_PER_MPI_LIMB - nbytes % BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  67) 		i %= BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  68) 		for (j = nlimbs; j > 0; j--) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  69) 			a = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  70) 			for (; i < BYTES_PER_MPI_LIMB; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  71) 				a <<= 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  72) 				a |= *buffer++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  73) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  74) 			i = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  75) 			val->d[j - 1] = a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  76) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  77) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  78) 	return val;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  79) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  80) EXPORT_SYMBOL_GPL(mpi_read_raw_data);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  81) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  82) MPI mpi_read_from_buffer(const void *xbuffer, unsigned *ret_nread)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  83) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  84) 	const uint8_t *buffer = xbuffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  85) 	unsigned int nbits, nbytes;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  86) 	MPI val;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  87) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  88) 	if (*ret_nread < 2)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  89) 		return ERR_PTR(-EINVAL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  90) 	nbits = buffer[0] << 8 | buffer[1];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  91) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  92) 	if (nbits > MAX_EXTERN_MPI_BITS) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  93) 		pr_info("MPI: mpi too large (%u bits)\n", nbits);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  94) 		return ERR_PTR(-EINVAL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  95) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  96) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  97) 	nbytes = DIV_ROUND_UP(nbits, 8);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  98) 	if (nbytes + 2 > *ret_nread) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  99) 		pr_info("MPI: mpi larger than buffer nbytes=%u ret_nread=%u\n",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) 				nbytes, *ret_nread);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) 		return ERR_PTR(-EINVAL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) 	val = mpi_read_raw_data(buffer + 2, nbytes);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) 	if (!val)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) 		return ERR_PTR(-ENOMEM);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) 	*ret_nread = nbytes + 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) 	return val;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) EXPORT_SYMBOL_GPL(mpi_read_from_buffer);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) /****************
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114)  * Fill the mpi VAL from the hex string in STR.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) int mpi_fromstr(MPI val, const char *str)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) 	int sign = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) 	int prepend_zero = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) 	int i, j, c, c1, c2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) 	unsigned int nbits, nbytes, nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) 	mpi_limb_t a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) 	if (*str == '-') {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) 		sign = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) 		str++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) 	/* Skip optional hex prefix.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) 	if (*str == '0' && str[1] == 'x')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131) 		str += 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133) 	nbits = strlen(str);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) 	if (nbits > MAX_EXTERN_SCAN_BYTES) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) 		mpi_clear(val);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) 	nbits *= 4;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) 	if ((nbits % 8))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) 		prepend_zero = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142) 	nbytes = (nbits+7) / 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) 	nlimbs = (nbytes+BYTES_PER_MPI_LIMB-1) / BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) 	if (val->alloced < nlimbs)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) 		mpi_resize(val, nlimbs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 147) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 148) 	i = BYTES_PER_MPI_LIMB - (nbytes % BYTES_PER_MPI_LIMB);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 149) 	i %= BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 150) 	j = val->nlimbs = nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 151) 	val->sign = sign;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 152) 	for (; j > 0; j--) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 153) 		a = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 154) 		for (; i < BYTES_PER_MPI_LIMB; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 155) 			if (prepend_zero) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 156) 				c1 = '0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 157) 				prepend_zero = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 158) 			} else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 159) 				c1 = *str++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 160) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 161) 			if (!c1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 162) 				mpi_clear(val);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 163) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 164) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 165) 			c2 = *str++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 166) 			if (!c2) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 167) 				mpi_clear(val);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 168) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 169) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 170) 			if (c1 >= '0' && c1 <= '9')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 171) 				c = c1 - '0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 172) 			else if (c1 >= 'a' && c1 <= 'f')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 173) 				c = c1 - 'a' + 10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 174) 			else if (c1 >= 'A' && c1 <= 'F')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 175) 				c = c1 - 'A' + 10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 176) 			else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 177) 				mpi_clear(val);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 178) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 179) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 180) 			c <<= 4;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 181) 			if (c2 >= '0' && c2 <= '9')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 182) 				c |= c2 - '0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 183) 			else if (c2 >= 'a' && c2 <= 'f')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 184) 				c |= c2 - 'a' + 10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 185) 			else if (c2 >= 'A' && c2 <= 'F')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 186) 				c |= c2 - 'A' + 10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 187) 			else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 188) 				mpi_clear(val);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 189) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 190) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 191) 			a <<= 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 192) 			a |= c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 193) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 194) 		i = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 195) 		val->d[j-1] = a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 196) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 197) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 198) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 199) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 200) EXPORT_SYMBOL_GPL(mpi_fromstr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 201) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 202) MPI mpi_scanval(const char *string)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 203) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 204) 	MPI a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 205) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 206) 	a = mpi_alloc(0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 207) 	if (!a)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 208) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 209) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 210) 	if (mpi_fromstr(a, string)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 211) 		mpi_free(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 212) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 213) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 214) 	mpi_normalize(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 215) 	return a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 216) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 217) EXPORT_SYMBOL_GPL(mpi_scanval);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 218) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 219) static int count_lzeros(MPI a)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 220) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 221) 	mpi_limb_t alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 222) 	int i, lzeros = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 223) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 224) 	for (i = a->nlimbs - 1; i >= 0; i--) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 225) 		alimb = a->d[i];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 226) 		if (alimb == 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 227) 			lzeros += sizeof(mpi_limb_t);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 228) 		} else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 229) 			lzeros += count_leading_zeros(alimb) / 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 230) 			break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 231) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 232) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 233) 	return lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 234) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 235) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 236) /**
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 237)  * mpi_read_buffer() - read MPI to a bufer provided by user (msb first)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 238)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 239)  * @a:		a multi precision integer
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 240)  * @buf:	bufer to which the output will be written to. Needs to be at
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 241)  *		leaset mpi_get_size(a) long.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 242)  * @buf_len:	size of the buf.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 243)  * @nbytes:	receives the actual length of the data written on success and
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 244)  *		the data to-be-written on -EOVERFLOW in case buf_len was too
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 245)  *		small.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 246)  * @sign:	if not NULL, it will be set to the sign of a.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 247)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 248)  * Return:	0 on success or error code in case of error
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 249)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 250) int mpi_read_buffer(MPI a, uint8_t *buf, unsigned buf_len, unsigned *nbytes,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 251) 		    int *sign)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 252) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 253) 	uint8_t *p;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 254) #if BYTES_PER_MPI_LIMB == 4
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 255) 	__be32 alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 256) #elif BYTES_PER_MPI_LIMB == 8
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 257) 	__be64 alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 258) #else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 259) #error please implement for this limb size.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 260) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 261) 	unsigned int n = mpi_get_size(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 262) 	int i, lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 263) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 264) 	if (!buf || !nbytes)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 265) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 266) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 267) 	if (sign)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 268) 		*sign = a->sign;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 269) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 270) 	lzeros = count_lzeros(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 271) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 272) 	if (buf_len < n - lzeros) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 273) 		*nbytes = n - lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 274) 		return -EOVERFLOW;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 275) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 276) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 277) 	p = buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 278) 	*nbytes = n - lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 279) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 280) 	for (i = a->nlimbs - 1 - lzeros / BYTES_PER_MPI_LIMB,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 281) 			lzeros %= BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 282) 		i >= 0; i--) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 283) #if BYTES_PER_MPI_LIMB == 4
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 284) 		alimb = cpu_to_be32(a->d[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 285) #elif BYTES_PER_MPI_LIMB == 8
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 286) 		alimb = cpu_to_be64(a->d[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 287) #else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 288) #error please implement for this limb size.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 289) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 290) 		memcpy(p, (u8 *)&alimb + lzeros, BYTES_PER_MPI_LIMB - lzeros);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 291) 		p += BYTES_PER_MPI_LIMB - lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 292) 		lzeros = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 293) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 294) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 295) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 296) EXPORT_SYMBOL_GPL(mpi_read_buffer);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 297) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 298) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 299)  * mpi_get_buffer() - Returns an allocated buffer with the MPI (msb first).
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 300)  * Caller must free the return string.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 301)  * This function does return a 0 byte buffer with nbytes set to zero if the
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 302)  * value of A is zero.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 303)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 304)  * @a:		a multi precision integer.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 305)  * @nbytes:	receives the length of this buffer.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 306)  * @sign:	if not NULL, it will be set to the sign of the a.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 307)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 308)  * Return:	Pointer to MPI buffer or NULL on error
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 309)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 310) void *mpi_get_buffer(MPI a, unsigned *nbytes, int *sign)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 311) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 312) 	uint8_t *buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 313) 	unsigned int n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 314) 	int ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 315) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 316) 	if (!nbytes)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 317) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 318) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 319) 	n = mpi_get_size(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 320) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 321) 	if (!n)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 322) 		n++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 323) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 324) 	buf = kmalloc(n, GFP_KERNEL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 325) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 326) 	if (!buf)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 327) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 328) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 329) 	ret = mpi_read_buffer(a, buf, n, nbytes, sign);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 330) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 331) 	if (ret) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 332) 		kfree(buf);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 333) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 334) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 335) 	return buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 336) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 337) EXPORT_SYMBOL_GPL(mpi_get_buffer);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 338) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 339) /**
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 340)  * mpi_write_to_sgl() - Funnction exports MPI to an sgl (msb first)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 341)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 342)  * This function works in the same way as the mpi_read_buffer, but it
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 343)  * takes an sgl instead of u8 * buf.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 344)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 345)  * @a:		a multi precision integer
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 346)  * @sgl:	scatterlist to write to. Needs to be at least
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 347)  *		mpi_get_size(a) long.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 348)  * @nbytes:	the number of bytes to write.  Leading bytes will be
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 349)  *		filled with zero.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 350)  * @sign:	if not NULL, it will be set to the sign of a.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 351)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 352)  * Return:	0 on success or error code in case of error
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 353)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 354) int mpi_write_to_sgl(MPI a, struct scatterlist *sgl, unsigned nbytes,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 355) 		     int *sign)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 356) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 357) 	u8 *p, *p2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 358) #if BYTES_PER_MPI_LIMB == 4
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 359) 	__be32 alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 360) #elif BYTES_PER_MPI_LIMB == 8
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 361) 	__be64 alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 362) #else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 363) #error please implement for this limb size.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 364) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 365) 	unsigned int n = mpi_get_size(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 366) 	struct sg_mapping_iter miter;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 367) 	int i, x, buf_len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 368) 	int nents;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 369) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 370) 	if (sign)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 371) 		*sign = a->sign;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 372) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 373) 	if (nbytes < n)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 374) 		return -EOVERFLOW;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 375) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 376) 	nents = sg_nents_for_len(sgl, nbytes);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 377) 	if (nents < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 378) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 379) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 380) 	sg_miter_start(&miter, sgl, nents, SG_MITER_ATOMIC | SG_MITER_TO_SG);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 381) 	sg_miter_next(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 382) 	buf_len = miter.length;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 383) 	p2 = miter.addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 384) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 385) 	while (nbytes > n) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 386) 		i = min_t(unsigned, nbytes - n, buf_len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 387) 		memset(p2, 0, i);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 388) 		p2 += i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 389) 		nbytes -= i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 390) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 391) 		buf_len -= i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 392) 		if (!buf_len) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 393) 			sg_miter_next(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 394) 			buf_len = miter.length;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 395) 			p2 = miter.addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 396) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 397) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 398) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 399) 	for (i = a->nlimbs - 1; i >= 0; i--) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 400) #if BYTES_PER_MPI_LIMB == 4
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 401) 		alimb = a->d[i] ? cpu_to_be32(a->d[i]) : 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 402) #elif BYTES_PER_MPI_LIMB == 8
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 403) 		alimb = a->d[i] ? cpu_to_be64(a->d[i]) : 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 404) #else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 405) #error please implement for this limb size.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 406) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 407) 		p = (u8 *)&alimb;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 408) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 409) 		for (x = 0; x < sizeof(alimb); x++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 410) 			*p2++ = *p++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 411) 			if (!--buf_len) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 412) 				sg_miter_next(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 413) 				buf_len = miter.length;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 414) 				p2 = miter.addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 415) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 416) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 417) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 418) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 419) 	sg_miter_stop(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 420) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 421) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 422) EXPORT_SYMBOL_GPL(mpi_write_to_sgl);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 423) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 424) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 425)  * mpi_read_raw_from_sgl() - Function allocates an MPI and populates it with
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 426)  *			     data from the sgl
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 427)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 428)  * This function works in the same way as the mpi_read_raw_data, but it
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 429)  * takes an sgl instead of void * buffer. i.e. it allocates
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 430)  * a new MPI and reads the content of the sgl to the MPI.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 431)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 432)  * @sgl:	scatterlist to read from
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 433)  * @nbytes:	number of bytes to read
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 434)  *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 435)  * Return:	Pointer to a new MPI or NULL on error
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 436)  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 437) MPI mpi_read_raw_from_sgl(struct scatterlist *sgl, unsigned int nbytes)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 438) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 439) 	struct sg_mapping_iter miter;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 440) 	unsigned int nbits, nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 441) 	int x, j, z, lzeros, ents;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 442) 	unsigned int len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 443) 	const u8 *buff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 444) 	mpi_limb_t a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 445) 	MPI val = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 446) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 447) 	ents = sg_nents_for_len(sgl, nbytes);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 448) 	if (ents < 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 449) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 450) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 451) 	sg_miter_start(&miter, sgl, ents, SG_MITER_ATOMIC | SG_MITER_FROM_SG);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 452) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 453) 	lzeros = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 454) 	len = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 455) 	while (nbytes > 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 456) 		while (len && !*buff) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 457) 			lzeros++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 458) 			len--;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 459) 			buff++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 460) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 461) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 462) 		if (len && *buff)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 463) 			break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 464) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 465) 		sg_miter_next(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 466) 		buff = miter.addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 467) 		len = miter.length;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 468) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 469) 		nbytes -= lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 470) 		lzeros = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 471) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 472) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 473) 	miter.consumed = lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 474) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 475) 	nbytes -= lzeros;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 476) 	nbits = nbytes * 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 477) 	if (nbits > MAX_EXTERN_MPI_BITS) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 478) 		sg_miter_stop(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 479) 		pr_info("MPI: mpi too large (%u bits)\n", nbits);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 480) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 481) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 482) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 483) 	if (nbytes > 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 484) 		nbits -= count_leading_zeros(*buff) - (BITS_PER_LONG - 8);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 485) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 486) 	sg_miter_stop(&miter);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 487) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 488) 	nlimbs = DIV_ROUND_UP(nbytes, BYTES_PER_MPI_LIMB);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 489) 	val = mpi_alloc(nlimbs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 490) 	if (!val)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 491) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 492) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 493) 	val->nbits = nbits;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 494) 	val->sign = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 495) 	val->nlimbs = nlimbs;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 496) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 497) 	if (nbytes == 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 498) 		return val;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 499) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 500) 	j = nlimbs - 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 501) 	a = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 502) 	z = BYTES_PER_MPI_LIMB - nbytes % BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 503) 	z %= BYTES_PER_MPI_LIMB;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 504) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 505) 	while (sg_miter_next(&miter)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 506) 		buff = miter.addr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 507) 		len = miter.length;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 508) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 509) 		for (x = 0; x < len; x++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 510) 			a <<= 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 511) 			a |= *buff++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 512) 			if (((z + x + 1) % BYTES_PER_MPI_LIMB) == 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 513) 				val->d[j--] = a;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 514) 				a = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 515) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 516) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 517) 		z += x;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 518) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 519) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 520) 	return val;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 521) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 522) EXPORT_SYMBOL_GPL(mpi_read_raw_from_sgl);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 523) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 524) /* Perform a two's complement operation on buffer P of size N bytes.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 525) static void twocompl(unsigned char *p, unsigned int n)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 526) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 527) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 528) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 529) 	for (i = n-1; i >= 0 && !p[i]; i--)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 530) 		;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 531) 	if (i >= 0) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 532) 		if ((p[i] & 0x01))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 533) 			p[i] = (((p[i] ^ 0xfe) | 0x01) & 0xff);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 534) 		else if ((p[i] & 0x02))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 535) 			p[i] = (((p[i] ^ 0xfc) | 0x02) & 0xfe);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 536) 		else if ((p[i] & 0x04))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 537) 			p[i] = (((p[i] ^ 0xf8) | 0x04) & 0xfc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 538) 		else if ((p[i] & 0x08))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 539) 			p[i] = (((p[i] ^ 0xf0) | 0x08) & 0xf8);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 540) 		else if ((p[i] & 0x10))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 541) 			p[i] = (((p[i] ^ 0xe0) | 0x10) & 0xf0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 542) 		else if ((p[i] & 0x20))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 543) 			p[i] = (((p[i] ^ 0xc0) | 0x20) & 0xe0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 544) 		else if ((p[i] & 0x40))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 545) 			p[i] = (((p[i] ^ 0x80) | 0x40) & 0xc0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 546) 		else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 547) 			p[i] = 0x80;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 548) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 549) 		for (i--; i >= 0; i--)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 550) 			p[i] ^= 0xff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 551) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 552) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 553) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 554) int mpi_print(enum gcry_mpi_format format, unsigned char *buffer,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 555) 			size_t buflen, size_t *nwritten, MPI a)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 556) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 557) 	unsigned int nbits = mpi_get_nbits(a);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 558) 	size_t len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 559) 	size_t dummy_nwritten;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 560) 	int negative;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 561) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 562) 	if (!nwritten)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 563) 		nwritten = &dummy_nwritten;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 564) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 565) 	/* Libgcrypt does no always care to set clear the sign if the value
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 566) 	 * is 0.  For printing this is a bit of a surprise, in particular
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 567) 	 * because if some of the formats don't support negative numbers but
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 568) 	 * should be able to print a zero.  Thus we need this extra test
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 569) 	 * for a negative number.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 570) 	 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 571) 	if (a->sign && mpi_cmp_ui(a, 0))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 572) 		negative = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 573) 	else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 574) 		negative = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 575) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 576) 	len = buflen;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 577) 	*nwritten = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 578) 	if (format == GCRYMPI_FMT_STD) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 579) 		unsigned char *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 580) 		int extra = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 581) 		unsigned int n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 582) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 583) 		tmp = mpi_get_buffer(a, &n, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 584) 		if (!tmp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 585) 			return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 586) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 587) 		if (negative) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 588) 			twocompl(tmp, n);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 589) 			if (!(*tmp & 0x80)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 590) 				/* Need to extend the sign.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 591) 				n++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 592) 				extra = 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 593) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 594) 		} else if (n && (*tmp & 0x80)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 595) 			/* Positive but the high bit of the returned buffer is set.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 596) 			 * Thus we need to print an extra leading 0x00 so that the
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 597) 			 * output is interpreted as a positive number.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 598) 			 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 599) 			n++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 600) 			extra = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 601) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 602) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 603) 		if (buffer && n > len) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 604) 			/* The provided buffer is too short. */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 605) 			kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 606) 			return -E2BIG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 607) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 608) 		if (buffer) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 609) 			unsigned char *s = buffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 610) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 611) 			if (extra == 1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 612) 				*s++ = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 613) 			else if (extra)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 614) 				*s++ = 0xff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 615) 			memcpy(s, tmp, n-!!extra);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 616) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 617) 		kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 618) 		*nwritten = n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 619) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 620) 	} else if (format == GCRYMPI_FMT_USG) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 621) 		unsigned int n = (nbits + 7)/8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 622) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 623) 		/* Note:  We ignore the sign for this format.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 624) 		/* FIXME: for performance reasons we should put this into
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 625) 		 * mpi_aprint because we can then use the buffer directly.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 626) 		 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 627) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 628) 		if (buffer && n > len)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 629) 			return -E2BIG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 630) 		if (buffer) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 631) 			unsigned char *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 632) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 633) 			tmp = mpi_get_buffer(a, &n, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 634) 			if (!tmp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 635) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 636) 			memcpy(buffer, tmp, n);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 637) 			kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 638) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 639) 		*nwritten = n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 640) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 641) 	} else if (format == GCRYMPI_FMT_PGP) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 642) 		unsigned int n = (nbits + 7)/8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 643) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 644) 		/* The PGP format can only handle unsigned integers.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 645) 		if (negative)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 646) 			return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 647) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 648) 		if (buffer && n+2 > len)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 649) 			return -E2BIG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 650) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 651) 		if (buffer) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 652) 			unsigned char *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 653) 			unsigned char *s = buffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 654) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 655) 			s[0] = nbits >> 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 656) 			s[1] = nbits;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 657) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 658) 			tmp = mpi_get_buffer(a, &n, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 659) 			if (!tmp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 660) 				return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 661) 			memcpy(s+2, tmp, n);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 662) 			kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 663) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 664) 		*nwritten = n+2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 665) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 666) 	} else if (format == GCRYMPI_FMT_SSH) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 667) 		unsigned char *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 668) 		int extra = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 669) 		unsigned int n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 670) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 671) 		tmp = mpi_get_buffer(a, &n, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 672) 		if (!tmp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 673) 			return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 674) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 675) 		if (negative) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 676) 			twocompl(tmp, n);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 677) 			if (!(*tmp & 0x80)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 678) 				/* Need to extend the sign.  */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 679) 				n++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 680) 				extra = 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 681) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 682) 		} else if (n && (*tmp & 0x80)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 683) 			n++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 684) 			extra = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 685) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 686) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 687) 		if (buffer && n+4 > len) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 688) 			kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 689) 			return -E2BIG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 690) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 691) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 692) 		if (buffer) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 693) 			unsigned char *s = buffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 694) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 695) 			*s++ = n >> 24;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 696) 			*s++ = n >> 16;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 697) 			*s++ = n >> 8;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 698) 			*s++ = n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 699) 			if (extra == 1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 700) 				*s++ = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 701) 			else if (extra)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 702) 				*s++ = 0xff;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 703) 			memcpy(s, tmp, n-!!extra);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 704) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 705) 		kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 706) 		*nwritten = 4+n;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 707) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 708) 	} else if (format == GCRYMPI_FMT_HEX) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 709) 		unsigned char *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 710) 		int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 711) 		int extra = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 712) 		unsigned int n = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 713) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 714) 		tmp = mpi_get_buffer(a, &n, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 715) 		if (!tmp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 716) 			return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 717) 		if (!n || (*tmp & 0x80))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 718) 			extra = 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 719) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 720) 		if (buffer && 2*n + extra + negative + 1 > len) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 721) 			kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 722) 			return -E2BIG;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 723) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 724) 		if (buffer) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 725) 			unsigned char *s = buffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 726) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 727) 			if (negative)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 728) 				*s++ = '-';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 729) 			if (extra) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 730) 				*s++ = '0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 731) 				*s++ = '0';
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 732) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 733) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 734) 			for (i = 0; i < n; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 735) 				unsigned int c = tmp[i];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 736) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 737) 				*s++ = (c >> 4) < 10 ? '0'+(c>>4) : 'A'+(c>>4)-10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 738) 				c &= 15;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 739) 				*s++ = c < 10 ? '0'+c : 'A'+c-10;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 740) 			}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 741) 			*s++ = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 742) 			*nwritten = s - buffer;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 743) 		} else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 744) 			*nwritten = 2*n + extra + negative + 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 745) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 746) 		kfree(tmp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 747) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 748) 	} else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 749) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 750) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 751) EXPORT_SYMBOL_GPL(mpi_print);