blob: b75b1a32fbd88e69b56c8c5d83ff170d4d781b44 [file] [log] [blame]
Fred Drake8c081a12001-10-12 20:57:55 +00001/*
2 * This is the High Performance Python Profiler portion of HotShot.
3 */
4
Tim Peters885d4572001-11-28 20:27:42 +00005#include "Python.h"
6#include "compile.h"
7#include "eval.h"
8#include "frameobject.h"
9#include "structmember.h"
Fred Drake8c081a12001-10-12 20:57:55 +000010
Fred Drake8c081a12001-10-12 20:57:55 +000011/*
12 * Which timer to use should be made more configurable, but that should not
Tim Petersfeab23f2001-10-13 00:11:10 +000013 * be difficult. This will do for now.
Fred Drake8c081a12001-10-12 20:57:55 +000014 */
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000015#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +000016#include <windows.h>
Tim Peters1566a172001-10-12 22:08:39 +000017#include <direct.h> /* for getcwd() */
Tim Peters7d99ff22001-10-13 07:37:52 +000018typedef __int64 hs_time;
19#define GETTIMEOFDAY(P_HS_TIME) \
20 { LARGE_INTEGER _temp; \
21 QueryPerformanceCounter(&_temp); \
22 *(P_HS_TIME) = _temp.QuadPart; }
23
Tim Petersfeab23f2001-10-13 00:11:10 +000024
Fred Drake8c081a12001-10-12 20:57:55 +000025#else
26#ifndef HAVE_GETTIMEOFDAY
27#error "This module requires gettimeofday() on non-Windows platforms!"
28#endif
Jack Janseneddc1442003-11-20 01:44:59 +000029#if (defined(PYOS_OS2) && defined(PYCC_GCC))
Jack Jansen963659a2001-10-23 22:26:16 +000030#include <sys/time.h>
31#else
Fred Drake8c081a12001-10-12 20:57:55 +000032#include <sys/resource.h>
33#include <sys/times.h>
Jack Jansen963659a2001-10-23 22:26:16 +000034#endif
Fred Drake8c081a12001-10-12 20:57:55 +000035typedef struct timeval hs_time;
36#endif
37
38#if !defined(__cplusplus) && !defined(inline)
39#ifdef __GNUC__
40#define inline __inline
41#endif
42#endif
43
44#ifndef inline
45#define inline
46#endif
47
48#define BUFFERSIZE 10240
49
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000050#if defined(PYOS_OS2) && defined(PYCC_GCC)
51#define PATH_MAX 260
52#endif
53
Martin v. Löwis8eb92a02002-09-19 08:03:21 +000054#if defined(__sgi) && _COMPILER_VERSION>700 && !defined(PATH_MAX)
55/* fix PATH_MAX not being defined with MIPSPro 7.x
56 if mode is ANSI C (default) */
57#define PATH_MAX 1024
58#endif
59
Tim Peters1566a172001-10-12 22:08:39 +000060#ifndef PATH_MAX
61# ifdef MAX_PATH
62# define PATH_MAX MAX_PATH
Martin v. Löwis21ee4092002-09-30 16:19:48 +000063# elif defined (_POSIX_PATH_MAX)
64# define PATH_MAX _POSIX_PATH_MAX
Tim Peters1566a172001-10-12 22:08:39 +000065# else
66# error "Need a defn. for PATH_MAX in _hotshot.c"
67# endif
68#endif
69
Fred Drake8c081a12001-10-12 20:57:55 +000070typedef struct {
71 PyObject_HEAD
72 PyObject *filemap;
73 PyObject *logfilename;
74 int index;
75 unsigned char buffer[BUFFERSIZE];
76 FILE *logfp;
77 int lineevents;
78 int linetimings;
Fred Drake30d1c752001-10-15 22:11:02 +000079 int frametimings;
Fred Drake8c081a12001-10-12 20:57:55 +000080 /* size_t filled; */
81 int active;
82 int next_fileno;
Fred Drake8c081a12001-10-12 20:57:55 +000083 hs_time prev_timeofday;
84} ProfilerObject;
85
86typedef struct {
87 PyObject_HEAD
Fred Drake4c2e1af2001-10-29 20:45:57 +000088 PyObject *info;
Fred Drake8c081a12001-10-12 20:57:55 +000089 FILE *logfp;
Fred Drake8c081a12001-10-12 20:57:55 +000090 int linetimings;
Fred Drake30d1c752001-10-15 22:11:02 +000091 int frametimings;
Fred Drake8c081a12001-10-12 20:57:55 +000092} LogReaderObject;
93
94static PyObject * ProfilerError = NULL;
95
96
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000097#ifndef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +000098#ifdef GETTIMEOFDAY_NO_TZ
99#define GETTIMEOFDAY(ptv) gettimeofday((ptv))
100#else
101#define GETTIMEOFDAY(ptv) gettimeofday((ptv), (struct timezone *)NULL)
102#endif
103#endif
104
105
106/* The log reader... */
107
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000108PyDoc_STRVAR(logreader_close__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +0000109"close()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000110"Close the log file, preventing additional records from being read.");
Fred Drake8c081a12001-10-12 20:57:55 +0000111
112static PyObject *
113logreader_close(LogReaderObject *self, PyObject *args)
114{
Fred Drake666bf522002-07-18 19:11:44 +0000115 if (self->logfp != NULL) {
116 fclose(self->logfp);
117 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +0000118 }
Fred Drake666bf522002-07-18 19:11:44 +0000119 Py_INCREF(Py_None);
120
121 return Py_None;
122}
123
124PyDoc_STRVAR(logreader_fileno__doc__,
125"fileno() -> file descriptor\n"
126"Returns the file descriptor for the log file, if open.\n"
127"Raises ValueError if the log file is closed.");
128
129static PyObject *
130logreader_fileno(LogReaderObject *self)
131{
132 if (self->logfp == NULL) {
133 PyErr_SetString(PyExc_ValueError,
134 "logreader's file object already closed");
135 return NULL;
136 }
137 return PyInt_FromLong(fileno(self->logfp));
Fred Drake8c081a12001-10-12 20:57:55 +0000138}
139
Fred Drake8c081a12001-10-12 20:57:55 +0000140
141/* Log File Format
142 * ---------------
143 *
144 * The log file consists of a sequence of variable-length records.
145 * Each record is identified with a record type identifier in two
146 * bits of the first byte. The two bits are the "least significant"
147 * bits of the byte.
148 *
149 * Low bits: Opcode: Meaning:
150 * 0x00 ENTER enter a frame
151 * 0x01 EXIT exit a frame
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000152 * 0x02 LINENO execution moved onto a different line
Fred Drake8c081a12001-10-12 20:57:55 +0000153 * 0x03 OTHER more bits are needed to deecode
154 *
155 * If the type is OTHER, the record is not packed so tightly, and the
156 * remaining bits are used to disambiguate the record type. These
157 * records are not used as frequently so compaction is not an issue.
158 * Each of the first three record types has a highly tailored
159 * structure that allows it to be packed tightly.
160 *
161 * The OTHER records have the following identifiers:
162 *
163 * First byte: Opcode: Meaning:
164 * 0x13 ADD_INFO define a key/value pair
165 * 0x23 DEFINE_FILE define an int->filename mapping
166 * 0x33 LINE_TIMES indicates if LINENO events have tdeltas
Fred Drake30d1c752001-10-15 22:11:02 +0000167 * 0x43 DEFINE_FUNC define a (fileno,lineno)->funcname mapping
168 * 0x53 FRAME_TIMES indicates if ENTER/EXIT events have tdeltas
Fred Drake8c081a12001-10-12 20:57:55 +0000169 *
170 * Packed Integers
171 *
172 * "Packed integers" are non-negative integer values encoded as a
173 * sequence of bytes. Each byte is encoded such that the most
174 * significant bit is set if the next byte is also part of the
175 * integer. Each byte provides bits to the least-significant end of
176 * the result; the accumulated value must be shifted up to place the
177 * new bits into the result.
178 *
179 * "Modified packed integers" are packed integers where only a portion
180 * of the first byte is used. In the rest of the specification, these
181 * are referred to as "MPI(n,name)", where "n" is the number of bits
182 * discarded from the least-signicant positions of the byte, and
183 * "name" is a name being given to those "discarded" bits, since they
184 * are a field themselves.
185 *
186 * ENTER records:
187 *
188 * MPI(2,type) fileno -- type is 0x00
Fred Drake8c081a12001-10-12 20:57:55 +0000189 * PI lineno
Fred Drake30d1c752001-10-15 22:11:02 +0000190 * PI tdelta -- iff frame times are enabled
Fred Drake8c081a12001-10-12 20:57:55 +0000191 *
192 * EXIT records
193 *
Fred Drake30d1c752001-10-15 22:11:02 +0000194 * MPI(2,type) tdelta -- type is 0x01; tdelta will be 0
195 * if frame times are disabled
Fred Drake8c081a12001-10-12 20:57:55 +0000196 *
197 * LINENO records
198 *
199 * MPI(2,type) lineno -- type is 0x02
200 * PI tdelta -- iff LINENO includes it
201 *
202 * ADD_INFO records
203 *
Fred Drake30d1c752001-10-15 22:11:02 +0000204 * BYTE type -- always 0x13
Fred Drake8c081a12001-10-12 20:57:55 +0000205 * PI len1 -- length of first string
206 * BYTE string1[len1] -- len1 bytes of string data
207 * PI len2 -- length of second string
208 * BYTE string2[len2] -- len2 bytes of string data
209 *
210 * DEFINE_FILE records
211 *
Fred Drake30d1c752001-10-15 22:11:02 +0000212 * BYTE type -- always 0x23
Fred Drake8c081a12001-10-12 20:57:55 +0000213 * PI fileno
214 * PI len -- length of filename
215 * BYTE filename[len] -- len bytes of string data
216 *
Fred Drake30d1c752001-10-15 22:11:02 +0000217 * DEFINE_FUNC records
218 *
219 * BYTE type -- always 0x43
220 * PI fileno
221 * PI lineno
222 * PI len -- length of funcname
223 * BYTE funcname[len] -- len bytes of string data
224 *
Fred Drake8c081a12001-10-12 20:57:55 +0000225 * LINE_TIMES records
Fred Drake30d1c752001-10-15 22:11:02 +0000226 *
227 * This record can be used only before the start of ENTER/EXIT/LINENO
228 * records. If have_tdelta is true, LINENO records will include the
229 * tdelta field, otherwise it will be omitted. If this record is not
230 * given, LINENO records will not contain the tdelta field.
231 *
232 * BYTE type -- always 0x33
Fred Drake8c081a12001-10-12 20:57:55 +0000233 * BYTE have_tdelta -- 0 if LINENO does *not* have
234 * timing information
Fred Drake30d1c752001-10-15 22:11:02 +0000235 * FRAME_TIMES records
236 *
237 * This record can be used only before the start of ENTER/EXIT/LINENO
238 * records. If have_tdelta is true, ENTER and EXIT records will
239 * include the tdelta field, otherwise it will be omitted. If this
240 * record is not given, ENTER and EXIT records will contain the tdelta
241 * field.
242 *
243 * BYTE type -- always 0x53
244 * BYTE have_tdelta -- 0 if ENTER/EXIT do *not* have
245 * timing information
Fred Drake8c081a12001-10-12 20:57:55 +0000246 */
247
248#define WHAT_ENTER 0x00
249#define WHAT_EXIT 0x01
250#define WHAT_LINENO 0x02
251#define WHAT_OTHER 0x03 /* only used in decoding */
252#define WHAT_ADD_INFO 0x13
253#define WHAT_DEFINE_FILE 0x23
254#define WHAT_LINE_TIMES 0x33
Fred Drake30d1c752001-10-15 22:11:02 +0000255#define WHAT_DEFINE_FUNC 0x43
256#define WHAT_FRAME_TIMES 0x53
Fred Drake8c081a12001-10-12 20:57:55 +0000257
258#define ERR_NONE 0
259#define ERR_EOF -1
260#define ERR_EXCEPTION -2
Fred Drake4c2e1af2001-10-29 20:45:57 +0000261#define ERR_BAD_RECTYPE -3
Fred Drake8c081a12001-10-12 20:57:55 +0000262
263#define PISIZE (sizeof(int) + 1)
264#define MPISIZE (PISIZE + 1)
265
266/* Maximum size of "normal" events -- nothing that contains string data */
267#define MAXEVENTSIZE (MPISIZE + PISIZE*2)
268
269
270/* Unpack a packed integer; if "discard" is non-zero, unpack a modified
271 * packed integer with "discard" discarded bits.
272 */
273static int
274unpack_packed_int(LogReaderObject *self, int *pvalue, int discard)
275{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000276 int c;
Fred Drake8c081a12001-10-12 20:57:55 +0000277 int accum = 0;
278 int bits = 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000279 int cont;
280
281 do {
Fred Drake8c081a12001-10-12 20:57:55 +0000282 /* read byte */
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000283 if ((c = fgetc(self->logfp)) == EOF)
284 return ERR_EOF;
285 accum |= ((c & 0x7F) >> discard) << bits;
Fred Drake8c081a12001-10-12 20:57:55 +0000286 bits += (7 - discard);
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000287 cont = c & 0x80;
Fred Drake8c081a12001-10-12 20:57:55 +0000288 discard = 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000289 } while (cont);
290
Fred Drake8c081a12001-10-12 20:57:55 +0000291 *pvalue = accum;
292
293 return 0;
294}
295
296/* Unpack a string, which is encoded as a packed integer giving the
297 * length of the string, followed by the string data.
298 */
299static int
300unpack_string(LogReaderObject *self, PyObject **pvalue)
301{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000302 int i;
Fred Drake8c081a12001-10-12 20:57:55 +0000303 int len;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000304 int err;
Guido van Rossum0b624f62002-07-20 00:38:01 +0000305 int ch;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000306 char *buf;
307
308 if ((err = unpack_packed_int(self, &len, 0)))
309 return err;
Fred Drake8c081a12001-10-12 20:57:55 +0000310
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000311 buf = malloc(len);
312 for (i=0; i < len; i++) {
Guido van Rossum0b624f62002-07-20 00:38:01 +0000313 ch = fgetc(self->logfp);
314 buf[i] = ch;
315 if (ch == EOF) {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000316 free(buf);
317 return ERR_EOF;
Fred Drake8c081a12001-10-12 20:57:55 +0000318 }
319 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000320 *pvalue = PyString_FromStringAndSize(buf, len);
321 free(buf);
322 if (*pvalue == NULL) {
323 return ERR_EXCEPTION;
324 }
325 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000326}
327
328
Fred Drake4c2e1af2001-10-29 20:45:57 +0000329static int
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000330unpack_add_info(LogReaderObject *self)
Fred Drake4c2e1af2001-10-29 20:45:57 +0000331{
332 PyObject *key;
333 PyObject *value = NULL;
334 int err;
335
Fred Drake4c2e1af2001-10-29 20:45:57 +0000336 err = unpack_string(self, &key);
337 if (!err) {
338 err = unpack_string(self, &value);
339 if (err)
340 Py_DECREF(key);
341 else {
342 PyObject *list = PyDict_GetItem(self->info, key);
343 if (list == NULL) {
344 list = PyList_New(0);
345 if (list == NULL) {
346 err = ERR_EXCEPTION;
347 goto finally;
348 }
349 if (PyDict_SetItem(self->info, key, list)) {
Armin Rigoa4127692004-08-03 08:33:55 +0000350 Py_DECREF(list);
Fred Drake4c2e1af2001-10-29 20:45:57 +0000351 err = ERR_EXCEPTION;
352 goto finally;
353 }
Armin Rigoa4127692004-08-03 08:33:55 +0000354 Py_DECREF(list);
Fred Drake4c2e1af2001-10-29 20:45:57 +0000355 }
356 if (PyList_Append(list, value))
357 err = ERR_EXCEPTION;
358 }
359 }
360 finally:
361 Py_XDECREF(key);
362 Py_XDECREF(value);
363 return err;
364}
365
366
367static void
Fred Drake666bf522002-07-18 19:11:44 +0000368eof_error(LogReaderObject *self)
Fred Drake4c2e1af2001-10-29 20:45:57 +0000369{
Fred Drake666bf522002-07-18 19:11:44 +0000370 fclose(self->logfp);
371 self->logfp = NULL;
Fred Drake4c2e1af2001-10-29 20:45:57 +0000372 PyErr_SetString(PyExc_EOFError,
373 "end of file with incomplete profile record");
374}
375
Fred Drake8c081a12001-10-12 20:57:55 +0000376static PyObject *
377logreader_tp_iternext(LogReaderObject *self)
378{
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000379 int c;
380 int what;
Fred Drake8c081a12001-10-12 20:57:55 +0000381 int err = ERR_NONE;
382 int lineno = -1;
383 int fileno = -1;
384 int tdelta = -1;
385 PyObject *s1 = NULL, *s2 = NULL;
386 PyObject *result = NULL;
387#if 0
388 unsigned char b0, b1;
389#endif
390
391 if (self->logfp == NULL) {
392 PyErr_SetString(ProfilerError,
393 "cannot iterate over closed LogReader object");
394 return NULL;
395 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000396
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000397restart:
398 /* decode the record type */
Fred Drake666bf522002-07-18 19:11:44 +0000399 if ((c = fgetc(self->logfp)) == EOF) {
400 fclose(self->logfp);
401 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +0000402 return NULL;
Fred Drake666bf522002-07-18 19:11:44 +0000403 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000404 what = c & WHAT_OTHER;
405 if (what == WHAT_OTHER)
406 what = c; /* need all the bits for type */
407 else
408 ungetc(c, self->logfp); /* type byte includes packed int */
Fred Drake8c081a12001-10-12 20:57:55 +0000409
Fred Drake8c081a12001-10-12 20:57:55 +0000410 switch (what) {
411 case WHAT_ENTER:
412 err = unpack_packed_int(self, &fileno, 2);
413 if (!err) {
Fred Drake30d1c752001-10-15 22:11:02 +0000414 err = unpack_packed_int(self, &lineno, 0);
415 if (self->frametimings && !err)
416 err = unpack_packed_int(self, &tdelta, 0);
Fred Drake8c081a12001-10-12 20:57:55 +0000417 }
418 break;
419 case WHAT_EXIT:
420 err = unpack_packed_int(self, &tdelta, 2);
421 break;
422 case WHAT_LINENO:
423 err = unpack_packed_int(self, &lineno, 2);
424 if (self->linetimings && !err)
425 err = unpack_packed_int(self, &tdelta, 0);
426 break;
427 case WHAT_ADD_INFO:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000428 err = unpack_add_info(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000429 break;
430 case WHAT_DEFINE_FILE:
431 err = unpack_packed_int(self, &fileno, 0);
432 if (!err) {
433 err = unpack_string(self, &s1);
434 if (!err) {
435 Py_INCREF(Py_None);
436 s2 = Py_None;
437 }
438 }
439 break;
Fred Drake30d1c752001-10-15 22:11:02 +0000440 case WHAT_DEFINE_FUNC:
441 err = unpack_packed_int(self, &fileno, 0);
442 if (!err) {
443 err = unpack_packed_int(self, &lineno, 0);
444 if (!err)
445 err = unpack_string(self, &s1);
446 }
447 break;
Fred Drake8c081a12001-10-12 20:57:55 +0000448 case WHAT_LINE_TIMES:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000449 if ((c = fgetc(self->logfp)) == EOF)
Fred Drake8c081a12001-10-12 20:57:55 +0000450 err = ERR_EOF;
451 else {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000452 self->linetimings = c ? 1 : 0;
453 goto restart;
454 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000455 break;
Fred Drake30d1c752001-10-15 22:11:02 +0000456 case WHAT_FRAME_TIMES:
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000457 if ((c = fgetc(self->logfp)) == EOF)
Fred Drake30d1c752001-10-15 22:11:02 +0000458 err = ERR_EOF;
459 else {
Neil Schemenauer8b6b4912002-05-29 18:19:14 +0000460 self->frametimings = c ? 1 : 0;
461 goto restart;
462 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000463 break;
Fred Drake8c081a12001-10-12 20:57:55 +0000464 default:
Fred Drake4c2e1af2001-10-29 20:45:57 +0000465 err = ERR_BAD_RECTYPE;
Fred Drake8c081a12001-10-12 20:57:55 +0000466 }
Fred Drake4c2e1af2001-10-29 20:45:57 +0000467 if (err == ERR_BAD_RECTYPE) {
468 PyErr_SetString(PyExc_ValueError,
469 "unknown record type in log file");
470 }
471 else if (err == ERR_EOF) {
Fred Drake666bf522002-07-18 19:11:44 +0000472 eof_error(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000473 }
474 else if (!err) {
475 result = PyTuple_New(4);
476 PyTuple_SET_ITEM(result, 0, PyInt_FromLong(what));
477 PyTuple_SET_ITEM(result, 2, PyInt_FromLong(fileno));
Fred Drake30d1c752001-10-15 22:11:02 +0000478 if (s1 == NULL)
Fred Drake8c081a12001-10-12 20:57:55 +0000479 PyTuple_SET_ITEM(result, 1, PyInt_FromLong(tdelta));
Fred Drake30d1c752001-10-15 22:11:02 +0000480 else
Fred Drake8c081a12001-10-12 20:57:55 +0000481 PyTuple_SET_ITEM(result, 1, s1);
Fred Drake30d1c752001-10-15 22:11:02 +0000482 if (s2 == NULL)
483 PyTuple_SET_ITEM(result, 3, PyInt_FromLong(lineno));
484 else
Fred Drake8c081a12001-10-12 20:57:55 +0000485 PyTuple_SET_ITEM(result, 3, s2);
Fred Drake8c081a12001-10-12 20:57:55 +0000486 }
487 /* The only other case is err == ERR_EXCEPTION, in which case the
488 * exception is already set.
489 */
490#if 0
491 b0 = self->buffer[self->index];
492 b1 = self->buffer[self->index + 1];
493 if (b0 & 1) {
494 /* This is a line-number event. */
495 what = PyTrace_LINE;
496 lineno = ((b0 & ~1) << 7) + b1;
497 self->index += 2;
498 }
499 else {
500 what = (b0 & 0x0E) >> 1;
501 tdelta = ((b0 & 0xF0) << 4) + b1;
502 if (what == PyTrace_CALL) {
503 /* we know there's a 2-byte file ID & 2-byte line number */
504 fileno = ((self->buffer[self->index + 2] << 8)
505 + self->buffer[self->index + 3]);
506 lineno = ((self->buffer[self->index + 4] << 8)
507 + self->buffer[self->index + 5]);
508 self->index += 6;
509 }
510 else
511 self->index += 2;
512 }
513#endif
514 return result;
515}
516
517static void
518logreader_dealloc(LogReaderObject *self)
519{
520 if (self->logfp != NULL) {
521 fclose(self->logfp);
522 self->logfp = NULL;
523 }
Armin Rigoa4127692004-08-03 08:33:55 +0000524 Py_XDECREF(self->info);
Fred Drake8c081a12001-10-12 20:57:55 +0000525 PyObject_Del(self);
526}
527
528static PyObject *
529logreader_sq_item(LogReaderObject *self, int index)
530{
531 PyObject *result = logreader_tp_iternext(self);
532 if (result == NULL && !PyErr_Occurred()) {
533 PyErr_SetString(PyExc_IndexError, "no more events in log");
534 return NULL;
535 }
536 return result;
537}
538
Fred Drake62c1e3c2001-12-04 21:40:53 +0000539static void
540do_stop(ProfilerObject *self);
Fred Drake8c081a12001-10-12 20:57:55 +0000541
542static int
543flush_data(ProfilerObject *self)
544{
545 /* Need to dump data to the log file... */
546 size_t written = fwrite(self->buffer, 1, self->index, self->logfp);
Tim Peters1566a172001-10-12 22:08:39 +0000547 if (written == (size_t)self->index)
Fred Drake8c081a12001-10-12 20:57:55 +0000548 self->index = 0;
549 else {
550 memmove(self->buffer, &self->buffer[written],
551 self->index - written);
552 self->index -= written;
553 if (written == 0) {
554 char *s = PyString_AsString(self->logfilename);
555 PyErr_SetFromErrnoWithFilename(PyExc_IOError, s);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000556 do_stop(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000557 return -1;
558 }
559 }
560 if (written > 0) {
561 if (fflush(self->logfp)) {
562 char *s = PyString_AsString(self->logfilename);
563 PyErr_SetFromErrnoWithFilename(PyExc_IOError, s);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000564 do_stop(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000565 return -1;
566 }
567 }
568 return 0;
569}
570
Fred Drake62c1e3c2001-12-04 21:40:53 +0000571static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000572pack_packed_int(ProfilerObject *self, int value)
573{
574 unsigned char partial;
575
576 do {
577 partial = value & 0x7F;
578 value >>= 7;
579 if (value)
580 partial |= 0x80;
581 self->buffer[self->index] = partial;
582 self->index++;
583 } while (value);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000584 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000585}
586
587/* Encode a modified packed integer, with a subfield of modsize bits
588 * containing the value "subfield". The value of subfield is not
589 * checked to ensure it actually fits in modsize bits.
590 */
Fred Drake62c1e3c2001-12-04 21:40:53 +0000591static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000592pack_modified_packed_int(ProfilerObject *self, int value,
593 int modsize, int subfield)
594{
595 const int maxvalues[] = {-1, 1, 3, 7, 15, 31, 63, 127};
596
597 int bits = 7 - modsize;
598 int partial = value & maxvalues[bits];
599 unsigned char b = subfield | (partial << modsize);
600
601 if (partial != value) {
602 b |= 0x80;
603 self->buffer[self->index] = b;
604 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000605 return pack_packed_int(self, value >> bits);
Fred Drake8c081a12001-10-12 20:57:55 +0000606 }
Fred Drake62c1e3c2001-12-04 21:40:53 +0000607 self->buffer[self->index] = b;
608 self->index++;
609 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000610}
611
Fred Drake62c1e3c2001-12-04 21:40:53 +0000612static int
Fred Drake30d1c752001-10-15 22:11:02 +0000613pack_string(ProfilerObject *self, const char *s, int len)
Fred Drake8c081a12001-10-12 20:57:55 +0000614{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000615 if (len + PISIZE + self->index >= BUFFERSIZE) {
616 if (flush_data(self) < 0)
617 return -1;
618 }
619 if (pack_packed_int(self, len) < 0)
620 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000621 memcpy(self->buffer + self->index, s, len);
622 self->index += len;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000623 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000624}
625
Fred Drake62c1e3c2001-12-04 21:40:53 +0000626static int
Fred Drake8c081a12001-10-12 20:57:55 +0000627pack_add_info(ProfilerObject *self, const char *s1, const char *s2)
628{
629 int len1 = strlen(s1);
630 int len2 = strlen(s2);
631
Fred Drake62c1e3c2001-12-04 21:40:53 +0000632 if (len1 + len2 + PISIZE*2 + 1 + self->index >= BUFFERSIZE) {
633 if (flush_data(self) < 0)
634 return -1;
635 }
Fred Drake8c081a12001-10-12 20:57:55 +0000636 self->buffer[self->index] = WHAT_ADD_INFO;
637 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000638 if (pack_string(self, s1, len1) < 0)
639 return -1;
640 return pack_string(self, s2, len2);
Fred Drake8c081a12001-10-12 20:57:55 +0000641}
642
Fred Drake62c1e3c2001-12-04 21:40:53 +0000643static int
Fred Drake8c081a12001-10-12 20:57:55 +0000644pack_define_file(ProfilerObject *self, int fileno, const char *filename)
645{
646 int len = strlen(filename);
647
Fred Drake62c1e3c2001-12-04 21:40:53 +0000648 if (len + PISIZE*2 + 1 + self->index >= BUFFERSIZE) {
649 if (flush_data(self) < 0)
650 return -1;
651 }
Fred Drake8c081a12001-10-12 20:57:55 +0000652 self->buffer[self->index] = WHAT_DEFINE_FILE;
653 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000654 if (pack_packed_int(self, fileno) < 0)
655 return -1;
656 return pack_string(self, filename, len);
Fred Drake30d1c752001-10-15 22:11:02 +0000657}
658
Fred Drake62c1e3c2001-12-04 21:40:53 +0000659static int
Fred Drake30d1c752001-10-15 22:11:02 +0000660pack_define_func(ProfilerObject *self, int fileno, int lineno,
661 const char *funcname)
662{
663 int len = strlen(funcname);
664
Fred Drake62c1e3c2001-12-04 21:40:53 +0000665 if (len + PISIZE*3 + 1 + self->index >= BUFFERSIZE) {
666 if (flush_data(self) < 0)
667 return -1;
668 }
Fred Drake30d1c752001-10-15 22:11:02 +0000669 self->buffer[self->index] = WHAT_DEFINE_FUNC;
670 self->index++;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000671 if (pack_packed_int(self, fileno) < 0)
672 return -1;
673 if (pack_packed_int(self, lineno) < 0)
674 return -1;
675 return pack_string(self, funcname, len);
Fred Drake8c081a12001-10-12 20:57:55 +0000676}
677
Fred Drake62c1e3c2001-12-04 21:40:53 +0000678static int
Fred Drake8c081a12001-10-12 20:57:55 +0000679pack_line_times(ProfilerObject *self)
680{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000681 if (2 + self->index >= BUFFERSIZE) {
682 if (flush_data(self) < 0)
683 return -1;
684 }
Fred Drake8c081a12001-10-12 20:57:55 +0000685 self->buffer[self->index] = WHAT_LINE_TIMES;
686 self->buffer[self->index + 1] = self->linetimings ? 1 : 0;
687 self->index += 2;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000688 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000689}
690
Fred Drake62c1e3c2001-12-04 21:40:53 +0000691static int
Fred Drake30d1c752001-10-15 22:11:02 +0000692pack_frame_times(ProfilerObject *self)
693{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000694 if (2 + self->index >= BUFFERSIZE) {
695 if (flush_data(self) < 0)
696 return -1;
697 }
Fred Drake30d1c752001-10-15 22:11:02 +0000698 self->buffer[self->index] = WHAT_FRAME_TIMES;
699 self->buffer[self->index + 1] = self->frametimings ? 1 : 0;
700 self->index += 2;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000701 return 0;
Fred Drake30d1c752001-10-15 22:11:02 +0000702}
703
Fred Drake62c1e3c2001-12-04 21:40:53 +0000704static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000705pack_enter(ProfilerObject *self, int fileno, int tdelta, int lineno)
706{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000707 if (MPISIZE + PISIZE*2 + self->index >= BUFFERSIZE) {
708 if (flush_data(self) < 0)
709 return -1;
710 }
Fred Drake8c081a12001-10-12 20:57:55 +0000711 pack_modified_packed_int(self, fileno, 2, WHAT_ENTER);
Fred Drake8c081a12001-10-12 20:57:55 +0000712 pack_packed_int(self, lineno);
Fred Drake30d1c752001-10-15 22:11:02 +0000713 if (self->frametimings)
Fred Drake62c1e3c2001-12-04 21:40:53 +0000714 return pack_packed_int(self, tdelta);
715 else
716 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000717}
718
Fred Drake62c1e3c2001-12-04 21:40:53 +0000719static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000720pack_exit(ProfilerObject *self, int tdelta)
721{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000722 if (MPISIZE + self->index >= BUFFERSIZE) {
723 if (flush_data(self) < 0)
724 return -1;
Fred Drake30d1c752001-10-15 22:11:02 +0000725 }
Fred Drake62c1e3c2001-12-04 21:40:53 +0000726 if (self->frametimings)
727 return pack_modified_packed_int(self, tdelta, 2, WHAT_EXIT);
728 self->buffer[self->index] = WHAT_EXIT;
729 self->index++;
730 return 0;
Fred Drake8c081a12001-10-12 20:57:55 +0000731}
732
Fred Drake62c1e3c2001-12-04 21:40:53 +0000733static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000734pack_lineno(ProfilerObject *self, int lineno)
735{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000736 if (MPISIZE + self->index >= BUFFERSIZE) {
737 if (flush_data(self) < 0)
738 return -1;
739 }
740 return pack_modified_packed_int(self, lineno, 2, WHAT_LINENO);
Fred Drake8c081a12001-10-12 20:57:55 +0000741}
742
Fred Drake62c1e3c2001-12-04 21:40:53 +0000743static inline int
Fred Drake8c081a12001-10-12 20:57:55 +0000744pack_lineno_tdelta(ProfilerObject *self, int lineno, int tdelta)
745{
Fred Drake62c1e3c2001-12-04 21:40:53 +0000746 if (MPISIZE + PISIZE + self->index >= BUFFERSIZE) {
747 if (flush_data(self) < 0)
748 return 0;
749 }
750 if (pack_modified_packed_int(self, lineno, 2, WHAT_LINENO) < 0)
751 return -1;
752 return pack_packed_int(self, tdelta);
Fred Drake8c081a12001-10-12 20:57:55 +0000753}
754
755static inline int
756get_fileno(ProfilerObject *self, PyCodeObject *fcode)
757{
Fred Drake30d1c752001-10-15 22:11:02 +0000758 /* This is only used for ENTER events. */
759
760 PyObject *obj;
761 PyObject *dict;
Fred Drake8c081a12001-10-12 20:57:55 +0000762 int fileno;
763
Fred Drake30d1c752001-10-15 22:11:02 +0000764 obj = PyDict_GetItem(self->filemap, fcode->co_filename);
765 if (obj == NULL) {
Fred Drake8c081a12001-10-12 20:57:55 +0000766 /* first sighting of this file */
Fred Drake30d1c752001-10-15 22:11:02 +0000767 dict = PyDict_New();
768 if (dict == NULL) {
Fred Drake8c081a12001-10-12 20:57:55 +0000769 return -1;
770 }
Fred Drake30d1c752001-10-15 22:11:02 +0000771 fileno = self->next_fileno;
772 obj = Py_BuildValue("iN", fileno, dict);
773 if (obj == NULL) {
774 return -1;
775 }
776 if (PyDict_SetItem(self->filemap, fcode->co_filename, obj)) {
777 Py_DECREF(obj);
Fred Drake8c081a12001-10-12 20:57:55 +0000778 return -1;
779 }
780 self->next_fileno++;
Fred Drake30d1c752001-10-15 22:11:02 +0000781 Py_DECREF(obj);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000782 if (pack_define_file(self, fileno,
783 PyString_AS_STRING(fcode->co_filename)) < 0)
784 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000785 }
786 else {
787 /* already know this ID */
Fred Drake30d1c752001-10-15 22:11:02 +0000788 fileno = PyInt_AS_LONG(PyTuple_GET_ITEM(obj, 0));
789 dict = PyTuple_GET_ITEM(obj, 1);
790 }
791 /* make sure we save a function name for this (fileno, lineno) */
792 obj = PyInt_FromLong(fcode->co_firstlineno);
793 if (obj == NULL) {
794 /* We just won't have it saved; too bad. */
795 PyErr_Clear();
796 }
797 else {
798 PyObject *name = PyDict_GetItem(dict, obj);
799 if (name == NULL) {
Fred Drake62c1e3c2001-12-04 21:40:53 +0000800 if (pack_define_func(self, fileno, fcode->co_firstlineno,
Armin Rigoa4127692004-08-03 08:33:55 +0000801 PyString_AS_STRING(fcode->co_name)) < 0) {
802 Py_DECREF(obj);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000803 return -1;
Armin Rigoa4127692004-08-03 08:33:55 +0000804 }
805 if (PyDict_SetItem(dict, obj, fcode->co_name)) {
806 Py_DECREF(obj);
Fred Drake30d1c752001-10-15 22:11:02 +0000807 return -1;
Armin Rigoa4127692004-08-03 08:33:55 +0000808 }
Fred Drake30d1c752001-10-15 22:11:02 +0000809 }
Armin Rigoa4127692004-08-03 08:33:55 +0000810 Py_DECREF(obj);
Fred Drake8c081a12001-10-12 20:57:55 +0000811 }
812 return fileno;
813}
814
815static inline int
816get_tdelta(ProfilerObject *self)
817{
818 int tdelta;
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000819#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +0000820 hs_time tv;
Tim Peters7d99ff22001-10-13 07:37:52 +0000821 hs_time diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000822
Tim Peters7d99ff22001-10-13 07:37:52 +0000823 GETTIMEOFDAY(&tv);
824 diff = tv - self->prev_timeofday;
825 tdelta = (int)diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000826#else
827 struct timeval tv;
828
829 GETTIMEOFDAY(&tv);
830
Neal Norwitz7a135162004-06-13 20:45:11 +0000831 tdelta = tv.tv_usec - self->prev_timeofday.tv_usec;
832 if (tv.tv_sec != self->prev_timeofday.tv_sec)
833 tdelta += (tv.tv_sec - self->prev_timeofday.tv_sec) * 1000000;
Fred Drake8c081a12001-10-12 20:57:55 +0000834#endif
Neal Norwitz7a135162004-06-13 20:45:11 +0000835 /* time can go backwards on some multiprocessor systems or by NTP */
836 if (tdelta < 0)
837 return 0;
838
Fred Drake8c081a12001-10-12 20:57:55 +0000839 self->prev_timeofday = tv;
840 return tdelta;
841}
842
843
844/* The workhorse: the profiler callback function. */
845
846static int
847profiler_callback(ProfilerObject *self, PyFrameObject *frame, int what,
848 PyObject *arg)
849{
Fred Drake30d1c752001-10-15 22:11:02 +0000850 int tdelta = -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000851 int fileno;
852
Fred Drake30d1c752001-10-15 22:11:02 +0000853 if (self->frametimings)
854 tdelta = get_tdelta(self);
Fred Drake8c081a12001-10-12 20:57:55 +0000855 switch (what) {
856 case PyTrace_CALL:
857 fileno = get_fileno(self, frame->f_code);
858 if (fileno < 0)
859 return -1;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000860 if (pack_enter(self, fileno, tdelta,
861 frame->f_code->co_firstlineno) < 0)
862 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000863 break;
864 case PyTrace_RETURN:
Fred Drake62c1e3c2001-12-04 21:40:53 +0000865 if (pack_exit(self, tdelta) < 0)
866 return -1;
Fred Drake8c081a12001-10-12 20:57:55 +0000867 break;
868 default:
869 /* should never get here */
870 break;
871 }
872 return 0;
873}
874
875
876/* Alternate callback when we want PyTrace_LINE events */
877
878static int
879tracer_callback(ProfilerObject *self, PyFrameObject *frame, int what,
880 PyObject *arg)
881{
882 int fileno;
883
Fred Drake8c081a12001-10-12 20:57:55 +0000884 switch (what) {
885 case PyTrace_CALL:
886 fileno = get_fileno(self, frame->f_code);
887 if (fileno < 0)
888 return -1;
Fred Drake62c1e3c2001-12-04 21:40:53 +0000889 return pack_enter(self, fileno,
890 self->frametimings ? get_tdelta(self) : -1,
891 frame->f_code->co_firstlineno);
892
Fred Drake8c081a12001-10-12 20:57:55 +0000893 case PyTrace_RETURN:
Fred Drake62c1e3c2001-12-04 21:40:53 +0000894 return pack_exit(self, get_tdelta(self));
895
Tim Peters1566a172001-10-12 22:08:39 +0000896 case PyTrace_LINE:
Fred Drake8c081a12001-10-12 20:57:55 +0000897 if (self->linetimings)
Michael W. Hudson806d1c82002-09-11 17:09:45 +0000898 return pack_lineno_tdelta(self, frame->f_lineno,
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000899 get_tdelta(self));
Fred Drake8c081a12001-10-12 20:57:55 +0000900 else
Michael W. Hudson02ff6a92002-09-11 15:36:32 +0000901 return pack_lineno(self, frame->f_lineno);
Fred Drake62c1e3c2001-12-04 21:40:53 +0000902
Fred Drake8c081a12001-10-12 20:57:55 +0000903 default:
904 /* ignore PyTrace_EXCEPTION */
905 break;
906 }
907 return 0;
908}
909
910
911/* A couple of useful helper functions. */
912
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000913#ifdef MS_WINDOWS
Tim Petersfeab23f2001-10-13 00:11:10 +0000914static LARGE_INTEGER frequency = {0, 0};
Fred Drake8c081a12001-10-12 20:57:55 +0000915#endif
916
917static unsigned long timeofday_diff = 0;
918static unsigned long rusage_diff = 0;
919
920static void
921calibrate(void)
922{
923 hs_time tv1, tv2;
924
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000925#ifdef MS_WINDOWS
Tim Peters7d99ff22001-10-13 07:37:52 +0000926 hs_time diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000927 QueryPerformanceFrequency(&frequency);
928#endif
929
930 GETTIMEOFDAY(&tv1);
931 while (1) {
932 GETTIMEOFDAY(&tv2);
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000933#ifdef MS_WINDOWS
Tim Peters7d99ff22001-10-13 07:37:52 +0000934 diff = tv2 - tv1;
935 if (diff != 0) {
936 timeofday_diff = (unsigned long)diff;
Fred Drake8c081a12001-10-12 20:57:55 +0000937 break;
938 }
939#else
940 if (tv1.tv_sec != tv2.tv_sec || tv1.tv_usec != tv2.tv_usec) {
941 if (tv1.tv_sec == tv2.tv_sec)
942 timeofday_diff = tv2.tv_usec - tv1.tv_usec;
943 else
944 timeofday_diff = (1000000 - tv1.tv_usec) + tv2.tv_usec;
945 break;
946 }
947#endif
948 }
Jack Janseneddc1442003-11-20 01:44:59 +0000949#if defined(MS_WINDOWS) || defined(PYOS_OS2) || \
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000950 defined(__VMS)
Fred Drake8c081a12001-10-12 20:57:55 +0000951 rusage_diff = -1;
952#else
953 {
954 struct rusage ru1, ru2;
955
956 getrusage(RUSAGE_SELF, &ru1);
957 while (1) {
958 getrusage(RUSAGE_SELF, &ru2);
959 if (ru1.ru_utime.tv_sec != ru2.ru_utime.tv_sec) {
960 rusage_diff = ((1000000 - ru1.ru_utime.tv_usec)
961 + ru2.ru_utime.tv_usec);
962 break;
963 }
964 else if (ru1.ru_utime.tv_usec != ru2.ru_utime.tv_usec) {
965 rusage_diff = ru2.ru_utime.tv_usec - ru1.ru_utime.tv_usec;
966 break;
967 }
968 else if (ru1.ru_stime.tv_sec != ru2.ru_stime.tv_sec) {
969 rusage_diff = ((1000000 - ru1.ru_stime.tv_usec)
970 + ru2.ru_stime.tv_usec);
971 break;
972 }
973 else if (ru1.ru_stime.tv_usec != ru2.ru_stime.tv_usec) {
974 rusage_diff = ru2.ru_stime.tv_usec - ru1.ru_stime.tv_usec;
975 break;
976 }
977 }
978 }
979#endif
980}
981
982static void
983do_start(ProfilerObject *self)
984{
985 self->active = 1;
986 GETTIMEOFDAY(&self->prev_timeofday);
987 if (self->lineevents)
988 PyEval_SetTrace((Py_tracefunc) tracer_callback, (PyObject *)self);
989 else
990 PyEval_SetProfile((Py_tracefunc) profiler_callback, (PyObject *)self);
991}
992
993static void
994do_stop(ProfilerObject *self)
995{
996 if (self->active) {
997 self->active = 0;
998 if (self->lineevents)
999 PyEval_SetTrace(NULL, NULL);
1000 else
1001 PyEval_SetProfile(NULL, NULL);
1002 }
1003 if (self->index > 0) {
1004 /* Best effort to dump out any remaining data. */
1005 flush_data(self);
1006 }
1007}
1008
1009static int
1010is_available(ProfilerObject *self)
1011{
1012 if (self->active) {
1013 PyErr_SetString(ProfilerError, "profiler already active");
1014 return 0;
1015 }
1016 if (self->logfp == NULL) {
1017 PyErr_SetString(ProfilerError, "profiler already closed");
1018 return 0;
1019 }
1020 return 1;
1021}
1022
1023
1024/* Profiler object interface methods. */
1025
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001026PyDoc_STRVAR(addinfo__doc__,
Fred Drake4c2e1af2001-10-29 20:45:57 +00001027"addinfo(key, value)\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001028"Insert an ADD_INFO record into the log.");
Fred Drake4c2e1af2001-10-29 20:45:57 +00001029
1030static PyObject *
1031profiler_addinfo(ProfilerObject *self, PyObject *args)
1032{
1033 PyObject *result = NULL;
1034 char *key, *value;
1035
1036 if (PyArg_ParseTuple(args, "ss:addinfo", &key, &value)) {
1037 if (self->logfp == NULL)
1038 PyErr_SetString(ProfilerError, "profiler already closed");
1039 else {
Fred Drake62c1e3c2001-12-04 21:40:53 +00001040 if (pack_add_info(self, key, value) == 0) {
1041 result = Py_None;
1042 Py_INCREF(result);
1043 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001044 }
1045 }
1046 return result;
1047}
1048
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001049PyDoc_STRVAR(close__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001050"close()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001051"Shut down this profiler and close the log files, even if its active.");
Fred Drake8c081a12001-10-12 20:57:55 +00001052
1053static PyObject *
Fred Drake666bf522002-07-18 19:11:44 +00001054profiler_close(ProfilerObject *self)
Fred Drake8c081a12001-10-12 20:57:55 +00001055{
Fred Drake666bf522002-07-18 19:11:44 +00001056 do_stop(self);
1057 if (self->logfp != NULL) {
1058 fclose(self->logfp);
1059 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001060 }
Fred Drake666bf522002-07-18 19:11:44 +00001061 Py_INCREF(Py_None);
1062 return Py_None;
1063}
1064
1065#define fileno__doc__ logreader_fileno__doc__
1066
1067static PyObject *
1068profiler_fileno(ProfilerObject *self)
1069{
1070 if (self->logfp == NULL) {
1071 PyErr_SetString(PyExc_ValueError,
1072 "profiler's file object already closed");
1073 return NULL;
1074 }
1075 return PyInt_FromLong(fileno(self->logfp));
Fred Drake8c081a12001-10-12 20:57:55 +00001076}
1077
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001078PyDoc_STRVAR(runcall__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001079"runcall(callable[, args[, kw]]) -> callable()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001080"Profile a specific function call, returning the result of that call.");
Fred Drake8c081a12001-10-12 20:57:55 +00001081
1082static PyObject *
1083profiler_runcall(ProfilerObject *self, PyObject *args)
1084{
1085 PyObject *result = NULL;
1086 PyObject *callargs = NULL;
1087 PyObject *callkw = NULL;
1088 PyObject *callable;
1089
1090 if (PyArg_ParseTuple(args, "O|OO:runcall",
1091 &callable, &callargs, &callkw)) {
1092 if (is_available(self)) {
1093 do_start(self);
1094 result = PyEval_CallObjectWithKeywords(callable, callargs, callkw);
1095 do_stop(self);
1096 }
1097 }
1098 return result;
1099}
1100
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001101PyDoc_STRVAR(runcode__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001102"runcode(code, globals[, locals])\n"
1103"Execute a code object while collecting profile data. If locals is\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001104"omitted, globals is used for the locals as well.");
Fred Drake8c081a12001-10-12 20:57:55 +00001105
1106static PyObject *
1107profiler_runcode(ProfilerObject *self, PyObject *args)
1108{
1109 PyObject *result = NULL;
1110 PyCodeObject *code;
1111 PyObject *globals;
1112 PyObject *locals = NULL;
1113
1114 if (PyArg_ParseTuple(args, "O!O!|O:runcode",
1115 &PyCode_Type, &code,
1116 &PyDict_Type, &globals,
1117 &locals)) {
1118 if (is_available(self)) {
1119 if (locals == NULL || locals == Py_None)
1120 locals = globals;
1121 else if (!PyDict_Check(locals)) {
1122 PyErr_SetString(PyExc_TypeError,
1123 "locals must be a dictionary or None");
1124 return NULL;
1125 }
1126 do_start(self);
1127 result = PyEval_EvalCode(code, globals, locals);
1128 do_stop(self);
1129#if 0
1130 if (!PyErr_Occurred()) {
1131 result = Py_None;
1132 Py_INCREF(result);
1133 }
1134#endif
1135 }
1136 }
1137 return result;
1138}
1139
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001140PyDoc_STRVAR(start__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001141"start()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001142"Install this profiler for the current thread.");
Fred Drake8c081a12001-10-12 20:57:55 +00001143
1144static PyObject *
1145profiler_start(ProfilerObject *self, PyObject *args)
1146{
1147 PyObject *result = NULL;
1148
Fred Drake666bf522002-07-18 19:11:44 +00001149 if (is_available(self)) {
1150 do_start(self);
1151 result = Py_None;
1152 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001153 }
1154 return result;
1155}
1156
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001157PyDoc_STRVAR(stop__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001158"stop()\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001159"Remove this profiler from the current thread.");
Fred Drake8c081a12001-10-12 20:57:55 +00001160
1161static PyObject *
1162profiler_stop(ProfilerObject *self, PyObject *args)
1163{
1164 PyObject *result = NULL;
1165
Fred Drake666bf522002-07-18 19:11:44 +00001166 if (!self->active)
1167 PyErr_SetString(ProfilerError, "profiler not active");
1168 else {
1169 do_stop(self);
1170 result = Py_None;
1171 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001172 }
1173 return result;
1174}
1175
1176
1177/* Python API support. */
1178
1179static void
1180profiler_dealloc(ProfilerObject *self)
1181{
1182 do_stop(self);
1183 if (self->logfp != NULL)
1184 fclose(self->logfp);
1185 Py_XDECREF(self->filemap);
1186 Py_XDECREF(self->logfilename);
1187 PyObject_Del((PyObject *)self);
1188}
1189
Fred Drake8c081a12001-10-12 20:57:55 +00001190static PyMethodDef profiler_methods[] = {
Fred Drake4c2e1af2001-10-29 20:45:57 +00001191 {"addinfo", (PyCFunction)profiler_addinfo, METH_VARARGS, addinfo__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001192 {"close", (PyCFunction)profiler_close, METH_NOARGS, close__doc__},
1193 {"fileno", (PyCFunction)profiler_fileno, METH_NOARGS, fileno__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001194 {"runcall", (PyCFunction)profiler_runcall, METH_VARARGS, runcall__doc__},
1195 {"runcode", (PyCFunction)profiler_runcode, METH_VARARGS, runcode__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001196 {"start", (PyCFunction)profiler_start, METH_NOARGS, start__doc__},
1197 {"stop", (PyCFunction)profiler_stop, METH_NOARGS, stop__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001198 {NULL, NULL}
1199};
1200
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001201static PyMemberDef profiler_members[] = {
Fred Drake30d1c752001-10-15 22:11:02 +00001202 {"frametimings", T_LONG, offsetof(ProfilerObject, linetimings), READONLY},
1203 {"lineevents", T_LONG, offsetof(ProfilerObject, lineevents), READONLY},
1204 {"linetimings", T_LONG, offsetof(ProfilerObject, linetimings), READONLY},
Fred Drake8c081a12001-10-12 20:57:55 +00001205 {NULL}
1206};
1207
1208static PyObject *
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001209profiler_get_closed(ProfilerObject *self, void *closure)
Fred Drake8c081a12001-10-12 20:57:55 +00001210{
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001211 PyObject *result = (self->logfp == NULL) ? Py_True : Py_False;
1212 Py_INCREF(result);
Fred Drake8c081a12001-10-12 20:57:55 +00001213 return result;
1214}
1215
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001216static PyGetSetDef profiler_getsets[] = {
Fred Draked1eb8b62002-07-17 18:54:20 +00001217 {"closed", (getter)profiler_get_closed, NULL,
Fred Drake5c3ed3d2002-07-17 19:38:05 +00001218 PyDoc_STR("True if the profiler's output file has already been closed.")},
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001219 {NULL}
1220};
1221
Fred Drake8c081a12001-10-12 20:57:55 +00001222
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001223PyDoc_STRVAR(profiler_object__doc__,
Tim Petersfeab23f2001-10-13 00:11:10 +00001224"High-performance profiler object.\n"
1225"\n"
1226"Methods:\n"
1227"\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001228"close(): Stop the profiler and close the log files.\n"
Fred Drake666bf522002-07-18 19:11:44 +00001229"fileno(): Returns the file descriptor of the log file.\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001230"runcall(): Run a single function call with profiling enabled.\n"
1231"runcode(): Execute a code object with profiling enabled.\n"
1232"start(): Install the profiler and return.\n"
1233"stop(): Remove the profiler.\n"
Tim Petersfeab23f2001-10-13 00:11:10 +00001234"\n"
1235"Attributes (read-only):\n"
1236"\n"
Fred Drake30d1c752001-10-15 22:11:02 +00001237"closed: True if the profiler has already been closed.\n"
1238"frametimings: True if ENTER/EXIT events collect timing information.\n"
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001239"lineevents: True if line events are reported to the profiler.\n"
1240"linetimings: True if line events collect timing information.");
Fred Drake8c081a12001-10-12 20:57:55 +00001241
1242static PyTypeObject ProfilerType = {
1243 PyObject_HEAD_INIT(NULL)
1244 0, /* ob_size */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001245 "_hotshot.ProfilerType", /* tp_name */
Fred Drake8c081a12001-10-12 20:57:55 +00001246 (int) sizeof(ProfilerObject), /* tp_basicsize */
1247 0, /* tp_itemsize */
1248 (destructor)profiler_dealloc, /* tp_dealloc */
1249 0, /* tp_print */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001250 0, /* tp_getattr */
Fred Drake8c081a12001-10-12 20:57:55 +00001251 0, /* tp_setattr */
1252 0, /* tp_compare */
1253 0, /* tp_repr */
1254 0, /* tp_as_number */
1255 0, /* tp_as_sequence */
1256 0, /* tp_as_mapping */
1257 0, /* tp_hash */
1258 0, /* tp_call */
1259 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00001260 PyObject_GenericGetAttr, /* tp_getattro */
Fred Drake8c081a12001-10-12 20:57:55 +00001261 0, /* tp_setattro */
1262 0, /* tp_as_buffer */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001263 Py_TPFLAGS_DEFAULT, /* tp_flags */
Fred Drake8c081a12001-10-12 20:57:55 +00001264 profiler_object__doc__, /* tp_doc */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001265 0, /* tp_traverse */
1266 0, /* tp_clear */
1267 0, /* tp_richcompare */
1268 0, /* tp_weaklistoffset */
1269 0, /* tp_iter */
1270 0, /* tp_iternext */
1271 profiler_methods, /* tp_methods */
1272 profiler_members, /* tp_members */
1273 profiler_getsets, /* tp_getset */
1274 0, /* tp_base */
1275 0, /* tp_dict */
1276 0, /* tp_descr_get */
1277 0, /* tp_descr_set */
Fred Drake8c081a12001-10-12 20:57:55 +00001278};
1279
1280
1281static PyMethodDef logreader_methods[] = {
Fred Drake666bf522002-07-18 19:11:44 +00001282 {"close", (PyCFunction)logreader_close, METH_NOARGS,
Fred Drake8c081a12001-10-12 20:57:55 +00001283 logreader_close__doc__},
Fred Drake666bf522002-07-18 19:11:44 +00001284 {"fileno", (PyCFunction)logreader_fileno, METH_NOARGS,
1285 logreader_fileno__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001286 {NULL, NULL}
1287};
1288
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001289static PyMemberDef logreader_members[] = {
Fred Drake5c3ed3d2002-07-17 19:38:05 +00001290 {"info", T_OBJECT, offsetof(LogReaderObject, info), RO,
1291 PyDoc_STR("Dictionary mapping informational keys to lists of values.")},
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001292 {NULL}
1293};
Fred Drake8c081a12001-10-12 20:57:55 +00001294
1295
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001296PyDoc_STRVAR(logreader__doc__,
1297"logreader(filename) --> log-iterator\n\
1298Create a log-reader for the timing information file.");
Fred Drake8c081a12001-10-12 20:57:55 +00001299
1300static PySequenceMethods logreader_as_sequence = {
1301 0, /* sq_length */
1302 0, /* sq_concat */
1303 0, /* sq_repeat */
1304 (intargfunc)logreader_sq_item, /* sq_item */
1305 0, /* sq_slice */
1306 0, /* sq_ass_item */
1307 0, /* sq_ass_slice */
1308 0, /* sq_contains */
1309 0, /* sq_inplace_concat */
1310 0, /* sq_inplace_repeat */
1311};
1312
Fred Drake666bf522002-07-18 19:11:44 +00001313static PyObject *
1314logreader_get_closed(LogReaderObject *self, void *closure)
1315{
1316 PyObject *result = (self->logfp == NULL) ? Py_True : Py_False;
1317 Py_INCREF(result);
1318 return result;
1319}
1320
1321static PyGetSetDef logreader_getsets[] = {
1322 {"closed", (getter)logreader_get_closed, NULL,
1323 PyDoc_STR("True if the logreader's input file has already been closed.")},
1324 {NULL}
1325};
1326
Fred Drake8c081a12001-10-12 20:57:55 +00001327static PyTypeObject LogReaderType = {
1328 PyObject_HEAD_INIT(NULL)
1329 0, /* ob_size */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001330 "_hotshot.LogReaderType", /* tp_name */
Fred Drake8c081a12001-10-12 20:57:55 +00001331 (int) sizeof(LogReaderObject), /* tp_basicsize */
1332 0, /* tp_itemsize */
1333 (destructor)logreader_dealloc, /* tp_dealloc */
1334 0, /* tp_print */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001335 0, /* tp_getattr */
Fred Drake8c081a12001-10-12 20:57:55 +00001336 0, /* tp_setattr */
1337 0, /* tp_compare */
1338 0, /* tp_repr */
1339 0, /* tp_as_number */
1340 &logreader_as_sequence, /* tp_as_sequence */
1341 0, /* tp_as_mapping */
1342 0, /* tp_hash */
1343 0, /* tp_call */
1344 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00001345 PyObject_GenericGetAttr, /* tp_getattro */
Fred Drake8c081a12001-10-12 20:57:55 +00001346 0, /* tp_setattro */
1347 0, /* tp_as_buffer */
Fred Drake4c2e1af2001-10-29 20:45:57 +00001348 Py_TPFLAGS_DEFAULT, /* tp_flags */
Fred Drake8c081a12001-10-12 20:57:55 +00001349 logreader__doc__, /* tp_doc */
Fred Drake8c081a12001-10-12 20:57:55 +00001350 0, /* tp_traverse */
1351 0, /* tp_clear */
1352 0, /* tp_richcompare */
1353 0, /* tp_weaklistoffset */
Raymond Hettinger1da1dbf2003-03-17 19:46:11 +00001354 PyObject_SelfIter, /* tp_iter */
Fred Drake8c081a12001-10-12 20:57:55 +00001355 (iternextfunc)logreader_tp_iternext,/* tp_iternext */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001356 logreader_methods, /* tp_methods */
1357 logreader_members, /* tp_members */
Fred Drake666bf522002-07-18 19:11:44 +00001358 logreader_getsets, /* tp_getset */
Guido van Rossum9cb64b92002-07-17 16:15:35 +00001359 0, /* tp_base */
1360 0, /* tp_dict */
1361 0, /* tp_descr_get */
1362 0, /* tp_descr_set */
Fred Drake8c081a12001-10-12 20:57:55 +00001363};
1364
1365static PyObject *
1366hotshot_logreader(PyObject *unused, PyObject *args)
1367{
1368 LogReaderObject *self = NULL;
1369 char *filename;
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001370 int c;
1371 int err = 0;
Fred Drake8c081a12001-10-12 20:57:55 +00001372
1373 if (PyArg_ParseTuple(args, "s:logreader", &filename)) {
1374 self = PyObject_New(LogReaderObject, &LogReaderType);
1375 if (self != NULL) {
Fred Drake30d1c752001-10-15 22:11:02 +00001376 self->frametimings = 1;
Fred Drake8c081a12001-10-12 20:57:55 +00001377 self->linetimings = 0;
Fred Drake4c2e1af2001-10-29 20:45:57 +00001378 self->info = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001379 self->logfp = fopen(filename, "rb");
1380 if (self->logfp == NULL) {
1381 PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
1382 Py_DECREF(self);
1383 self = NULL;
Fred Drake4c2e1af2001-10-29 20:45:57 +00001384 goto finally;
1385 }
1386 self->info = PyDict_New();
1387 if (self->info == NULL) {
1388 Py_DECREF(self);
1389 goto finally;
1390 }
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001391 /* read initial info */
1392 for (;;) {
1393 if ((c = fgetc(self->logfp)) == EOF) {
Fred Drake666bf522002-07-18 19:11:44 +00001394 eof_error(self);
Neil Schemenauer8b6b4912002-05-29 18:19:14 +00001395 break;
1396 }
1397 if (c != WHAT_ADD_INFO) {
1398 ungetc(c, self->logfp);
1399 break;
1400 }
1401 err = unpack_add_info(self);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001402 if (err) {
1403 if (err == ERR_EOF)
Fred Drake666bf522002-07-18 19:11:44 +00001404 eof_error(self);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001405 else
1406 PyErr_SetString(PyExc_RuntimeError,
1407 "unexpected error");
1408 break;
1409 }
Fred Drake8c081a12001-10-12 20:57:55 +00001410 }
1411 }
1412 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001413 finally:
Fred Drake8c081a12001-10-12 20:57:55 +00001414 return (PyObject *) self;
1415}
1416
1417
1418/* Return a Python string that represents the version number without the
1419 * extra cruft added by revision control, even if the right options were
1420 * given to the "cvs export" command to make it not include the extra
1421 * cruft.
1422 */
1423static char *
1424get_version_string(void)
1425{
1426 static char *rcsid = "$Revision$";
1427 char *rev = rcsid;
1428 char *buffer;
1429 int i = 0;
1430
Neal Norwitz3afb2d22002-03-20 21:32:07 +00001431 while (*rev && !isdigit((int)*rev))
Fred Drake8c081a12001-10-12 20:57:55 +00001432 ++rev;
1433 while (rev[i] != ' ' && rev[i] != '\0')
1434 ++i;
1435 buffer = malloc(i + 1);
1436 if (buffer != NULL) {
1437 memmove(buffer, rev, i);
1438 buffer[i] = '\0';
1439 }
1440 return buffer;
1441}
1442
1443/* Write out a RFC 822-style header with various useful bits of
1444 * information to make the output easier to manage.
1445 */
1446static int
1447write_header(ProfilerObject *self)
1448{
1449 char *buffer;
1450 char cwdbuffer[PATH_MAX];
1451 PyObject *temp;
1452 int i, len;
1453
1454 buffer = get_version_string();
1455 if (buffer == NULL) {
1456 PyErr_NoMemory();
1457 return -1;
1458 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001459 pack_add_info(self, "hotshot-version", buffer);
1460 pack_add_info(self, "requested-frame-timings",
1461 (self->frametimings ? "yes" : "no"));
1462 pack_add_info(self, "requested-line-events",
Fred Drake8c081a12001-10-12 20:57:55 +00001463 (self->lineevents ? "yes" : "no"));
Fred Drake4c2e1af2001-10-29 20:45:57 +00001464 pack_add_info(self, "requested-line-timings",
1465 (self->linetimings ? "yes" : "no"));
1466 pack_add_info(self, "platform", Py_GetPlatform());
1467 pack_add_info(self, "executable", Py_GetProgramFullPath());
Fred Drakef12a68c2001-11-09 15:59:36 +00001468 free(buffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001469 buffer = (char *) Py_GetVersion();
1470 if (buffer == NULL)
1471 PyErr_Clear();
1472 else
Fred Drake4c2e1af2001-10-29 20:45:57 +00001473 pack_add_info(self, "executable-version", buffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001474
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001475#ifdef MS_WINDOWS
Tim Peters885d4572001-11-28 20:27:42 +00001476 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%I64d", frequency.QuadPart);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001477 pack_add_info(self, "reported-performance-frequency", cwdbuffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001478#else
Tim Peters885d4572001-11-28 20:27:42 +00001479 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%lu", rusage_diff);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001480 pack_add_info(self, "observed-interval-getrusage", cwdbuffer);
Tim Peters885d4572001-11-28 20:27:42 +00001481 PyOS_snprintf(cwdbuffer, sizeof(cwdbuffer), "%lu", timeofday_diff);
Fred Drake4c2e1af2001-10-29 20:45:57 +00001482 pack_add_info(self, "observed-interval-gettimeofday", cwdbuffer);
Fred Drake8c081a12001-10-12 20:57:55 +00001483#endif
Fred Drake8c081a12001-10-12 20:57:55 +00001484
Fred Drake4c2e1af2001-10-29 20:45:57 +00001485 pack_add_info(self, "current-directory",
Fred Drake8c081a12001-10-12 20:57:55 +00001486 getcwd(cwdbuffer, sizeof cwdbuffer));
1487
1488 temp = PySys_GetObject("path");
1489 len = PyList_GET_SIZE(temp);
1490 for (i = 0; i < len; ++i) {
1491 PyObject *item = PyList_GET_ITEM(temp, i);
1492 buffer = PyString_AsString(item);
Fred Draked1eb8b62002-07-17 18:54:20 +00001493 if (buffer == NULL) {
1494 pack_add_info(self, "sys-path-entry", "<non-string-path-entry>");
1495 PyErr_Clear();
1496 }
1497 else {
1498 pack_add_info(self, "sys-path-entry", buffer);
1499 }
Fred Drake8c081a12001-10-12 20:57:55 +00001500 }
Fred Drake4c2e1af2001-10-29 20:45:57 +00001501 pack_frame_times(self);
1502 pack_line_times(self);
1503
Fred Drake8c081a12001-10-12 20:57:55 +00001504 return 0;
1505}
1506
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001507PyDoc_STRVAR(profiler__doc__,
1508"profiler(logfilename[, lineevents[, linetimes]]) -> profiler\n\
1509Create a new profiler object.");
Fred Drake8c081a12001-10-12 20:57:55 +00001510
1511static PyObject *
1512hotshot_profiler(PyObject *unused, PyObject *args)
1513{
1514 char *logfilename;
1515 ProfilerObject *self = NULL;
1516 int lineevents = 0;
1517 int linetimings = 1;
1518
1519 if (PyArg_ParseTuple(args, "s|ii:profiler", &logfilename,
1520 &lineevents, &linetimings)) {
1521 self = PyObject_New(ProfilerObject, &ProfilerType);
1522 if (self == NULL)
1523 return NULL;
Fred Drake30d1c752001-10-15 22:11:02 +00001524 self->frametimings = 1;
Fred Drake8c081a12001-10-12 20:57:55 +00001525 self->lineevents = lineevents ? 1 : 0;
1526 self->linetimings = (lineevents && linetimings) ? 1 : 0;
1527 self->index = 0;
1528 self->active = 0;
1529 self->next_fileno = 0;
Tim Peters1566a172001-10-12 22:08:39 +00001530 self->logfp = NULL;
Fred Drake8c081a12001-10-12 20:57:55 +00001531 self->logfilename = PyTuple_GET_ITEM(args, 0);
1532 Py_INCREF(self->logfilename);
1533 self->filemap = PyDict_New();
1534 if (self->filemap == NULL) {
1535 Py_DECREF(self);
1536 return NULL;
1537 }
1538 self->logfp = fopen(logfilename, "wb");
1539 if (self->logfp == NULL) {
1540 Py_DECREF(self);
1541 PyErr_SetFromErrnoWithFilename(PyExc_IOError, logfilename);
1542 return NULL;
1543 }
1544 if (timeofday_diff == 0) {
1545 /* Run this several times since sometimes the first
1546 * doesn't give the lowest values, and we're really trying
1547 * to determine the lowest.
1548 */
1549 calibrate();
1550 calibrate();
1551 calibrate();
1552 }
1553 if (write_header(self))
1554 /* some error occurred, exception has been set */
1555 self = NULL;
1556 }
1557 return (PyObject *) self;
1558}
1559
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001560PyDoc_STRVAR(coverage__doc__,
1561"coverage(logfilename) -> profiler\n\
Fred Drake30d1c752001-10-15 22:11:02 +00001562Returns a profiler that doesn't collect any timing information, which is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001563useful in building a coverage analysis tool.");
Fred Drake30d1c752001-10-15 22:11:02 +00001564
1565static PyObject *
1566hotshot_coverage(PyObject *unused, PyObject *args)
1567{
1568 char *logfilename;
1569 PyObject *result = NULL;
1570
1571 if (PyArg_ParseTuple(args, "s:coverage", &logfilename)) {
1572 result = hotshot_profiler(unused, args);
1573 if (result != NULL) {
1574 ProfilerObject *self = (ProfilerObject *) result;
1575 self->frametimings = 0;
1576 self->linetimings = 0;
1577 self->lineevents = 1;
1578 }
1579 }
1580 return result;
1581}
1582
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001583PyDoc_VAR(resolution__doc__) =
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001584#ifdef MS_WINDOWS
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001585PyDoc_STR(
Tim Petersfeab23f2001-10-13 00:11:10 +00001586"resolution() -> (performance-counter-ticks, update-frequency)\n"
1587"Return the resolution of the timer provided by the QueryPerformanceCounter()\n"
1588"function. The first value is the smallest observed change, and the second\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001589"is the result of QueryPerformanceFrequency()."
1590)
Fred Drake8c081a12001-10-12 20:57:55 +00001591#else
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001592PyDoc_STR(
Tim Petersfeab23f2001-10-13 00:11:10 +00001593"resolution() -> (gettimeofday-usecs, getrusage-usecs)\n"
1594"Return the resolution of the timers provided by the gettimeofday() and\n"
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001595"getrusage() system calls, or -1 if the call is not supported."
1596)
Fred Drake8c081a12001-10-12 20:57:55 +00001597#endif
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001598;
Fred Drake8c081a12001-10-12 20:57:55 +00001599
1600static PyObject *
1601hotshot_resolution(PyObject *unused, PyObject *args)
1602{
1603 PyObject *result = NULL;
1604
1605 if (PyArg_ParseTuple(args, ":resolution")) {
1606 if (timeofday_diff == 0) {
1607 calibrate();
1608 calibrate();
1609 calibrate();
1610 }
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001611#ifdef MS_WINDOWS
Fred Drake8c081a12001-10-12 20:57:55 +00001612 result = Py_BuildValue("ii", timeofday_diff, frequency.LowPart);
1613#else
1614 result = Py_BuildValue("ii", timeofday_diff, rusage_diff);
1615#endif
1616 }
1617 return result;
1618}
1619
1620
1621static PyMethodDef functions[] = {
Fred Drake30d1c752001-10-15 22:11:02 +00001622 {"coverage", hotshot_coverage, METH_VARARGS, coverage__doc__},
Fred Drake8c081a12001-10-12 20:57:55 +00001623 {"profiler", hotshot_profiler, METH_VARARGS, profiler__doc__},
1624 {"logreader", hotshot_logreader, METH_VARARGS, logreader__doc__},
1625 {"resolution", hotshot_resolution, METH_VARARGS, resolution__doc__},
1626 {NULL, NULL}
1627};
1628
1629
1630void
1631init_hotshot(void)
1632{
1633 PyObject *module;
1634
1635 LogReaderType.ob_type = &PyType_Type;
1636 ProfilerType.ob_type = &PyType_Type;
1637 module = Py_InitModule("_hotshot", functions);
1638 if (module != NULL) {
1639 char *s = get_version_string();
1640
1641 PyModule_AddStringConstant(module, "__version__", s);
1642 free(s);
1643 Py_INCREF(&LogReaderType);
1644 PyModule_AddObject(module, "LogReaderType",
1645 (PyObject *)&LogReaderType);
1646 Py_INCREF(&ProfilerType);
1647 PyModule_AddObject(module, "ProfilerType",
1648 (PyObject *)&ProfilerType);
1649
1650 if (ProfilerError == NULL)
1651 ProfilerError = PyErr_NewException("hotshot.ProfilerError",
1652 NULL, NULL);
1653 if (ProfilerError != NULL) {
1654 Py_INCREF(ProfilerError);
1655 PyModule_AddObject(module, "ProfilerError", ProfilerError);
1656 }
1657 PyModule_AddIntConstant(module, "WHAT_ENTER", WHAT_ENTER);
1658 PyModule_AddIntConstant(module, "WHAT_EXIT", WHAT_EXIT);
1659 PyModule_AddIntConstant(module, "WHAT_LINENO", WHAT_LINENO);
1660 PyModule_AddIntConstant(module, "WHAT_OTHER", WHAT_OTHER);
1661 PyModule_AddIntConstant(module, "WHAT_ADD_INFO", WHAT_ADD_INFO);
1662 PyModule_AddIntConstant(module, "WHAT_DEFINE_FILE", WHAT_DEFINE_FILE);
Fred Drake30d1c752001-10-15 22:11:02 +00001663 PyModule_AddIntConstant(module, "WHAT_DEFINE_FUNC", WHAT_DEFINE_FUNC);
Fred Drake8c081a12001-10-12 20:57:55 +00001664 PyModule_AddIntConstant(module, "WHAT_LINE_TIMES", WHAT_LINE_TIMES);
1665 }
1666}