home · contact · privacy
Fixed bug that freed non-existing map objects, triggered segfaults.
[plomrogue] / src / map_objects.c
1 /* map_objects.c */
2
3 #include "map_objects.h"
4 #include <stdlib.h> /* for malloc(), calloc(), free(), atoi() */
5 #include <stdint.h> /* for uint8_t */
6 #include <stdio.h> /* for FILE typedef */
7 #include <string.h> /* for strchr(), strlen(), memcpy(), strtok() */
8 #include "readwrite.h" /* for [read/write]_uint[8/16/23][_bigendian]() */
9 #include "misc.h" /* for textfile_sizes(), find_passable_pos() */
10 #include "main.h" /* for World struct */
11 #include "rexit.h" /* for err_exit() */
12
13
14
15 /* Return pointer to newly allocated map object struct of size "size". If first
16  * in map object chain ("first" pointing to !0), point "start" to it.
17  *
18  * Returns NULL instead of MapObj pointer if malloc() failed.
19  */
20 static struct MapObj * get_next_map_obj(void * start, char * first,
21                                         size_t size, struct MapObj * map_obj);
22
23
24
25 /* Map-object-type-specific helpers to (build|write|read)_map_objects(). */
26 static void build_map_objects_itemdata(struct MapObjDef * map_obj_def,
27                                        void * start);
28 static void build_map_objects_monsterdata(struct MapObjDef * map_obj_def,
29                                           void * start);
30 static uint8_t write_map_objects_monsterdata(void * start, FILE * file);
31 static uint8_t read_map_objects_monsterdata( void * start, FILE * file);
32
33
34
35 static struct MapObj * get_next_map_obj(void * start, char * first,
36                                         size_t size, struct MapObj * map_obj)
37 {
38     if (* first)
39     {
40         struct MapObj * * z = start;
41         map_obj = malloc(size);
42         * z = map_obj;
43         * first = 0;
44     }
45     else
46     {
47         map_obj->next = malloc(size);
48         map_obj = map_obj->next;
49     }
50     return map_obj;
51 }
52
53
54
55 static void build_map_objects_itemdata(struct MapObjDef * map_obj_def,
56                                        void * start)
57 {
58   struct Item * i = (struct Item *) start;
59   i->map_obj.type = map_obj_def->id;
60 }
61
62
63
64 static void build_map_objects_monsterdata(struct MapObjDef * map_obj_def,
65                                           void * start)
66 {
67     struct Monster * m = (struct Monster *) start;
68     m->map_obj.type = map_obj_def->id;
69     struct MonsterDef * md = (struct MonsterDef *) map_obj_def;
70     m->hitpoints = md->hitpoints_start;
71 }
72
73
74
75 static uint8_t write_map_objects_monsterdata(void * start, FILE * file)
76 {
77     struct Monster * m = (struct Monster *) start;
78     return write_uint8(m->hitpoints, file);
79 }
80
81
82
83 static uint8_t read_map_objects_monsterdata (void * start, FILE * file)
84 {
85     struct Monster * m = (struct Monster *) start;
86     return read_uint8(file, &m->hitpoints);
87 }
88
89
90
91 extern void init_map_object_defs(struct World * world, char * filename)
92 {
93     world->item_def    = 0;
94     world->monster_def = 0;
95     char * err = "Trouble in init_map_object_defs() with fopen().";
96     FILE * file = fopen(filename, "r");
97     exit_err(NULL == file, world, err);
98     uint16_t linemax;
99     err = "Trouble in init_map_object_defs() with textfile_sizes().";
100     exit_err(textfile_sizes(file, &linemax, NULL), world, err);
101     struct MapObjDef  mod;
102     struct ItemDef    id;
103     struct MonsterDef md;
104     struct ItemDef    * * p_p_id  = &world->item_def;
105     struct MonsterDef * * p_p_md  = &world->monster_def;
106     err = "Trouble in init_map_object_defs() with malloc() or calloc().";
107     char * defline = malloc(linemax);
108     exit_err(NULL == defline, world, err);
109     char * line_p;
110     char * delim = " ";
111     while (fgets(defline, linemax, file))
112     {
113         mod.next    = 0;
114         mod.id      = atoi(strtok(defline, delim));
115         mod.m_or_i  = * strtok(NULL, delim);
116         mod.mapchar = * strtok(NULL, delim);
117         if ('i' == mod.m_or_i)
118         {
119             line_p = strtok(NULL, delim);
120         }
121         else
122         {
123             md.corpse_id       = atoi(strtok(NULL, delim));
124             md.hitpoints_start = atoi(strtok(NULL, delim));
125             line_p             = strtok(NULL, delim);
126         }
127         mod.desc = calloc(strlen(line_p), sizeof(char));
128         exit_err(NULL == mod.desc, world, err);
129         memcpy(mod.desc, line_p, strlen(line_p) - 1);
130         if ('i' == mod.m_or_i)
131         {
132             id.map_obj_def = mod;
133             * p_p_id       = malloc(sizeof(struct ItemDef));
134             exit_err(NULL == p_p_id, world, err);
135             * * p_p_id     = id;
136             p_p_id         = (struct ItemDef    * *) * p_p_id;
137         }
138         else
139         {
140             md.map_obj_def = mod;
141             * p_p_md       = malloc(sizeof(struct MonsterDef));
142             exit_err(NULL == p_p_md, world, err);
143             * * p_p_md     = md;
144             p_p_md         = (struct MonsterDef * *) * p_p_md;
145         }
146     }
147     free(defline);
148     err = "Trouble in init_map_object_defs() with fclose().";
149     exit_err(fclose(file), world, err);
150 }
151
152
153
154 extern void free_item_defs(struct ItemDef * id_start)
155 {
156     if (0 != id_start->map_obj_def.next)
157     {
158         free_item_defs((struct ItemDef *) id_start->map_obj_def.next);
159     }
160     free(id_start->map_obj_def.desc);
161     free(id_start);
162 }
163
164
165
166
167 extern void free_monster_defs(struct MonsterDef * md_start)
168 {
169     if (0 != md_start->map_obj_def.next)
170     {
171         free_monster_defs((struct MonsterDef *) md_start->map_obj_def.next);
172     }
173     free(md_start->map_obj_def.desc);
174     free(md_start);
175 }
176
177
178
179 extern uint8_t write_map_objects(struct World * world, void * start,
180                                  FILE * file)
181 {
182     struct MapObj * map_obj;
183     struct MapObjDef * mod;
184     for (map_obj = start; map_obj != 0; map_obj = map_obj->next)
185     {
186         if (   write_uint8(map_obj->type, file)
187             || write_uint16_bigendian(map_obj->pos.y + 1, file)
188             || write_uint16_bigendian(map_obj->pos.x + 1, file))
189         {
190             return 1;
191         }
192         mod = get_map_obj_def(world, map_obj->type);
193         if ('m' == mod->m_or_i)
194         {
195             if (write_map_objects_monsterdata(map_obj, file))
196             {
197                 return 1;
198             }
199         }
200     }
201     return write_uint16_bigendian(0, file);
202 }
203
204
205
206 extern uint8_t read_map_objects(struct World * world, void * start, FILE * file)
207 {
208     struct MapObj * map_obj;
209     struct MapObjDef * mod;
210     size_t size;
211     uint8_t type;
212     char first = 1;
213     long pos;
214     uint16_t read_uint16 = 0;
215     char * err = "Trouble in read_map_objects() with get_next_map_obj().";
216     while (1)
217     {
218         pos = ftell(file);
219         if (read_uint16_bigendian(file, &read_uint16))
220         {
221             return 1;
222         }
223         if (0 == read_uint16)
224         {
225             break;
226         }
227         fseek(file, pos, SEEK_SET);
228         if (read_uint8(file, &type))
229         {
230             return 1;
231         }
232         mod = get_map_obj_def(world, type);
233         if ('m' == mod->m_or_i)
234         {
235             size = sizeof(struct Monster);
236         }
237         else
238         {
239             size = sizeof(struct Item);
240         }
241         map_obj = get_next_map_obj(start, &first, size, map_obj);
242         exit_err(NULL == map_obj, world, err);
243         map_obj->type = type;
244         if (   read_uint16_bigendian(file, &map_obj->pos.y)
245             || read_uint16_bigendian(file, &map_obj->pos.x))
246         {
247             return 1;
248         }
249         map_obj->pos.y--;
250         map_obj->pos.x--;
251         if ('m' == mod->m_or_i)
252         {
253             if (read_map_objects_monsterdata(map_obj, file))
254             {
255                 return 1;
256             }
257         }
258     }
259     if (!first)
260     {
261         map_obj->next = 0;
262     }
263     return 0;
264 }
265
266
267
268 extern void * build_map_objects(struct World * world, void * start, char def_id,
269                                 uint8_t n)
270 {
271     uint8_t i;
272     struct MapObj * mo;
273     char first = 1;
274     struct MapObjDef * mod = get_map_obj_def(world, def_id);
275     size_t size = 0;
276     char * err = "Trouble in build_map_objects() with get_next_map_obj().";
277     if ('i' == mod->m_or_i)
278     {
279         size = sizeof(struct Item);
280     }
281     else
282     {
283         size = sizeof(struct Monster);
284     }
285     for (i = 0; i < n; i++)
286     {
287         mo = get_next_map_obj(start, &first, size, mo);
288         exit_err(NULL == mo, world, err);
289         mo->pos = find_passable_pos(world->map);
290         if ('i' == mod->m_or_i)
291         {
292             build_map_objects_itemdata(mod, mo);
293         }
294         else
295         {
296             build_map_objects_monsterdata(mod, mo);
297         }
298
299     }
300     if (!first)
301     {
302         mo->next = 0;
303     }
304     return &mo->next;
305 }
306
307
308
309 extern void free_items(struct Item * item)
310 {
311     if (0 == item)
312     {
313         return;
314     }
315     free_items((struct Item *) item->map_obj.next);
316     free(item);
317 }
318
319
320
321 extern void free_monsters(struct Monster * monster)
322 {
323     if (0 == monster)
324     {
325         return;
326     }
327     free_monsters((struct Monster *) monster->map_obj.next);
328     free(monster);
329 }
330
331
332
333 extern struct MapObjDef * get_map_obj_def(struct World * world, char def_id)
334 {
335     struct MapObjDef * d = NULL;
336     for (d = (struct MapObjDef *) world->monster_def;
337          d->id != def_id && 0 != d->next;
338          d = d->next);
339     if (d->id != def_id)
340     {
341         for (d = (struct MapObjDef *) world->item_def;
342              d->id != def_id && 0 != d->next;
343              d = d->next);
344     }
345     return d;
346 }