| 1 |
#ifndef lint |
| 2 |
static const char RCSid[] = "$Id: savestr.c,v 2.13 2021/05/07 03:09:45 greg Exp $"; |
| 3 |
#endif |
| 4 |
/* |
| 5 |
* savestr.c - routines for efficient string storage. |
| 6 |
* |
| 7 |
* Savestr(s) stores a shared read-only string. |
| 8 |
* Freestr(s) indicates a client is finished with a string. |
| 9 |
* All strings must be null-terminated. There is |
| 10 |
* no imposed length limit. |
| 11 |
* Strings stored with savestr(s) can be equated |
| 12 |
* reliably using their pointer values. |
| 13 |
* Calls to savestr(s) and freestr(s) should be |
| 14 |
* balanced (obviously). The last call to freestr(s) |
| 15 |
* frees memory associated with the string; it should |
| 16 |
* never be referenced again. |
| 17 |
* |
| 18 |
* External symbols declared in rtio.h |
| 19 |
*/ |
| 20 |
|
| 21 |
#include "copyright.h" |
| 22 |
|
| 23 |
#include <stdlib.h> |
| 24 |
|
| 25 |
#include "rterror.h" |
| 26 |
#include "rtio.h" |
| 27 |
|
| 28 |
#ifndef NHASH |
| 29 |
#define NHASH 2039 /* hash table size (prime!) */ |
| 30 |
#endif |
| 31 |
|
| 32 |
typedef struct s_head { |
| 33 |
struct s_head *next; /* next in hash list */ |
| 34 |
int nl; /* links count */ |
| 35 |
} S_HEAD; /* followed by the string itself */ |
| 36 |
|
| 37 |
static S_HEAD *stab[NHASH]; |
| 38 |
|
| 39 |
#define hash(s) (shash(s)%NHASH) |
| 40 |
|
| 41 |
#define string(sp) ((char *)((sp)+1)) |
| 42 |
|
| 43 |
#define salloc(str) (S_HEAD *)malloc(sizeof(S_HEAD)+1+strlen(str)) |
| 44 |
|
| 45 |
#define sfree(sp) free((void *)(sp)) |
| 46 |
|
| 47 |
|
| 48 |
char * |
| 49 |
savestr(const char *str) /* save a string */ |
| 50 |
{ |
| 51 |
int hval; |
| 52 |
S_HEAD *sp; |
| 53 |
|
| 54 |
if (str == NULL) |
| 55 |
return(NULL); |
| 56 |
if (!*str) |
| 57 |
return ""; |
| 58 |
hval = hash(str); |
| 59 |
for (sp = stab[hval]; sp != NULL; sp = sp->next) |
| 60 |
if (!strcmp(str, string(sp))) { |
| 61 |
sp->nl++; |
| 62 |
return(string(sp)); |
| 63 |
} |
| 64 |
if ((sp = salloc(str)) == NULL) { |
| 65 |
eputs("Out of memory in savestr\n"); |
| 66 |
quit(1); |
| 67 |
} |
| 68 |
strcpy(string(sp), str); |
| 69 |
sp->nl = 1; |
| 70 |
sp->next = stab[hval]; |
| 71 |
stab[hval] = sp; |
| 72 |
return(string(sp)); |
| 73 |
} |
| 74 |
|
| 75 |
|
| 76 |
void |
| 77 |
freestr(char *s) /* free a string */ |
| 78 |
{ |
| 79 |
int hval; |
| 80 |
S_HEAD *spl, *sp; |
| 81 |
|
| 82 |
if (s == NULL || !*s) |
| 83 |
return; |
| 84 |
hval = hash(s); |
| 85 |
for (spl = NULL, sp = stab[hval]; sp != NULL; spl = sp, sp = sp->next) |
| 86 |
if (s == string(sp)) { |
| 87 |
if (--sp->nl > 0) |
| 88 |
return; |
| 89 |
if (spl != NULL) |
| 90 |
spl->next = sp->next; |
| 91 |
else |
| 92 |
stab[hval] = sp->next; |
| 93 |
sfree(sp); |
| 94 |
return; |
| 95 |
} |
| 96 |
} |
| 97 |
|
| 98 |
|
| 99 |
int |
| 100 |
shash(const char *s) |
| 101 |
{ |
| 102 |
int h = 0; |
| 103 |
|
| 104 |
while (*s) |
| 105 |
h = (h<<1 & 0x7fff) ^ (*s++ & 0xff); |
| 106 |
return(h); |
| 107 |
} |