mirror of
https://github.com/tmux/tmux.git
synced 2026-05-08 14:52:17 +00:00
Add time limit to a couple of other loops.
This commit is contained in:
74
format.c
74
format.c
@@ -4125,15 +4125,33 @@ found:
|
||||
return (found);
|
||||
}
|
||||
|
||||
/* Check if format has not taken too long. */
|
||||
static int
|
||||
format_check_time(struct format_expand_state *es)
|
||||
{
|
||||
uint64_t t = get_timer();
|
||||
|
||||
if (t - es->start_time < FORMAT_TIME_LIMIT)
|
||||
return (1);
|
||||
t -= es->start_time;
|
||||
|
||||
format_log(es, "reached time limit (%llu)", (unsigned long long)t);
|
||||
return (0);
|
||||
}
|
||||
|
||||
/* Unescape escaped characters. */
|
||||
static char *
|
||||
format_unescape(const char *s)
|
||||
format_unescape(struct format_expand_state *es, const char *s)
|
||||
{
|
||||
char *out, *cp;
|
||||
int brackets = 0;
|
||||
|
||||
cp = out = xmalloc(strlen(s) + 1);
|
||||
for (; *s != '\0'; s++) {
|
||||
if (!format_check_time(es)){
|
||||
free(out);
|
||||
return (xstrdup(""));
|
||||
}
|
||||
if (*s == '#' && s[1] == '{')
|
||||
brackets++;
|
||||
if (brackets == 0 &&
|
||||
@@ -4152,13 +4170,17 @@ format_unescape(const char *s)
|
||||
|
||||
/* Remove escaped characters. */
|
||||
static char *
|
||||
format_strip(const char *s)
|
||||
format_strip(struct format_expand_state *es, const char *s)
|
||||
{
|
||||
char *out, *cp;
|
||||
int brackets = 0;
|
||||
|
||||
cp = out = xmalloc(strlen(s) + 1);
|
||||
for (; *s != '\0'; s++) {
|
||||
if (!format_check_time(es)){
|
||||
free(out);
|
||||
return (xstrdup(""));
|
||||
}
|
||||
if (*s == '#' && s[1] == '{')
|
||||
brackets++;
|
||||
if (*s == '#' && strchr(",#{}:", s[1]) != NULL) {
|
||||
@@ -4174,13 +4196,16 @@ format_strip(const char *s)
|
||||
return (out);
|
||||
}
|
||||
|
||||
|
||||
/* Skip until end. */
|
||||
const char *
|
||||
format_skip(const char *s, const char *end)
|
||||
static const char *
|
||||
format_skip1(struct format_expand_state *es, const char *s, const char *end)
|
||||
{
|
||||
int brackets = 0;
|
||||
|
||||
for (; *s != '\0'; s++) {
|
||||
if (es != NULL && !format_check_time(es))
|
||||
return (NULL);
|
||||
if (*s == '#' && s[1] == '{')
|
||||
brackets++;
|
||||
if (*s == '#' &&
|
||||
@@ -4199,6 +4224,13 @@ format_skip(const char *s, const char *end)
|
||||
return (s);
|
||||
}
|
||||
|
||||
/* Skip until end. */
|
||||
const char *
|
||||
format_skip(const char *s, const char *end)
|
||||
{
|
||||
return (format_skip1(NULL, s, end));
|
||||
}
|
||||
|
||||
/* Return left and right alternatives separated by commas. */
|
||||
static int
|
||||
format_choose(struct format_expand_state *es, const char *s, char **left,
|
||||
@@ -4207,7 +4239,7 @@ format_choose(struct format_expand_state *es, const char *s, char **left,
|
||||
const char *cp;
|
||||
char *left0, *right0;
|
||||
|
||||
cp = format_skip(s, ",");
|
||||
cp = format_skip1(es, s, ",");
|
||||
if (cp == NULL)
|
||||
return (-1);
|
||||
left0 = xstrndup(s, cp - s);
|
||||
@@ -4225,20 +4257,6 @@ format_choose(struct format_expand_state *es, const char *s, char **left,
|
||||
return (0);
|
||||
}
|
||||
|
||||
/* Check format has not taken too lon. */
|
||||
static int
|
||||
format_check_time(struct format_expand_state *es)
|
||||
{
|
||||
uint64_t t = get_timer();
|
||||
|
||||
if (t - es->start_time < FORMAT_TIME_LIMIT)
|
||||
return (1);
|
||||
t -= es->start_time;
|
||||
|
||||
format_log(es, "reached time limit (%llu)", (unsigned long long)t);
|
||||
return (0);
|
||||
}
|
||||
|
||||
/* Is this true? */
|
||||
int
|
||||
format_true(const char *s)
|
||||
@@ -4352,7 +4370,7 @@ format_build_modifiers(struct format_expand_state *es, const char **s,
|
||||
|
||||
/* Single argument with no wrapper character. */
|
||||
if (!ispunct((u_char)cp[1]) || cp[1] == '-') {
|
||||
end = format_skip(cp + 1, ":;");
|
||||
end = format_skip1(es, cp + 1, ":;");
|
||||
if (end == NULL)
|
||||
break;
|
||||
|
||||
@@ -4375,7 +4393,7 @@ format_build_modifiers(struct format_expand_state *es, const char **s,
|
||||
cp++;
|
||||
break;
|
||||
}
|
||||
end = format_skip(cp + 1, last);
|
||||
end = format_skip1(es, cp + 1, last);
|
||||
if (end == NULL)
|
||||
break;
|
||||
cp++;
|
||||
@@ -4489,7 +4507,7 @@ format_bool_op_n(struct format_expand_state *es, const char *fmt, int and)
|
||||
cp1 = fmt;
|
||||
|
||||
while (and ? result : !result) {
|
||||
cp2 = format_skip(cp1, ",");
|
||||
cp2 = format_skip1(es, cp1, ",");
|
||||
|
||||
if (cp2 == NULL)
|
||||
raw = xstrdup(cp1);
|
||||
@@ -5069,7 +5087,7 @@ format_replace(struct format_expand_state *es, const char *key, size_t keylen,
|
||||
else if (fm->argc >= 2 &&
|
||||
strchr(fm->argv[0], 'f') != NULL) {
|
||||
free(time_format);
|
||||
time_format = format_strip(fm->argv[1]);
|
||||
time_format = format_strip(es, fm->argv[1]);
|
||||
}
|
||||
break;
|
||||
case 'q':
|
||||
@@ -5185,7 +5203,7 @@ format_replace(struct format_expand_state *es, const char *key, size_t keylen,
|
||||
/* Is this a literal string? */
|
||||
if (modifiers & FORMAT_LITERAL) {
|
||||
format_log(es, "literal string is '%s'", copy);
|
||||
value = format_unescape(copy);
|
||||
value = format_unescape(es, copy);
|
||||
goto done;
|
||||
}
|
||||
|
||||
@@ -5338,7 +5356,7 @@ format_replace(struct format_expand_state *es, const char *key, size_t keylen,
|
||||
*/
|
||||
cp = copy + 1;
|
||||
while (1) {
|
||||
cp2 = format_skip(cp, ",");
|
||||
cp2 = format_skip1(es, cp, ",");
|
||||
if (cp2 == NULL) {
|
||||
format_log(es,
|
||||
"no condition matched in '%s'; using last "
|
||||
@@ -5373,7 +5391,7 @@ format_replace(struct format_expand_state *es, const char *key, size_t keylen,
|
||||
}
|
||||
|
||||
cp = cp2 + 1;
|
||||
cp2 = format_skip(cp, ",");
|
||||
cp2 = format_skip1(es, cp, ",");
|
||||
if (format_true(found)) {
|
||||
format_log(es, "condition '%s' is true",
|
||||
condition);
|
||||
@@ -5622,7 +5640,7 @@ format_expand1(struct format_expand_state *es, const char *fmt)
|
||||
fmt += n + 1;
|
||||
continue;
|
||||
case '{':
|
||||
ptr = format_skip((char *)fmt - 2, "}");
|
||||
ptr = format_skip1(es, (char *)fmt - 2, "}");
|
||||
if (ptr == NULL)
|
||||
break;
|
||||
n = ptr - fmt;
|
||||
@@ -5645,7 +5663,7 @@ format_expand1(struct format_expand_state *es, const char *fmt)
|
||||
n++;
|
||||
}
|
||||
if (*ptr == '[') {
|
||||
style_end = format_skip(fmt - 2, "]");
|
||||
style_end = format_skip1(es, fmt - 2, "]");
|
||||
format_log(es, "found #*%zu[", n);
|
||||
while (len - off < n + 2) {
|
||||
buf = xreallocarray(buf, 2, len);
|
||||
|
||||
Reference in New Issue
Block a user