15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2009 The RE2 Authors.  All Rights Reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// license that can be found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdio.h>
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef WIN32
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/resource.h>
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "util/test.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)DEFINE_string(test_tmpdir, "/var/tmp", "temp directory");
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct Test {
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  void (*fn)(void);
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const char *name;
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static Test tests[10000];
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static int ntests;
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RegisterTest(void (*fn)(void), const char *name) {
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  tests[ntests].fn = fn;
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  tests[ntests++].name = name;
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace re2 {
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int64 VirtualProcessSize() {
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef WIN32
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  struct rusage ru;
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  getrusage(RUSAGE_SELF, &ru);
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return (int64)ru.ru_maxrss*1024;
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return 0;
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace re2
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int main(int argc, char **argv) {
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < ntests; i++) {
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    printf("%s\n", tests[i].name);
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    tests[i].fn();
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  printf("PASS\n");
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return 0;
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
46