X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fcmem.c;h=17535d874c013e25c1a528d3bb0760e3b42a176f;hb=e8b475a846a6c819b5f7b4afbca67aa072294349;hp=2274c2ee21517c57f59cdac60e1dfa1d631da84a;hpb=7f7bde5dd1305cfc598fe09030c69aad0dc01368;p=YACASL2.git

diff --git a/src/cmem.c b/src/cmem.c
index 2274c2e..17535d8 100644
--- a/src/cmem.c
+++ b/src/cmem.c
@@ -1,43 +1,27 @@
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <assert.h>
-#include <stdbool.h>
 #include "cmem.h"
 
-/**
- * mallocを実行し、0で初期化
- * メモリを確保できない場合はエラーを出力して終了
- */
-void *malloc_chk(size_t size, char *tag)
+void *malloc_chk(size_t size, const char *tag)
 {
     void *p;
 
     if((p = malloc(size)) == NULL) {
         fprintf(stderr, "%s: cannot allocate memory\n", tag);
-        exit(-1);
+        exit(1);
     }
     return memset(p, 0, size);
 }
 
-/**
- * callocを実行
- * メモリを確保できない場合はエラーを出力して終了
- */
 void *calloc_chk(size_t nmemb, size_t size, char *tag)
 {
     void *p;
 
     if((p = calloc(nmemb, size)) == NULL) {
         fprintf(stderr, "%s: cannot allocate memory\n", tag);
-        exit(-1);
+        exit(1);
     }
     return p;
 }
 
-/**
- * malloc_chkを実行してメモリを確保し、コピーした文字列を返す
- */
 char *strdup_chk(const char *s, char *tag)
 {
     assert(s != NULL);
@@ -48,11 +32,17 @@ char *strdup_chk(const char *s, char *tag)
     return t;
 }
 
-/**
- * メモリを解放
- */
-void free_chk(void *ptr, char *tag)
+char *strndup_chk(const char *s, size_t len, char *tag)
 {
-    free(ptr);
-    ptr = NULL;
+    assert(s != NULL);
+    char *t;
+
+    if(len < strlen(s)) {
+        t = malloc_chk(len + 1, tag);
+        strncpy(t, s, len);
+        t[len] = '\0';
+    } else {
+        t = strdup_chk(s, tag);
+    }
+    return t;
 }