зеркало из https://github.com/mozilla/pjs.git
345 строки
8.5 KiB
C
345 строки
8.5 KiB
C
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/*
|
|
* The contents of this file are subject to the Mozilla Public
|
|
* License Version 1.1 (the "License"); you may not use this file
|
|
* except in compliance with the License. You may obtain a copy of
|
|
* the License at http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS
|
|
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
|
* implied. See the License for the specific language governing
|
|
* rights and limitations under the License.
|
|
*
|
|
* The Original Code is the Netscape security libraries.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998-2000 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*
|
|
* Alternatively, the contents of this file may be used under the
|
|
* terms of the GNU General Public License Version 2 or later (the
|
|
* "GPL"), in which case the provisions of the GPL are applicable
|
|
* instead of those above. If you wish to allow use of your
|
|
* version of this file only under the terms of the GPL and not to
|
|
* allow others to use your version of this file under the MPL,
|
|
* indicate your decision by deleting the provisions above and
|
|
* replace them with the notice and other provisions required by
|
|
* the GPL. If you do not delete the provisions above, a recipient
|
|
* may use your version of this file under either the MPL or the
|
|
* GPL.
|
|
*/
|
|
|
|
/***********************************************************************
|
|
**
|
|
** Name: dbmalloc.c
|
|
**
|
|
** Description: Testing malloc (OBSOLETE)
|
|
**
|
|
** Modification History:
|
|
** 14-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
|
|
** The debug mode will print all of the printfs associated with this test.
|
|
** The regress mode will be the default mode. Since the regress tool limits
|
|
** the output to a one line status:PASS or FAIL,all of the printf statements
|
|
** have been handled with an if (debug_mode) statement.
|
|
***********************************************************************/
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <time.h>
|
|
#include <string.h>
|
|
#include "nspr.h"
|
|
|
|
void
|
|
usage
|
|
(
|
|
void
|
|
)
|
|
{
|
|
fprintf(stderr, "Usage: dbmalloc ('-m'|'-s') '-f' num_fails ('-d'|'-n') filename [...]\n");
|
|
exit(0);
|
|
}
|
|
|
|
typedef struct node_struct
|
|
{
|
|
struct node_struct *next, *prev;
|
|
int line;
|
|
char value[4];
|
|
}
|
|
node_t,
|
|
*node_pt;
|
|
|
|
node_pt get_node(const char *line)
|
|
{
|
|
node_pt rv;
|
|
int l = strlen(line);
|
|
rv = (node_pt)PR_MALLOC(sizeof(node_t) + l + 1 - 4);
|
|
if( (node_pt)0 == rv ) return (node_pt)0;
|
|
memcpy(&rv->value[0], line, l+1);
|
|
rv->next = rv->prev = (node_pt)0;
|
|
return rv;
|
|
}
|
|
|
|
void
|
|
dump
|
|
(
|
|
const char *name,
|
|
node_pt node,
|
|
int mf,
|
|
int debug
|
|
)
|
|
{
|
|
if( (node_pt)0 != node->prev ) dump(name, node->prev, mf, debug);
|
|
if( 0 != debug ) printf("[%s]: %6d: %s", name, node->line, node->value);
|
|
if( node->line == mf ) fprintf(stderr, "[%s]: Line %d was allocated!\n", name, node->line);
|
|
if( (node_pt)0 != node->next ) dump(name, node->next, mf, debug);
|
|
return;
|
|
}
|
|
|
|
void
|
|
release
|
|
(
|
|
node_pt node
|
|
)
|
|
{
|
|
if( (node_pt)0 != node->prev ) release(node->prev);
|
|
if( (node_pt)0 != node->next ) release(node->next);
|
|
PR_DELETE(node);
|
|
}
|
|
|
|
int
|
|
t2
|
|
(
|
|
const char *name,
|
|
int mf,
|
|
int debug
|
|
)
|
|
{
|
|
int rv;
|
|
FILE *fp;
|
|
int l = 0;
|
|
node_pt head = (node_pt)0;
|
|
char buffer[ BUFSIZ ];
|
|
|
|
fp = fopen(name, "r");
|
|
if( (FILE *)0 == fp )
|
|
{
|
|
fprintf(stderr, "[%s]: Cannot open \"%s.\"\n", name, name);
|
|
return -1;
|
|
}
|
|
|
|
/* fgets mallocs a buffer, first time through. */
|
|
if( (char *)0 == fgets(buffer, BUFSIZ, fp) )
|
|
{
|
|
fprintf(stderr, "[%s]: \"%s\" is empty.\n", name, name);
|
|
(void)fclose(fp);
|
|
return -1;
|
|
}
|
|
|
|
rewind(fp);
|
|
|
|
if( PR_SUCCESS != PR_ClearMallocCount() )
|
|
{
|
|
fprintf(stderr, "[%s]: Cannot clear malloc count.\n", name);
|
|
(void)fclose(fp);
|
|
return -1;
|
|
}
|
|
|
|
if( PR_SUCCESS != PR_SetMallocCountdown(mf) )
|
|
{
|
|
fprintf(stderr, "[%s]: Cannot set malloc countdown to %d\n", name, mf);
|
|
(void)fclose(fp);
|
|
return -1;
|
|
}
|
|
|
|
while( fgets(buffer, BUFSIZ, fp) )
|
|
{
|
|
node_pt n;
|
|
node_pt *w = &head;
|
|
|
|
if( (strlen(buffer) == (BUFSIZ-1)) && (buffer[BUFSIZ-2] != '\n') )
|
|
buffer[BUFSIZ-2] == '\n';
|
|
|
|
l++;
|
|
|
|
n = get_node(buffer);
|
|
if( (node_pt)0 == n )
|
|
{
|
|
printf("[%s]: Line %d: malloc failure!\n", name, l);
|
|
continue;
|
|
}
|
|
|
|
n->line = l;
|
|
|
|
while( 1 )
|
|
{
|
|
int comp;
|
|
|
|
if( (node_pt)0 == *w )
|
|
{
|
|
*w = n;
|
|
break;
|
|
}
|
|
|
|
comp = strcmp((*w)->value, n->value);
|
|
if( comp < 0 ) w = &(*w)->next;
|
|
else w = &(*w)->prev;
|
|
}
|
|
}
|
|
|
|
(void)fclose(fp);
|
|
|
|
dump(name, head, mf, debug);
|
|
|
|
rv = PR_GetMallocCount();
|
|
PR_ClearMallocCountdown();
|
|
|
|
release(head);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int nf = 0;
|
|
int debug = 0;
|
|
|
|
void
|
|
test
|
|
(
|
|
const char *name
|
|
)
|
|
{
|
|
int n, i;
|
|
|
|
extern int nf, debug;
|
|
|
|
printf("[%s]: starting test 0\n", name);
|
|
n = t2(name, 0, debug);
|
|
if( -1 == n ) return;
|
|
printf("[%s]: test 0 had %ld allocations.\n", name, n);
|
|
|
|
if( 0 >= n ) return;
|
|
|
|
for( i = 0; i < nf; i++ )
|
|
{
|
|
int which = rand() % n;
|
|
if( 0 == which ) printf("[%s]: starting test %d -- no allocation should fail\n", name, i+1);
|
|
else printf("[%s]: starting test %d -- allocation %d should fail\n", name, i+1, which);
|
|
(void)t2(name, which, debug);
|
|
printf("[%s]: test %d done.\n", name, i+1);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
int
|
|
main
|
|
(
|
|
int argc,
|
|
char *argv[]
|
|
)
|
|
{
|
|
int okay = 0;
|
|
int multithread = 0;
|
|
|
|
struct threadlist
|
|
{
|
|
struct threadlist *next;
|
|
PRThread *thread;
|
|
}
|
|
*threadhead = (struct threadlist *)0;
|
|
|
|
extern int nf, debug;
|
|
|
|
srand(time(0));
|
|
|
|
PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
|
|
PR_STDIO_INIT();
|
|
|
|
printf("[main]: We %s using the debugging malloc.\n",
|
|
PR_IsDebuggingMalloc() ? "ARE" : "ARE NOT");
|
|
|
|
while( argv++, --argc )
|
|
{
|
|
if( '-' == argv[0][0] )
|
|
{
|
|
switch( argv[0][1] )
|
|
{
|
|
case 'f':
|
|
nf = atoi(argv[0][2] ? &argv[0][2] :
|
|
--argc ? *++argv : "0");
|
|
break;
|
|
case 'd':
|
|
debug = 1;
|
|
break;
|
|
case 'n':
|
|
debug = 0;
|
|
break;
|
|
case 'm':
|
|
multithread = 1;
|
|
break;
|
|
case 's':
|
|
multithread = 0;
|
|
break;
|
|
default:
|
|
usage();
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
FILE *fp = fopen(*argv, "r");
|
|
if( (FILE *)0 == fp )
|
|
{
|
|
fprintf(stderr, "Cannot open \"%s.\"\n", *argv);
|
|
continue;
|
|
}
|
|
|
|
okay++;
|
|
(void)fclose(fp);
|
|
if( multithread )
|
|
{
|
|
struct threadlist *n;
|
|
|
|
n = (struct threadlist *)malloc(sizeof(struct threadlist));
|
|
if( (struct threadlist *)0 == n )
|
|
{
|
|
fprintf(stderr, "This is getting tedious. \"%s\"\n", *argv);
|
|
continue;
|
|
}
|
|
|
|
n->next = threadhead;
|
|
n->thread = PR_CreateThread(PR_USER_THREAD, (void (*)(void *))test,
|
|
*argv, PR_PRIORITY_NORMAL,
|
|
PR_LOCAL_THREAD, PR_JOINABLE_THREAD,
|
|
0);
|
|
if( (PRThread *)0 == n->thread )
|
|
{
|
|
fprintf(stderr, "Can't create thread for \"%s.\"\n", *argv);
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
threadhead = n;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
test(*argv);
|
|
}
|
|
}
|
|
}
|
|
|
|
if( okay == 0 ) usage();
|
|
else while( (struct threadlist *)0 != threadhead )
|
|
{
|
|
struct threadlist *x = threadhead->next;
|
|
(void)PR_JoinThread(threadhead->thread);
|
|
PR_DELETE(threadhead);
|
|
threadhead = x;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|