mirror of https://github.com/rui314/chibicc
52 lines
961 B
C
52 lines
961 B
C
#include "test.h"
|
|
#include <stdarg.h>
|
|
|
|
int sum1(int x, ...) {
|
|
va_list ap;
|
|
va_start(ap, x);
|
|
|
|
for (;;) {
|
|
int y = va_arg(ap, int);
|
|
if (y == 0)
|
|
return x;
|
|
x += y;
|
|
}
|
|
}
|
|
|
|
int sum2(int x, ...) {
|
|
va_list ap;
|
|
va_start(ap, x);
|
|
|
|
for (;;) {
|
|
double y = va_arg(ap, double);
|
|
x += y;
|
|
|
|
int z = va_arg(ap, int);
|
|
if (z == 0)
|
|
return x;
|
|
x += z;
|
|
}
|
|
}
|
|
|
|
void fmt(char *buf, char *fmt, ...) {
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
|
|
va_list ap2;
|
|
va_copy(ap2, ap);
|
|
vsprintf(buf, fmt, ap2);
|
|
va_end(buf);
|
|
}
|
|
|
|
int main() {
|
|
ASSERT(6, sum1(1, 2, 3, 0));
|
|
ASSERT(55, sum1(1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 0));
|
|
ASSERT(21, sum2(1, 2.0, 3, 4.0, 5, 6.0, 0));
|
|
ASSERT(21, sum2(1, 2.0, 3, 4.0, 5, 6.0, 0));
|
|
ASSERT(210, sum2(1, 2.0, 3, 4.0, 5, 6.0, 7, 8.0, 9, 10.0, 11, 12.0, 13, 14.0, 15, 16.0, 17, 18.0, 19, 20.0, 0));
|
|
ASSERT(0, ({ char buf[100]; fmt(buf, "%d %d", 2, 3); strcmp(buf, "2 3"); }));
|
|
|
|
printf("OK\n");
|
|
return 0;
|
|
}
|