X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Fcmem.c;h=e921bb08f4678bf2c783e0e0621a303b34ea2cd5;hp=b83c79c3dc546166d7208b95da0bf3bcb0525df6;hb=58b56a6373dbd2b2ce8d30f7661ff15f584fb03f;hpb=555c5e8b851becc08ba661a9cb19f617d5a00c12

diff --git a/src/cmem.c b/src/cmem.c
index b83c79c..e921bb0 100644
--- a/src/cmem.c
+++ b/src/cmem.c
@@ -1,47 +1,58 @@
 #include "cmem.h"
 
-/* mallocを実行し、0で初期化 */
-/* メモリを確保できない場合はエラーを出力して終了 */
-void *malloc_chk(size_t size, char *tag)
+void *malloc_chk(size_t size, const char *tag)
 {
-    void *p;
+    void *p = NULL;
 
     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 *calloc_chk(size_t nmemb, size_t size, const char *tag)
 {
-    void *p;
+    void *p = NULL;
 
     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)
+char *strdup_chk(const char *s, const char *tag)
 {
     assert(s != NULL);
-    char *t;
+    char *t = NULL;
 
     t = malloc_chk(strlen(s) + 1, tag);
     strcpy(t, s);
     return t;
 }
 
-/* メモリがNULLの場合は解放 */
-void free_chk(void *ptr, char *tag)
+char *strndup_chk(const char *s, size_t len, const char *tag)
 {
-    if(ptr != NULL) {
-        free(ptr);
+    assert(s != NULL);
+    char *t = NULL;
+
+    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;
+}
+
+char *strip_end(char *s)
+{
+    int i;
+    i = strlen(s) - 1;
+    while(i > 0 && (s[i] == '\n' || s[i] == ' ' || s[i] == '\t')) {
+        s[i--] = '\0';
     }
+    return s;
 }