src/utils/flactimer/main.cpp

   1 /* flactimer - Runs a command and prints timing information
   2  * Copyright (C) 2007,2008,2009  Josh Coalson
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <string.h>
  21 #include <windows.h>
  22
  23 #define int64_t __int64
  24 #define uint64_t unsigned int64_t
  25
  26 static inline uint64_t time2nsec(const FILETIME &t)
  27 {
  28         uint64_t n = t.dwHighDateTime;
  29         n <<= 32;
  30         n |= (uint64_t)t.dwLowDateTime;
  31         return n * 100;
  32 }
  33
  34 static void printtime(FILE *fout, uint64_t nsec, uint64_t total)
  35 {
  36         unsigned pct = (unsigned)(100.0 * ((double)(int64_t)nsec / (double)(int64_t)total));
  37         uint64_t msec = nsec / 1000000; nsec -= msec * 1000000;
  38         uint64_t sec = msec / 1000; msec -= sec * 1000;
  39         uint64_t min = sec / 60; sec -= min * 60;
  40         uint64_t hour = min / 60; min -= hour * 60;
  41         fprintf(fout, " %5u.%03u = %02u:%02u:%02u.%03u = %3u%%\n",
  42                 (unsigned)((hour*60+min)*60+sec),
  43                 (unsigned)msec,
  44                 (unsigned)hour,
  45                 (unsigned)min,
  46                 (unsigned)sec,
  47                 (unsigned)msec,
  48                 pct
  49         );
  50 }
  51
  52 int main(int argc, char *argv[])
  53 {
  54         const char *usage = "usage: flactimer [-1 | -2 | -o outputfile] command\n";
  55         FILE *fout = stderr;
  56
  57         if(argc == 1 || (argc > 1 && 0 == strcmp(argv[1], "-h"))) {
  58                 fprintf(stderr, usage);
  59                 return 0;
  60         }
  61         argv++;
  62         argc--;
  63         if(0 == strcmp(argv[0], "-1") || 0 == strcmp(argv[0], "/1")) {
  64                 fout = stdout;
  65                 argv++;
  66                 argc--;
  67         }
  68         else if(0 == strcmp(argv[0], "-2") || 0 == strcmp(argv[0], "/2")) {
  69                 fout = stdout;
  70                 argv++;
  71                 argc--;
  72         }
  73         else if(0 == strcmp(argv[0], "-o")) {
  74                 if(argc < 2) {
  75                         fprintf(stderr, usage);
  76                         return 1;
  77                 }
  78                 fout = fopen(argv[1], "w");
  79                 if(!fout) {
  80                         fprintf(fout, "ERROR opening file %s for writing\n", argv[1]);
  81                         return 1;
  82                 }
  83                 argv += 2;
  84                 argc -= 2;
  85         }
  86         if(argc <= 0) {
  87                 fprintf(fout, "ERROR, no command!\n\n");
  88                 fprintf(fout, usage);
  89                 fclose(fout);
  90                 return 1;
  91         }
  92
  93         // improvement: double-quote all args
  94         int i, n = 0;
  95         for(i = 0; i < argc; i++) {
  96                 if(i > 0)
  97                         n++;
  98                 n += strlen(argv[i]);
  99         }
 100         char *args = (char*)malloc(n+1);
 101         if(!args) {
 102                 fprintf(fout, "ERROR, no memory\n");
 103                 fclose(fout);
 104                 return 1;
 105         }
 106         args[0] = '\0';
 107         for(i = 0; i < argc; i++) {
 108                 if(i > 0)
 109                         safe_strncat(args, sizeof(args), " ");
 110                 safe_strncat(args, sizeof(args), argv[i]);
 111         }
 112
 113         //fprintf(stderr, "@@@ cmd=[%s] args=[%s]\n", argv[0], args);
 114
 115         STARTUPINFO si;
 116         GetStartupInfo(&si);
 117
 118         DWORD wallclock_msec = GetTickCount();
 119
 120         PROCESS_INFORMATION pi;
 121         BOOL ok = CreateProcess(
 122                 argv[0], // lpApplicationName
 123                 args, // lpCommandLine
 124                 NULL, // lpProcessAttributes
 125                 NULL, // lpThreadAttributes
 126                 FALSE, // bInheritHandles
 127                 0, // dwCreationFlags
 128                 NULL, // lpEnvironment
 129                 NULL, // lpCurrentDirectory
 130                 &si, // lpStartupInfo (inherit from this proc?)
 131                 &pi // lpProcessInformation
 132         );
 133
 134         if(!ok) {
 135                 fprintf(fout, "ERROR running command\n");
 136                 free(args); //@@@ ok to free here or have to wait to wait till process is reaped?
 137                 fclose(fout);
 138                 return 1;
 139         }
 140
 141         //fprintf(stderr, "@@@ waiting...\n");
 142         WaitForSingleObject(pi.hProcess, INFINITE);
 143         //fprintf(stderr, "@@@ done\n");
 144
 145         wallclock_msec = GetTickCount() - wallclock_msec;
 146
 147         FILETIME creation_time;
 148         FILETIME exit_time;
 149         FILETIME kernel_time;
 150         FILETIME user_time;
 151         if(!GetProcessTimes(pi.hProcess, &creation_time, &exit_time, &kernel_time, &user_time)) {
 152                 fprintf(fout, "ERROR getting time info\n");
 153                 free(args); //@@@ ok to free here or have to wait to wait till process is reaped?
 154                 fclose(fout);
 155                 return 1;
 156         }
 157         uint64_t kernel_nsec = time2nsec(kernel_time);
 158         uint64_t user_nsec = time2nsec(user_time);
 159
 160         fprintf(fout, "Kernel Time  = "); printtime(fout, kernel_nsec, (uint64_t)wallclock_msec * 1000000);
 161         fprintf(fout, "User Time    = "); printtime(fout, user_nsec, (uint64_t)wallclock_msec * 1000000);
 162         fprintf(fout, "Process Time = "); printtime(fout, kernel_nsec+user_nsec, (uint64_t)wallclock_msec * 1000000);
 163         fprintf(fout, "Global Time  = "); printtime(fout, (uint64_t)wallclock_msec * 1000000, (uint64_t)wallclock_msec * 1000000);
 164
 165         CloseHandle(pi.hThread);
 166         CloseHandle(pi.hProcess);
 167
 168         free(args); //@@@ always causes crash, maybe CreateProcess takes ownership?
 169         fclose(fout);
 170         return 0;
 171 }