X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Fcmem.c;h=ef7a9ddb75e0fe1303aabcdf8084e586ddccdd4f;hp=3085ea6303a2d3c714c958477e2d2d4a393bb8c8;hb=9e6203ab5881638185e10b09b1f40428e95fb98c;hpb=3964fd135fcf5f6309639096437dd7f07e0c035f

diff --git a/src/cmem.c b/src/cmem.c
index 3085ea6..ef7a9dd 100644
--- a/src/cmem.c
+++ b/src/cmem.c
@@ -1,15 +1,6 @@
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <assert.h>
-#include <stdbool.h>
 #include "cmem.h"
 
-/**
- * mallocを実行し、0で初期化\n
- * メモリを確保できない場合はエラーを出力して終了
- */
-void *malloc_chk(size_t size, char *tag)
+void *malloc_chk(size_t size, const char *tag)
 {
     void *p;
 
@@ -20,10 +11,6 @@ void *malloc_chk(size_t size, char *tag)
     return memset(p, 0, size);
 }
 
-/**
- * callocを実行\n
- * メモリを確保できない場合はエラーを出力して終了
- */
 void *calloc_chk(size_t nmemb, size_t size, char *tag)
 {
     void *p;
@@ -35,9 +22,6 @@ void *calloc_chk(size_t nmemb, size_t size, char *tag)
     return p;
 }
 
-/**
- * malloc_chkを実行してメモリを確保し、コピーした文字列を返す
- */
 char *strdup_chk(const char *s, char *tag)
 {
     assert(s != NULL);
@@ -47,3 +31,14 @@ char *strdup_chk(const char *s, char *tag)
     strcpy(t, s);
     return t;
 }
+
+char *strndup_chk(const char *s, size_t len, char *tag)
+{
+    assert(s != NULL);
+    char *t;
+
+    t = malloc_chk(len + 1, tag);
+    strncpy(t, s, len);
+    t[len] = '\0';
+    return t;
+}