/* new memory functions */

#define LOG_MISC 32
#define MEMTBLSIZE 6666
#undef DEBUG

#include <stdio.h>

extern int serv;

unsigned memused=0;

#ifdef DEBUG
struct {
  void *ptr;
  int size;
  char file[20];
  int line;
} memtbl[MEMTBLSIZE];
#endif

/* initialize the memory structure */
void init_mem()
{
#ifdef DEBUG
  int i;
  for (i=0; i<MEMTBLSIZE; i++) memtbl[i].ptr=NULL;
#endif
}

/* tell someone the gory memory details */
void tell_mem_status(nick)
char *nick;
{
#ifdef DEBUG
  int i,tot=0; float per;
  for (i=0; i<MEMTBLSIZE; i++) if (memtbl[i].ptr!=NULL) tot++;
  per=((tot*1.0)/(MEMTBLSIZE*1.0))*100.0;
  mprintf(serv,"NOTICE %s :Memory table usage: %d/%d (%.1f%% full)\n",nick,
	  tot,MEMTBLSIZE,per);
#endif
  mprintf(serv,"NOTICE %s :Total memory allocated = %d bytes\n",nick,memused);
}

void tell_mem_status_dcc(z)
int z;
{
  int exp,i,tot=0; float per;
  exp=expected_memory();  /* in io.c ? */
#ifdef DEBUG
  for (i=0; i<MEMTBLSIZE; i++) if (memtbl[i].ptr!=NULL) tot++;
  per=((tot*1.0)/(MEMTBLSIZE*1.0))*100.0;
  tprintf(z,"Memory table usage: %d/%d (%.1f%% full)\n",tot,MEMTBLSIZE,per);
#endif
  per=((exp*1.0)/(memused*1.0))*100.0;
  if (per!=100.0)
    tprintf(z,"Memory fault: only accounting for %d/%d (%.1f%%)\n",
	    exp,memused,per);
#ifdef DEBUG
  tprintf(z,"Memory table itself occupies an additional %d bytes static\n",
	  MEMTBLSIZE*32);
#endif
}

void debug_mem_to_dcc(z)
int z;
{
#ifdef DEBUG
  unsigned long exp[7],use[7],l; int i,j; char fn[20],sofar[81];
  exp[0]=expmem_chan();
  exp[1]=expmem_chanprog();
  exp[2]=expmem_misc();
  exp[3]=expmem_edit();
  exp[4]=expmem_users();
  exp[5]=expmem_proc();
  exp[6]=expmem_dcc();
  for (i=0; i<7; i++) use[i]=0;
  for (i=0; i<MEMTBLSIZE; i++) if (memtbl[i].ptr!=NULL) {
    strcpy(fn,memtbl[i].file); l=memtbl[i].size;
    if (strcasecmp(fn,"chan.c")==0) use[0]+=l;
    if (strcasecmp(fn,"chanprog.c")==0) use[1]+=l;
    if (strcasecmp(fn,"misc.c")==0) use[2]+=l;
    if (strcasecmp(fn,"edit.c")==0) use[3]+=l;
    if (strcasecmp(fn,"userrec.c")==0) use[4]+=l;
    if (strcasecmp(fn,"proc.c")==0) use[5]+=l;
    if (strcasecmp(fn,"dcc.c")==0) use[6]+=l;
  }
  for (i=0; i<7; i++) {
    switch(i) {
    case 0: strcpy(fn,"chan.c"); break;
    case 1: strcpy(fn,"chanprog.c"); break;
    case 2: strcpy(fn,"misc.c"); break;
    case 3: strcpy(fn,"edit.c"); break;
    case 4: strcpy(fn,"userrec.c"); break;
    case 5: strcpy(fn,"proc.c"); break;
    case 6: strcpy(fn,"dcc.c"); break;
    }
    if (use[i]==exp[i])
      tprintf(z,"File '%-10s' accounted for %lu/%lu (ok)\n",fn,exp[i],use[i]);
    else {
      tprintf(z,"File '%-10s' accounted for %lu/%lu (debug follows:)\n",fn,
	      exp[i],use[i]);
      strcpy(sofar,"   ");
      for (j=0; j<MEMTBLSIZE; j++) 
	if (memtbl[j].ptr!=NULL) if (strcasecmp(memtbl[j].file,fn)==0) {
	  sprintf(&sofar[strlen(sofar)],"%-3d:(%03X) ",memtbl[j].line,
		  memtbl[j].size);
	  if (strlen(sofar)>60) {
	    sofar[strlen(sofar)-1]=0;
	    tprintf(z,"%s\n",sofar);
	    strcpy(sofar,"   ");
	  }
	}
      if (sofar[0]) {
	sofar[strlen(sofar)-1]=0;
	tprintf(z,"%s\n",sofar);
      }
    }
  }
  tprintf(z,"--- End of debug memory list.\n");
#else
  tprintf(z,"Compiled without extensive memory debugging (sorry).\n");
#endif
}

void *n_malloc(size,file,line)
int size,line; char *file;
{
  void *x; int i;
#ifdef DEBUG
  x=(void *)malloc(size);
  if (x==NULL) {
    log(LOG_MISC,"*** FAILED MALLOC %s (%d)",file,line);
    return NULL;
  }
  for (i=0; (i<MEMTBLSIZE) && (memtbl[i].ptr!=NULL); i++);
  if (i==MEMTBLSIZE) {
    log(LOG_MISC,"*** MEMORY TABLE FULL: %s (%d)",file,line);
    return x;
  }
  memtbl[i].ptr=x; memtbl[i].line=line; memtbl[i].size=size;
  strcpy(memtbl[i].file,file);
#else
  x=(void *)malloc(size+4); *(int *)(x)=size; 
  (char *)x+=4;
#endif
  memused+=size;
  return x;
}

void n_free(ptr,file,line)
void *ptr; char *file; int line;
{
  int i;
#ifdef DEBUG
  if (ptr==NULL) {
    log(LOG_MISC,"*** ATTEMPTING TO FREE NULL PTR: %s (%d)",file,line);
    return;
  }
  for (i=0; (i<MEMTBLSIZE) && (memtbl[i].ptr!=ptr); i++);
  if (i==MEMTBLSIZE) {
    log(LOG_MISC,"*** ATTEMPTING TO FREE NON-MALLOC'D PTR: %s (%d)",file,line);
    return;
  }
  memtbl[i].ptr=NULL;
  memused-=memtbl[i].size; free(ptr);
#else
  memused-=*(int *)((char *)(ptr)-4); free((char *)(ptr)-4);
#endif
}