fix two more stack overflows on win32 (in one case two 3.25mb entity_frame_t structs...
[divverent/darkplaces.git] / protocol.c
1
2 #include "quakedef.h"
3
4 void ClearStateToDefault(entity_state_t *s)
5 {
6         s->active = 0;
7         s->time = 0;
8         VectorClear(s->origin);
9         VectorClear(s->angles);
10         s->effects = 0;
11         s->modelindex = 0;
12         s->frame = 0;
13         s->colormap = 0;
14         s->skin = 0;
15         s->alpha = 255;
16         s->scale = 16;
17         s->glowsize = 0;
18         s->glowcolor = 254;
19         s->flags = 0;
20 }
21
22 // (server) clears the database to contain no frames (thus delta compression compresses against nothing)
23 void EntityFrame_ClearDatabase(entity_database_t *d)
24 {
25         memset(d, 0, sizeof(*d));
26 }
27
28 // (server and client) removes frames older than 'frame' from database
29 void EntityFrame_AckFrame(entity_database_t *d, int frame)
30 {
31         int i;
32         if (d->ackframe < frame)
33                 d->ackframe = frame;
34         for (i = 0;i < d->numframes && d->frames[i].framenum < frame;i++);
35         // ignore outdated frame acks (out of order packets)
36         if (i == 0)
37                 return;
38         d->numframes -= i;
39         // if some queue is left, slide it down to beginning of array
40         if (d->numframes)
41                 memmove(&d->frames[0], &d->frames[i], sizeof(d->frames[0]) * d->numframes);
42 }
43
44 // (server) clears frame, to prepare for adding entities
45 void EntityFrame_Clear(entity_frame_t *f, vec3_t eye)
46 {
47         f->time = 0;
48         f->framenum = 0;
49         f->numentities = 0;
50         if (eye == NULL)
51         {
52                 VectorClear(f->eye);
53         }
54         else
55         {
56                 VectorCopy(eye, f->eye);
57         }
58 }
59
60 // (server) allocates an entity slot in frame, returns NULL if full
61 entity_state_t *EntityFrame_NewEntity(entity_frame_t *f, int number)
62 {
63         entity_state_t *e;
64         if (f->numentities >= MAX_ENTITY_DATABASE)
65                 return NULL;
66         e = &f->entitydata[f->numentities++];
67         e->active = true;
68         e->number = number;
69         return e;
70 }
71
72 // (server and client) reads a frame from the database
73 void EntityFrame_FetchFrame(entity_database_t *d, int framenum, entity_frame_t *f)
74 {
75         int i, n;
76         EntityFrame_Clear(f, NULL);
77         for (i = 0;i < d->numframes && d->frames[i].framenum < framenum;i++);
78         if (i < d->numframes && framenum == d->frames[i].framenum)
79         {
80                 f->framenum = framenum;
81                 f->numentities = d->frames[i].endentity - d->frames[i].firstentity;
82                 n = MAX_ENTITY_DATABASE - (d->frames[i].firstentity % MAX_ENTITY_DATABASE);
83                 if (n > f->numentities)
84                         n = f->numentities;
85                 memcpy(f->entitydata, d->entitydata + d->frames[i].firstentity % MAX_ENTITY_DATABASE, sizeof(*f->entitydata) * n);
86                 if (f->numentities > n)
87                         memcpy(f->entitydata + n, d->entitydata, sizeof(*f->entitydata) * (f->numentities - n));
88                 VectorCopy(d->eye, f->eye);
89         }
90         else
91                 f->framenum = -1;
92 }
93
94 // (server and client) adds a entity_frame to the database, for future reference
95 void EntityFrame_AddFrame(entity_database_t *d, entity_frame_t *f)
96 {
97         int n, e;
98         entity_frameinfo_t *info;
99
100         VectorCopy(f->eye, d->eye);
101
102         // figure out how many entity slots are used already
103         if (d->numframes)
104         {
105                 n = d->frames[d->numframes - 1].endentity - d->frames[0].firstentity;
106                 if (n + f->numentities > MAX_ENTITY_DATABASE || d->numframes >= MAX_ENTITY_HISTORY)
107                 {
108                         // ran out of room, dump database
109                         EntityFrame_ClearDatabase(d);
110                 }
111         }
112
113         info = &d->frames[d->numframes];
114         info->framenum = f->framenum;
115         e = -1000;
116         // make sure we check the newly added frame as well, but we haven't incremented numframes yet
117         for (n = 0;n <= d->numframes;n++)
118         {
119                 if (e >= d->frames[n].framenum)
120                 {
121                         if (e == f->framenum)
122                                 Con_Printf("EntityFrame_AddFrame: tried to add out of sequence frame to database\n");
123                         else
124                                 Con_Printf("EntityFrame_AddFrame: out of sequence frames in database\n");
125                         return;
126                 }
127                 e = d->frames[n].framenum;
128         }
129         // if database still has frames after that...
130         if (d->numframes)
131                 info->firstentity = d->frames[d->numframes - 1].endentity;
132         else
133                 info->firstentity = 0;
134         info->endentity = info->firstentity + f->numentities;
135         d->numframes++;
136
137         n = info->firstentity % MAX_ENTITY_DATABASE;
138         e = MAX_ENTITY_DATABASE - n;
139         if (e > f->numentities)
140                 e = f->numentities;
141         memcpy(d->entitydata + n, f->entitydata, sizeof(entity_state_t) * e);
142         if (f->numentities > e)
143                 memcpy(d->entitydata, f->entitydata + e, sizeof(entity_state_t) * (f->numentities - e));
144 }
145
146 // (server) writes a frame to network stream
147 static entity_frame_t deltaframe; // FIXME?
148 void EntityFrame_Write(entity_database_t *d, entity_frame_t *f, sizebuf_t *msg)
149 {
150         int i, onum, bits, number;
151         entity_frame_t *o = &deltaframe;
152         entity_state_t *ent, *delta, baseline;
153
154         EntityFrame_AddFrame(d, f);
155
156         ClearStateToDefault(&baseline);
157         EntityFrame_FetchFrame(d, d->ackframe > 0 ? d->ackframe : -1, o);
158         MSG_WriteByte (msg, svc_entities);
159         MSG_WriteLong (msg, o->framenum);
160         MSG_WriteLong (msg, f->framenum);
161         MSG_WriteFloat (msg, f->eye[0]);
162         MSG_WriteFloat (msg, f->eye[1]);
163         MSG_WriteFloat (msg, f->eye[2]);
164
165         onum = 0;
166         for (i = 0;i < f->numentities;i++)
167         {
168                 ent = f->entitydata + i;
169                 number = ent->number;
170                 for (;onum < o->numentities && o->entitydata[onum].number < number;onum++)
171                 {
172                         // write remove message
173                         MSG_WriteShort(msg, o->entitydata[onum].number | 0x8000);
174                 }
175                 if (onum < o->numentities && (o->entitydata[onum].number == number))
176                 {
177                         // delta from previous frame
178                         delta = o->entitydata + onum;
179                         // advance to next entity in delta frame
180                         onum++;
181                 }
182                 else
183                 {
184                         // delta from baseline
185                         delta = &baseline;
186                 }
187                 bits = 0;
188                 if (ent->flags & RENDER_LOWPRECISION)
189                 {
190                         if ((int) ent->origin[0] != (int) delta->origin[0])
191                                 bits |= E_ORIGIN1;
192                         if ((int) ent->origin[1] != (int) delta->origin[1])
193                                 bits |= E_ORIGIN2;
194                         if ((int) ent->origin[2] != (int) delta->origin[2])
195                                 bits |= E_ORIGIN3;
196                 }
197                 else
198                 {
199                         if (fabs(ent->origin[0] - delta->origin[0]) > 0.01f)
200                                 bits |= E_ORIGIN1;
201                         if (fabs(ent->origin[1] - delta->origin[1]) > 0.01f)
202                                 bits |= E_ORIGIN2;
203                         if (fabs(ent->origin[2] - delta->origin[2]) > 0.01f)
204                                 bits |= E_ORIGIN3;
205                 }
206                 if ((qbyte) (ent->angles[0] * (256.0f / 360.0f)) != (qbyte) (delta->angles[0] * (256.0f / 360.0f)))
207                         bits |= E_ANGLE1;
208                 if ((qbyte) (ent->angles[1] * (256.0f / 360.0f)) != (qbyte) (delta->angles[1] * (256.0f / 360.0f)))
209                         bits |= E_ANGLE2;
210                 if ((qbyte) (ent->angles[2] * (256.0f / 360.0f)) != (qbyte) (delta->angles[2] * (256.0f / 360.0f)))
211                         bits |= E_ANGLE3;
212                 if ((ent->modelindex ^ delta->modelindex) & 0x00FF)
213                         bits |= E_MODEL1;
214                 if ((ent->modelindex ^ delta->modelindex) & 0xFF00)
215                         bits |= E_MODEL2;
216                 if ((ent->frame ^ delta->frame) & 0x00FF)
217                         bits |= E_FRAME1;
218                 if ((ent->frame ^ delta->frame) & 0xFF00)
219                         bits |= E_FRAME2;
220                 if ((ent->effects ^ delta->effects) & 0x00FF)
221                         bits |= E_EFFECTS1;
222                 if ((ent->effects ^ delta->effects) & 0xFF00)
223                         bits |= E_EFFECTS2;
224                 if (ent->colormap != delta->colormap)
225                         bits |= E_COLORMAP;
226                 if (ent->skin != delta->skin)
227                         bits |= E_SKIN;
228                 if (ent->alpha != delta->alpha)
229                         bits |= E_ALPHA;
230                 if (ent->scale != delta->scale)
231                         bits |= E_SCALE;
232                 if (ent->glowsize != delta->glowsize)
233                         bits |= E_GLOWSIZE;
234                 if (ent->glowcolor != delta->glowcolor)
235                         bits |= E_GLOWCOLOR;
236                 if (ent->flags != delta->flags)
237                         bits |= E_FLAGS;
238
239                 if (bits) // don't send anything if it hasn't changed
240                 {
241                         if (bits & 0xFF000000)
242                                 bits |= E_EXTEND3;
243                         if (bits & 0x00FF0000)
244                                 bits |= E_EXTEND2;
245                         if (bits & 0x0000FF00)
246                                 bits |= E_EXTEND1;
247
248                         MSG_WriteShort(msg, number);
249                         MSG_WriteByte(msg, bits & 0xFF);
250                         if (bits & E_EXTEND1)
251                         {
252                                 MSG_WriteByte(msg, (bits >> 8) & 0xFF);
253                                 if (bits & E_EXTEND2)
254                                 {
255                                         MSG_WriteByte(msg, (bits >> 16) & 0xFF);
256                                         if (bits & E_EXTEND3)
257                                                 MSG_WriteByte(msg, (bits >> 24) & 0xFF);
258                                 }
259                         }
260                         // LordHavoc: have to write flags first, as they can modify protocol
261                         if (bits & E_FLAGS)
262                                 MSG_WriteByte(msg, ent->flags);
263                         if (ent->flags & RENDER_LOWPRECISION)
264                         {
265                                 if (bits & E_ORIGIN1)
266                                         MSG_WriteShort(msg, ent->origin[0]);
267                                 if (bits & E_ORIGIN2)
268                                         MSG_WriteShort(msg, ent->origin[1]);
269                                 if (bits & E_ORIGIN3)
270                                         MSG_WriteShort(msg, ent->origin[2]);
271                         }
272                         else
273                         {
274                                 if (bits & E_ORIGIN1)
275                                         MSG_WriteFloat(msg, ent->origin[0]);
276                                 if (bits & E_ORIGIN2)
277                                         MSG_WriteFloat(msg, ent->origin[1]);
278                                 if (bits & E_ORIGIN3)
279                                         MSG_WriteFloat(msg, ent->origin[2]);
280                         }
281                         if (bits & E_ANGLE1)
282                                 MSG_WriteAngle(msg, ent->angles[0]);
283                         if (bits & E_ANGLE2)
284                                 MSG_WriteAngle(msg, ent->angles[1]);
285                         if (bits & E_ANGLE3)
286                                 MSG_WriteAngle(msg, ent->angles[2]);
287                         if (bits & E_MODEL1)
288                                 MSG_WriteByte(msg, ent->modelindex & 0xFF);
289                         if (bits & E_MODEL2)
290                                 MSG_WriteByte(msg, (ent->modelindex >> 8) & 0xFF);
291                         if (bits & E_FRAME1)
292                                 MSG_WriteByte(msg, ent->frame & 0xFF);
293                         if (bits & E_FRAME2)
294                                 MSG_WriteByte(msg, (ent->frame >> 8) & 0xFF);
295                         if (bits & E_EFFECTS1)
296                                 MSG_WriteByte(msg, ent->effects & 0xFF);
297                         if (bits & E_EFFECTS2)
298                                 MSG_WriteByte(msg, (ent->effects >> 8) & 0xFF);
299                         if (bits & E_COLORMAP)
300                                 MSG_WriteByte(msg, ent->colormap);
301                         if (bits & E_SKIN)
302                                 MSG_WriteByte(msg, ent->skin);
303                         if (bits & E_ALPHA)
304                                 MSG_WriteByte(msg, ent->alpha);
305                         if (bits & E_SCALE)
306                                 MSG_WriteByte(msg, ent->scale);
307                         if (bits & E_GLOWSIZE)
308                                 MSG_WriteByte(msg, ent->glowsize);
309                         if (bits & E_GLOWCOLOR)
310                                 MSG_WriteByte(msg, ent->glowcolor);
311                 }
312         }
313         for (;onum < o->numentities;onum++)
314         {
315                 // write remove message
316                 MSG_WriteShort(msg, o->entitydata[onum].number | 0x8000);
317         }
318         MSG_WriteShort(msg, 0xFFFF);
319 }
320
321 // (client) reads a frame from network stream
322 static entity_frame_t framedata; // FIXME?
323 void EntityFrame_Read(entity_database_t *d)
324 {
325         int number, removed, bits;
326         entity_frame_t *f = &framedata, *delta = &deltaframe;
327         entity_state_t *e, baseline, *old, *oldend;
328
329         ClearStateToDefault(&baseline);
330
331         EntityFrame_Clear(f, NULL);
332
333         // read the frame header info
334         f->time = cl.mtime[0];
335         number = MSG_ReadLong();
336         f->framenum = MSG_ReadLong();
337         f->eye[0] = MSG_ReadFloat();
338         f->eye[1] = MSG_ReadFloat();
339         f->eye[2] = MSG_ReadFloat();
340         EntityFrame_AckFrame(d, number);
341         EntityFrame_FetchFrame(d, number, delta);
342         old = delta->entitydata;
343         oldend = old + delta->numentities;
344         // read entities until we hit the magic 0xFFFF end tag
345         while ((number = (unsigned short) MSG_ReadShort()) != 0xFFFF)
346         {
347                 if (msg_badread)
348                         Host_Error("EntityFrame_Read: read error\n");
349                 removed = number & 0x8000;
350                 number &= 0x7FFF;
351                 if (number >= MAX_EDICTS)
352                         Host_Error("EntityFrame_Read: number (%i) >= MAX_EDICTS (%i)\n", number, MAX_EDICTS);
353
354                 // seek to entity, while copying any skipped entities (assume unchanged)
355                 while (old < oldend && old->number < number)
356                 {
357                         if (f->numentities >= MAX_ENTITY_DATABASE)
358                                 Host_Error("EntityFrame_Read: entity list too big\n");
359                         memcpy(f->entitydata + f->numentities, old, sizeof(entity_state_t));
360                         f->entitydata[f->numentities].time = cl.mtime[0];
361                         old++;
362                         f->numentities++;
363                 }
364                 if (removed)
365                 {
366                         if (old < oldend && old->number == number)
367                                 old++;
368                         else
369                                 Con_Printf("EntityFrame_Read: REMOVE on unused entity %i\n", number);
370                 }
371                 else
372                 {
373                         if (f->numentities >= MAX_ENTITY_DATABASE)
374                                 Host_Error("EntityFrame_Read: entity list too big\n");
375
376                         // reserve this slot
377                         e = f->entitydata + f->numentities++;
378
379                         if (old < oldend && old->number == number)
380                         {
381                                 // delta from old entity
382                                 memcpy(e, old++, sizeof(*e));
383                         }
384                         else
385                         {
386                                 // delta from baseline
387                                 memcpy(e, &baseline, sizeof(*e));
388                         }
389
390                         e->active = true;
391                         e->time = cl.mtime[0];
392                         e->number = number;
393
394                         bits = MSG_ReadByte();
395                         if (bits & E_EXTEND1)
396                         {
397                                 bits |= MSG_ReadByte() << 8;
398                                 if (bits & E_EXTEND2)
399                                 {
400                                         bits |= MSG_ReadByte() << 16;
401                                         if (bits & E_EXTEND3)
402                                                 bits |= MSG_ReadByte() << 24;
403                                 }
404                         }
405
406                         if (dpprotocol == DPPROTOCOL_VERSION2)
407                         {
408                                 if (bits & E_ORIGIN1)
409                                         e->origin[0] = (signed short) MSG_ReadShort();
410                                 if (bits & E_ORIGIN2)
411                                         e->origin[1] = (signed short) MSG_ReadShort();
412                                 if (bits & E_ORIGIN3)
413                                         e->origin[2] = (signed short) MSG_ReadShort();
414                         }
415                         else
416                         {
417                                 if (bits & E_FLAGS)
418                                         e->flags = MSG_ReadByte();
419                                 if (e->flags & RENDER_LOWPRECISION || dpprotocol == DPPROTOCOL_VERSION2)
420                                 {
421                                         if (bits & E_ORIGIN1)
422                                                 e->origin[0] = (signed short) MSG_ReadShort();
423                                         if (bits & E_ORIGIN2)
424                                                 e->origin[1] = (signed short) MSG_ReadShort();
425                                         if (bits & E_ORIGIN3)
426                                                 e->origin[2] = (signed short) MSG_ReadShort();
427                                 }
428                                 else
429                                 {
430                                         if (bits & E_ORIGIN1)
431                                                 e->origin[0] = MSG_ReadFloat();
432                                         if (bits & E_ORIGIN2)
433                                                 e->origin[1] = MSG_ReadFloat();
434                                         if (bits & E_ORIGIN3)
435                                                 e->origin[2] = MSG_ReadFloat();
436                                 }
437                         }
438                         if (bits & E_ANGLE1)
439                                 e->angles[0] = MSG_ReadAngle();
440                         if (bits & E_ANGLE2)
441                                 e->angles[1] = MSG_ReadAngle();
442                         if (bits & E_ANGLE3)
443                                 e->angles[2] = MSG_ReadAngle();
444                         if (bits & E_MODEL1)
445                                 e->modelindex = (e->modelindex & 0xFF00) | (unsigned int) MSG_ReadByte();
446                         if (bits & E_MODEL2)
447                                 e->modelindex = (e->modelindex & 0x00FF) | ((unsigned int) MSG_ReadByte() << 8);
448                         if (bits & E_FRAME1)
449                                 e->frame = (e->frame & 0xFF00) | (unsigned int) MSG_ReadByte();
450                         if (bits & E_FRAME2)
451                                 e->frame = (e->frame & 0x00FF) | ((unsigned int) MSG_ReadByte() << 8);
452                         if (bits & E_EFFECTS1)
453                                 e->effects = (e->effects & 0xFF00) | (unsigned int) MSG_ReadByte();
454                         if (bits & E_EFFECTS2)
455                                 e->effects = (e->effects & 0x00FF) | ((unsigned int) MSG_ReadByte() << 8);
456                         if (bits & E_COLORMAP)
457                                 e->colormap = MSG_ReadByte();
458                         if (bits & E_SKIN)
459                                 e->skin = MSG_ReadByte();
460                         if (bits & E_ALPHA)
461                                 e->alpha = MSG_ReadByte();
462                         if (bits & E_SCALE)
463                                 e->scale = MSG_ReadByte();
464                         if (bits & E_GLOWSIZE)
465                                 e->glowsize = MSG_ReadByte();
466                         if (bits & E_GLOWCOLOR)
467                                 e->glowcolor = MSG_ReadByte();
468                         if (dpprotocol == DPPROTOCOL_VERSION2)
469                                 if (bits & E_FLAGS)
470                                         e->flags = MSG_ReadByte();
471                 }
472         }
473         while (old < oldend)
474         {
475                 if (f->numentities >= MAX_ENTITY_DATABASE)
476                         Host_Error("EntityFrame_Read: entity list too big\n");
477                 memcpy(f->entitydata + f->numentities, old, sizeof(entity_state_t));
478                 f->entitydata[f->numentities].time = cl.mtime[0];
479                 old++;
480                 f->numentities++;
481         }
482         EntityFrame_AddFrame(d, f);
483 }
484
485
486 // (client) returns the frame number of the most recent frame recieved
487 int EntityFrame_MostRecentlyRecievedFrameNum(entity_database_t *d)
488 {
489         if (d->numframes)
490                 return d->frames[d->numframes - 1].framenum;
491         else
492                 return -1;
493 }
494