bignum.c 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. /*
  2. * Dropbear - a SSH2 server
  3. *
  4. * Copyright (c) 2002,2003 Matt Johnston
  5. * All rights reserved.
  6. *
  7. * Permission is hereby granted, free of charge, to any person obtaining a copy
  8. * of this software and associated documentation files (the "Software"), to deal
  9. * in the Software without restriction, including without limitation the rights
  10. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11. * copies of the Software, and to permit persons to whom the Software is
  12. * furnished to do so, subject to the following conditions:
  13. *
  14. * The above copyright notice and this permission notice shall be included in
  15. * all copies or substantial portions of the Software.
  16. *
  17. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  23. * SOFTWARE. */
  24. /* Contains helper functions for mp_int handling */
  25. #include "includes.h"
  26. #include "dbutil.h"
  27. /* wrapper for mp_init, failing fatally on errors (memory allocation) */
  28. void m_mp_init(mp_int *mp) {
  29. if (mp_init(mp) != MP_OKAY) {
  30. dropbear_exit("Mem alloc error");
  31. }
  32. }
  33. /* simplified duplication of bn_mp_multi's mp_init_multi, but die fatally
  34. * on error */
  35. void m_mp_init_multi(mp_int *mp, ...)
  36. {
  37. mp_int* cur_arg = mp;
  38. va_list args;
  39. va_start(args, mp); /* init args to next argument from caller */
  40. while (cur_arg != NULL) {
  41. if (mp_init(cur_arg) != MP_OKAY) {
  42. dropbear_exit("Mem alloc error");
  43. }
  44. cur_arg = va_arg(args, mp_int*);
  45. }
  46. va_end(args);
  47. }
  48. void m_mp_alloc_init_multi(mp_int **mp, ...)
  49. {
  50. mp_int** cur_arg = mp;
  51. va_list args;
  52. va_start(args, mp); /* init args to next argument from caller */
  53. while (cur_arg != NULL) {
  54. *cur_arg = m_malloc(sizeof(mp_int));
  55. if (mp_init(*cur_arg) != MP_OKAY) {
  56. dropbear_exit("Mem alloc error");
  57. }
  58. cur_arg = va_arg(args, mp_int**);
  59. }
  60. va_end(args);
  61. }
  62. void bytes_to_mp(mp_int *mp, const unsigned char* bytes, unsigned int len) {
  63. if (mp_read_unsigned_bin(mp, (unsigned char*)bytes, len) != MP_OKAY) {
  64. dropbear_exit("Mem alloc error");
  65. }
  66. }
  67. /* hash the ssh representation of the mp_int mp */
  68. void hash_process_mp(const struct ltc_hash_descriptor *hash_desc,
  69. hash_state *hs, mp_int *mp) {
  70. buffer * buf;
  71. buf = buf_new(512 + 20); /* max buffer is a 4096 bit key,
  72. plus header + some leeway*/
  73. buf_putmpint(buf, mp);
  74. hash_desc->process(hs, buf->data, buf->len);
  75. buf_free(buf);
  76. }