common.c

来自「NAT打洞」· C语言 代码 · 共 35 行

#include "common.h"
#include "arraylist.h"

#if !defined(WIN32) && !defined(__WIN32__)
ARRAYLIST *keylist;

static int started = 0;

void clean_keylist(void){
	int i;
	for(i = 0; i < arraylist_size(keylist); i++){
  	char *tmp = (char*)arraylist_get(keylist, i);
  	unlink(tmp);
    free(tmp);
  }
  free(keylist->list);
  free(keylist);
}

key_t get_key(void){
	char temp[255];
  strcpy(temp, "/tmp/pckeyXXXXXX");
  if(mkstemp(temp) != -1){
    if(!started){
    	keylist = arraylist();
      started = 1;
    }
    arraylist_add(keylist, strdup(temp));
		return ftok(temp, getpid());
  }
  return -1;
}
#endif