]> git.pond.sub.org Git - empserver/blob - src/lib/commands/retr.c
retreat lretreat: Fix crash when getstarg() fails
[empserver] / src / lib / commands / retr.c
1 /*
2  *  Empire - A multi-player, client/server Internet based war game.
3  *  Copyright (C) 1986-2015, Dave Pare, Jeff Bailey, Thomas Ruschak,
4  *                Ken Stevens, Steve McClure, Markus Armbruster
5  *
6  *  Empire is free software: you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation, either version 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  *
19  *  ---
20  *
21  *  See files README, COPYING and CREDITS in the root of the source
22  *  tree for related information and legal notices.  It is expected
23  *  that future projects/authors will amend these files as needed.
24  *
25  *  ---
26  *
27  *  retr.c: Set retreat conditionals for ships and land units
28  *
29  *  Known contributors to this file:
30  *     Ken Stevens, 1995
31  *     Steve McClure, 2000
32  *     Markus Armbruster, 2008-2014
33  */
34
35 #include <config.h>
36
37 #include <ctype.h>
38 #include "commands.h"
39 #include "empobj.h"
40 #include "land.h"
41 #include "path.h"
42 #include "retreat.h"
43 #include "ship.h"
44
45 /*
46  * Retreat flag characters
47  * 'X' means flag is not available
48  * Must agree with RET_ defines.
49  */
50 static char shp_rflagsc[] = "Xitshbdu";
51 static char lnd_rflagsc[] = "XiXXhbXX";
52
53 static int retreat(int);
54
55 int
56 retr(void)
57 {
58     return retreat(EF_SHIP);
59 }
60
61 int
62 lretr(void)
63 {
64     return retreat(EF_LAND);
65 }
66
67 static int
68 retreat(int type)
69 {
70     char *pq, *fl;
71     int nunits;
72     struct nstr_item ni;
73     union empobj_storage unit;
74     int rflags, ch, j;
75     unsigned i;
76     char *rflagsc, *p, *name, *rpath, *what;
77     int *rflagsp;
78     char buf1[1024];
79     char buf2[1024];
80
81     if (CANT_HAPPEN(type != EF_LAND && type != EF_SHIP))
82         type = EF_SHIP;
83     rflagsc = type == EF_SHIP ? shp_rflagsc : lnd_rflagsc;
84
85     if (!snxtitem(&ni, type, player->argp[1], NULL))
86         return RET_SYN;
87     nunits = 0;
88     if (player->argp[2] != NULL) {
89         pq = getstarg(player->argp[2], "Retreat path? ", buf1);
90         if (!pq)
91             return RET_SYN;
92         for (i = 0; i < RET_LEN - 1 && pq[i]; i++) {
93             if (chkdir(pq[i], DIR_STOP, DIR_LAST) < 0) {
94                 pr("'%c' is not a valid direction...\n", pq[i]);
95                 direrr(NULL, NULL, NULL);
96                 return RET_SYN;
97             }
98         }
99     } else
100         pq = NULL;
101
102     rflags = 0;
103     if (pq != NULL) {
104     again:
105         fl = getstarg(player->argp[3],
106                       "Retreat conditions ('?' to list available ones)? ",
107                       buf2);
108         if (!fl)
109             return RET_SYN;
110
111         for (i = 0; fl[i]; i++) {
112             ch = tolower(fl[i]);
113             if (ch == 'c') {
114                 *pq = 0;
115                 break;
116             }
117             if (ch == '?' && !player->argp[3]) {
118                 for (j = 1; rflagsc[j]; j++) {
119                     if (rflagsc[j] != 'X')
120                         pr("%c\tretreat when %s\n",
121                            rflagsc[j],
122                            symbol_by_value(1 << j, retreat_flags));
123                 }
124                 pr("c\tcancel retreat order\n");
125                 goto again;
126             }
127             p = strchr(rflagsc, ch);
128             if (!p) {
129                 pr("Bad retreat condition '%c'\n", fl[i]);
130                 return RET_SYN;
131             }
132             rflags |= 1 << (p - rflagsc);
133         }
134         if (*pq && !rflags)
135             return RET_SYN;
136         if (ni.sel == NS_GROUP && ni.group)
137             rflags |= RET_GROUP;
138         if (!*pq)
139             rflags = 0;
140     }
141
142     while (nxtitem(&ni, &unit)) {
143         if (!player->owner || unit.gen.own == 0)
144             continue;
145         if (type == EF_SHIP) {
146             if (nunits++ == 0) {
147                 if (player->god)
148                     pr("own ");
149                 pr("shp#     ship type       x,y   fl path       as flt?  flags\n");
150             }
151             name = mchr[unit.ship.shp_type].m_name;
152             rpath = unit.ship.shp_rpath;
153             rflagsp = &unit.ship.shp_rflags;
154         } else {
155             if (nunits++ == 0) {
156                 if (player->god)
157                     pr("own ");
158                 pr("lnd#     unit type       x,y   ar path       as army? flags\n");
159             }
160             name = lchr[unit.land.lnd_type].l_name;
161             rpath = unit.land.lnd_rpath;
162             rflagsp = &unit.land.lnd_rflags;
163         }
164         if (pq) {
165             strncpy(rpath, pq, RET_LEN - 1);
166             *rflagsp = rflags;
167             put_empobj(type, unit.gen.uid, &unit);
168         }
169         if (player->god)
170             pr("%3d ", unit.gen.own);
171         pr("%4d ", ni.cur);
172         pr("%-16.16s ", name);
173         prxy("%4d,%-4d ", unit.gen.x, unit.gen.y);
174         pr("%1.1s", &unit.gen.group);
175         pr(" %-11s", rpath);
176         rflags = *rflagsp;
177         if (rflags & RET_GROUP)
178             pr("Yes      ");
179         else
180             pr("         ");
181         for (j = 1; rflagsc[j]; j++) {
182             if ((1 << j) & rflags) {
183                 if (CANT_HAPPEN(rflagsc[j] == 'X'))
184                     continue;
185                 pr("%c", rflagsc[j]);
186             }
187         }
188         pr("\n");
189     }
190     what = type == EF_SHIP ? "ship" : "unit";
191     if (nunits == 0) {
192         if (player->argp[1])
193             pr("%s: No %s(s)\n", player->argp[1], what);
194         else
195             pr("%s: No %s(s)\n", "", what);
196         return RET_FAIL;
197     } else
198         pr("%d %s%s\n", nunits, what, splur(nunits));
199     return RET_OK;
200 }