^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);