mem.c

   1 /*
   2  * mem.c - Routines to cleanup memory after use.
   3  *
   4  * Copyright 2002-2004,2007 Jonathan McDowell <noodles@earth.li>
   5  *
   6  * This program is free software: you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the Free
   8  * Software Foundation; version 2 of the License.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along with
  16  * this program.  If not, see <https://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22
  23 #include "keystructs.h"
  24 #include "ll.h"
  25 #include "mem.h"
  26 #include "stats.h"
  27
  28 /**
  29  *      packet_dup - duplicate an OpenPGP packet.
  30  *      @packet: The packet to duplicate.
  31  *
  32  *      This function takes an OpenPGP packet structure and duplicates it,
  33  *      including the data part. It returns NULL if there is a problem
  34  *      allocating memory for the duplicate.
  35  */
  36 struct openpgp_packet *packet_dup(struct openpgp_packet *packet)
  37 {
  38         struct openpgp_packet *newpacket = NULL;
  39
  40         if (packet == NULL)
  41                 return NULL;
  42
  43         newpacket = malloc(sizeof (struct openpgp_packet));
  44         if (newpacket != NULL) {
  45                 newpacket->tag = packet->tag;
  46                 newpacket->newformat = packet->newformat;
  47                 newpacket->length = packet->length;
  48                 newpacket->data = malloc(newpacket->length);
  49                 if (newpacket->data != NULL) {
  50                         memcpy(newpacket->data, packet->data,
  51                                         newpacket->length);
  52                 }
  53         }
  54
  55         return newpacket;
  56 }
  57
  58 /**
  59  *      packet_list_add - Adds an OpenPGP packet list to another.
  60  *      @list: The packet list to add to.
  61  *      @list_end: The end of the packet list to add to.
  62  *      @packet_list: The packet list to add.
  63  *
  64  *      This function takes an OpenPGP packet list and adds it to another list,
  65  *      duplicating it in the process. The list to add to need not exists to
  66  *      begin with, in which case the function simply duplicates the supplied
  67  *      list.
  68  */
  69 void packet_list_add(struct openpgp_packet_list **list,
  70                 struct openpgp_packet_list **list_end,
  71                 struct openpgp_packet_list *packet_list)
  72 {
  73         for (; packet_list != NULL; packet_list = packet_list->next) {
  74                 ADD_PACKET_TO_LIST((*list_end),
  75                                 packet_dup(packet_list->packet));
  76                 if (*list == NULL) {
  77                         *list = *list_end;
  78                 }
  79         }
  80
  81         return;
  82 }
  83
  84 /**
  85  *      free_packet - free the memory used by an OpenPGP packet.
  86  *      @packet: The packet to free.
  87  *
  88  *      Takes an OpenPGP packet structure and frees the memory used by it,
  89  *      including the data part.
  90  */
  91 void free_packet(struct openpgp_packet *packet) {
  92         if (packet->data != NULL) {
  93                 free(packet->data);
  94                 packet->data = NULL;
  95         }
  96         free(packet);
  97 }
  98
  99 /**
 100  *      free_packet_list - free the memory used by an OpenPGP packet list.
 101  *      @packet_list: The packet list to free.
 102  *
 103  *      Takes an OpenPGP packet list structure and frees the memory used by the
 104  *      packets in it and the linked list overhead.
 105  */
 106 void free_packet_list(struct openpgp_packet_list *packet_list) {
 107         struct openpgp_packet_list *nextpacket = NULL;
 108
 109         while (packet_list != NULL) {
 110                 nextpacket = packet_list->next;
 111                 if (packet_list->packet != NULL) {
 112                         free_packet(packet_list->packet);
 113                 }
 114                 free(packet_list);
 115                 packet_list = nextpacket;
 116         }
 117 }
 118
 119 /**
 120  *      free_signedpacket_list - free an OpenPGP signed packet list.
 121  *      @signedpacket_list: The packet list to free.
 122  *
 123  *      Takes an OpenPGP signed packet list structure and frees the memory used
 124  *      by the packets and signatures it and the linked list overhead.
 125  */
 126 void free_signedpacket_list(
 127                 struct openpgp_signedpacket_list *signedpacket_list) {
 128         struct openpgp_signedpacket_list *nextpacket = NULL;
 129
 130         while (signedpacket_list != NULL) {
 131                 nextpacket = signedpacket_list->next;
 132                 if (signedpacket_list->packet != NULL) {
 133                         free_packet(signedpacket_list->packet);
 134                 }
 135                 if (signedpacket_list->sigs != NULL) {
 136                         free_packet_list(signedpacket_list->sigs);
 137                 }
 138                 free(signedpacket_list);
 139                 signedpacket_list = nextpacket;
 140         }
 141 }
 142
 143 /**
 144  *      free_publickey - free an OpenPGP public key structure.
 145  *      @key: The key to free.
 146  *
 147  *      Takes an OpenPGP key and frees the memory used by all the structures it
 148  *      contains.
 149  */
 150 void free_publickey(struct openpgp_publickey *key) {
 151         struct openpgp_publickey *nextkey = NULL;
 152
 153         while (key != NULL) {
 154                 nextkey = key->next;
 155                 if (key->publickey != NULL) {
 156                         free_packet(key->publickey);
 157                         key->publickey = NULL;
 158                 }
 159                 if (key->sigs != NULL) {
 160                         free_packet_list(key->sigs);
 161                         key->sigs = NULL;
 162                 }
 163                 if (key->uids != NULL) {
 164                         free_signedpacket_list(key->uids);
 165                         key->uids = NULL;
 166                 }
 167                 if (key->subkeys != NULL) {
 168                         free_signedpacket_list(key->subkeys);
 169                         key->subkeys = NULL;
 170                 }
 171                 free(key);
 172                 key = nextkey;
 173         }
 174 }