42 |
|
#include <sys/param.h> |
43 |
|
#include <sys/sysctl.h> |
44 |
|
#endif |
45 |
+ |
#ifdef HPUX |
46 |
+ |
#include <sys/param.h> |
47 |
+ |
#include <sys/pstat.h> |
48 |
+ |
#endif |
49 |
|
|
50 |
|
#include "tools.h" |
51 |
|
#include "statgrab.h" |
64 |
|
#include <sys/fcntl.h> |
65 |
|
#include <dirent.h> |
66 |
|
#endif |
67 |
+ |
#ifdef WIN32 |
68 |
+ |
#include "win32.h" |
69 |
+ |
#endif |
70 |
|
|
71 |
|
#if defined(SOLARIS) && defined(HAVE_LIBDEVINFO_H) |
72 |
|
struct map{ |
237 |
|
if(strcmp(ksp->ks_module, "md")==0) continue; |
238 |
|
if((kstat_read(kc, ksp, &kios))==-1) continue; |
239 |
|
strncpy(device_name, ksp->ks_name, sizeof device_name); |
240 |
< |
for(x=0;x<(sizeof device_name);x++){ |
240 |
> |
for(x=0;x<(int)(sizeof device_name);x++){ |
241 |
|
if( isdigit((int)device_name[x]) ) break; |
242 |
|
} |
243 |
|
if(x == sizeof device_name) x--; |
297 |
|
return match_string; |
298 |
|
} |
299 |
|
|
300 |
< |
long long sg_get_ll_match(char *line, regmatch_t *match){ |
294 |
< |
char *ptr; |
295 |
< |
long long num; |
296 |
< |
|
297 |
< |
ptr=line+match->rm_so; |
298 |
< |
num=atoll(ptr); |
299 |
< |
|
300 |
< |
return num; |
301 |
< |
} |
302 |
< |
#endif |
303 |
< |
|
300 |
> |
/* Cygwin (without a recent newlib) doesn't have atoll */ |
301 |
|
#ifndef HAVE_ATOLL |
302 |
|
static long long atoll(const char *s) { |
303 |
|
long long value = 0; |
318 |
|
} |
319 |
|
#endif |
320 |
|
|
321 |
+ |
long long sg_get_ll_match(char *line, regmatch_t *match){ |
322 |
+ |
char *ptr; |
323 |
+ |
long long num; |
324 |
+ |
|
325 |
+ |
ptr=line+match->rm_so; |
326 |
+ |
num=atoll(ptr); |
327 |
+ |
|
328 |
+ |
return num; |
329 |
+ |
} |
330 |
+ |
#endif |
331 |
+ |
|
332 |
|
/* $OpenBSD: strlcpy.c,v 1.8 2003/06/17 21:56:24 millert Exp $ */ |
333 |
|
|
334 |
|
/* |
447 |
|
return 0; |
448 |
|
} |
449 |
|
|
450 |
+ |
/* join two strings together */ |
451 |
+ |
int sg_concat_string(char **dest, const char *src) { |
452 |
+ |
char *new; |
453 |
+ |
int len = strlen(*dest) + strlen(src) + 1; |
454 |
+ |
|
455 |
+ |
new = sg_realloc(*dest, len); |
456 |
+ |
if (new == NULL) { |
457 |
+ |
return -1; |
458 |
+ |
} |
459 |
+ |
|
460 |
+ |
*dest = new; |
461 |
+ |
strcat(*dest, src); |
462 |
+ |
return 0; |
463 |
+ |
} |
464 |
+ |
|
465 |
|
#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD) |
466 |
|
kvm_t *sg_get_kvm() { |
467 |
|
static kvm_t *kvmd = NULL; |
504 |
|
mib[1] = VM_UVMEXP; |
505 |
|
|
506 |
|
if (sysctl(mib, 2, &uvm, &size, NULL, 0) < 0) { |
507 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_VM.VM_UVMEXP"); |
507 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_VM.VM_UVMEXP"); |
508 |
|
return NULL; |
509 |
|
} |
510 |
|
|
512 |
|
} |
513 |
|
#endif |
514 |
|
|
515 |
+ |
#ifdef HPUX |
516 |
+ |
struct pst_static *sg_get_pstat_static() { |
517 |
+ |
static int got = 0; |
518 |
+ |
static struct pst_static pst; |
519 |
+ |
|
520 |
+ |
if (!got) { |
521 |
+ |
if (pstat_getstatic(&pst, sizeof pst, 1, 0) == -1) { |
522 |
+ |
sg_set_error_with_errno(SG_ERROR_PSTAT, |
523 |
+ |
"pstat_static"); |
524 |
+ |
return NULL; |
525 |
+ |
} |
526 |
+ |
got = 1; |
527 |
+ |
} |
528 |
+ |
return &pst; |
529 |
+ |
} |
530 |
+ |
#endif |
531 |
+ |
|
532 |
|
int sg_init(){ |
533 |
< |
sg_set_error_with_errno(SG_ERROR_NONE, NULL, 0); |
533 |
> |
sg_set_error(SG_ERROR_NONE, NULL); |
534 |
|
|
535 |
|
#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD) |
536 |
|
if (sg_get_kvm() == NULL) { |
549 |
|
build_mapping(); |
550 |
|
#endif |
551 |
|
#endif |
552 |
+ |
#ifdef WIN32 |
553 |
+ |
return sg_win32_start_capture(); |
554 |
+ |
#endif |
555 |
|
return 0; |
556 |
|
} |
557 |
|
|
558 |
+ |
int sg_shutdown() { |
559 |
+ |
#ifdef WIN32 |
560 |
+ |
sg_win32_end_capture(); |
561 |
+ |
#endif |
562 |
+ |
return 0; |
563 |
+ |
} |
564 |
+ |
|
565 |
+ |
int sg_snapshot() { |
566 |
+ |
#ifdef WIN32 |
567 |
+ |
return sg_win32_snapshot(); |
568 |
+ |
#else |
569 |
+ |
return 0; |
570 |
+ |
#endif |
571 |
+ |
} |
572 |
+ |
|
573 |
|
int sg_drop_privileges() { |
574 |
+ |
#ifndef WIN32 |
575 |
+ |
#ifdef HAVE_SETEGID |
576 |
|
if (setegid(getgid()) != 0) { |
577 |
< |
sg_set_error(SG_ERROR_SETEGID, NULL); |
577 |
> |
#elif defined(HAVE_SETRESGID) |
578 |
> |
if (setresgid(getgid(), getgid(), getgid()) != 0) { |
579 |
> |
#else |
580 |
> |
{ |
581 |
> |
#endif |
582 |
> |
sg_set_error_with_errno(SG_ERROR_SETEGID, NULL); |
583 |
|
return -1; |
584 |
|
} |
585 |
+ |
#ifdef HAVE_SETEUID |
586 |
|
if (seteuid(getuid()) != 0) { |
587 |
< |
sg_set_error(SG_ERROR_SETEUID, NULL); |
587 |
> |
#elif defined(HAVE_SETRESUID) |
588 |
> |
if (setresuid(getuid(), getuid(), getuid()) != 0) { |
589 |
> |
#else |
590 |
> |
{ |
591 |
> |
#endif |
592 |
> |
sg_set_error_with_errno(SG_ERROR_SETEUID, NULL); |
593 |
|
return -1; |
594 |
|
} |
595 |
+ |
#endif /* WIN32 */ |
596 |
|
return 0; |
597 |
|
} |
598 |
|
|
600 |
|
void *tmp = NULL; |
601 |
|
tmp = realloc(ptr, size); |
602 |
|
if(tmp == NULL) { |
603 |
< |
sg_set_error(SG_ERROR_MALLOC, NULL); |
603 |
> |
sg_set_error_with_errno(SG_ERROR_MALLOC, NULL); |
604 |
|
} |
605 |
|
return tmp; |
606 |
|
} |
535 |
– |
|