X-Git-Url: http://woldlab.caltech.edu/gitweb/?p=samtools.git;a=blobdiff_plain;f=bam_index.c;h=bc89b5b24b40736902d191c5e06774c8e8f228b6;hp=9610a2656031657b796abc68bf32c1b6f1564745;hb=refs%2Fheads%2Fcalnh;hpb=b990d9d3dd518a9cf7b7f4a165c584d8d8d1cfb0 diff --git a/bam_index.c b/bam_index.c index 9610a26..bc89b5b 100644 --- a/bam_index.c +++ b/bam_index.c @@ -259,10 +259,16 @@ void bam_index_save(const bam_index_t *idx, FILE *fp) { int32_t i, size; khint_t k; - fwrite("BAI\1", 1, 4, fp); + if (fwrite("BAI\1", 1, 4, fp) < 4) { + fprintf(stderr, "[bam_index_save] failed to write magic number.\n"); + abort(); + } if (bam_is_be) { uint32_t x = idx->n; - fwrite(bam_swap_endian_4p(&x), 4, 1, fp); + if (fwrite(bam_swap_endian_4p(&x), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_ref.\n"); + abort(); + } } else fwrite(&idx->n, 4, 1, fp); for (i = 0; i < idx->n; ++i) { khash_t(i) *index = idx->index[i]; @@ -271,49 +277,91 @@ void bam_index_save(const bam_index_t *idx, FILE *fp) size = kh_size(index); if (bam_is_be) { // big endian uint32_t x = size; - fwrite(bam_swap_endian_4p(&x), 4, 1, fp); - } else fwrite(&size, 4, 1, fp); + if (fwrite(bam_swap_endian_4p(&x), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_bin.\n"); + abort(); + } + } else if (fwrite(&size, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_bin.\n"); + abort(); + } for (k = kh_begin(index); k != kh_end(index); ++k) { if (kh_exist(index, k)) { bam_binlist_t *p = &kh_value(index, k); if (bam_is_be) { // big endian uint32_t x; - x = kh_key(index, k); fwrite(bam_swap_endian_4p(&x), 4, 1, fp); - x = p->n; fwrite(bam_swap_endian_4p(&x), 4, 1, fp); + x = kh_key(index, k); + if (fwrite(bam_swap_endian_4p(&x), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write bin.\n"); + abort(); + } + x = p->n; + if (fwrite(bam_swap_endian_4p(&x), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_chunk.\n"); + abort(); + } for (x = 0; (int)x < p->n; ++x) { bam_swap_endian_8p(&p->list[x].u); bam_swap_endian_8p(&p->list[x].v); } - fwrite(p->list, 16, p->n, fp); + if (fwrite(p->list, 16, p->n, fp) < p->n) { + fprintf(stderr, "[bam_index_save] failed to write chunk pair.\n"); + abort(); + } + // hmmm...sandbagging PPC or reverting? for (x = 0; (int)x < p->n; ++x) { bam_swap_endian_8p(&p->list[x].u); bam_swap_endian_8p(&p->list[x].v); } } else { - fwrite(&kh_key(index, k), 4, 1, fp); - fwrite(&p->n, 4, 1, fp); - fwrite(p->list, 16, p->n, fp); + if (fwrite(&kh_key(index, k), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write bin.\n"); + abort(); + } + if (fwrite(&p->n, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_chunk.\n"); + abort(); + } + if (fwrite(p->list, 16, p->n, fp) < p->n) { + fprintf(stderr, "[bam_index_save] failed to write chunk pair.\n"); + abort(); + } } } } // write linear index (index2) if (bam_is_be) { int x = index2->n; - fwrite(bam_swap_endian_4p(&x), 4, 1, fp); - } else fwrite(&index2->n, 4, 1, fp); + if (fwrite(bam_swap_endian_4p(&x), 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_intv.\n"); + abort(); + } + } else if (fwrite(&index2->n, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_intv.\n"); + abort(); + } if (bam_is_be) { // big endian int x; for (x = 0; (int)x < index2->n; ++x) bam_swap_endian_8p(&index2->offset[x]); - fwrite(index2->offset, 8, index2->n, fp); + if (fwrite(index2->offset, 8, index2->n, fp) < index2->n) { + fprintf(stderr, "[bam_index_save] failed to write ioffset.\n"); + abort(); + } for (x = 0; (int)x < index2->n; ++x) bam_swap_endian_8p(&index2->offset[x]); - } else fwrite(index2->offset, 8, index2->n, fp); + } else if (fwrite(index2->offset, 8, index2->n, fp) < index2->n) { + fprintf(stderr, "[bam_index_save] failed to write ioffset.\n"); + abort(); + } } { // write the number of reads coor-less records. uint64_t x = idx->n_no_coor; if (bam_is_be) bam_swap_endian_8p(&x); - fwrite(&x, 8, 1, fp); + if (fwrite(&x, 8, 1, fp) < 1) { + fprintf(stderr, "[bam_index_save] failed to write n_no_coor.\n"); + abort(); + } } fflush(fp); } @@ -327,14 +375,22 @@ static bam_index_t *bam_index_load_core(FILE *fp) fprintf(stderr, "[bam_index_load_core] fail to load index.\n"); return 0; } - fread(magic, 1, 4, fp); + if (fread(magic, 1, 4, fp) < 4) { + fprintf(stderr, "[bam_index_load] failed to read magic number.\n"); + fclose(fp); + return 0; + } if (strncmp(magic, "BAI\1", 4)) { fprintf(stderr, "[bam_index_load] wrong magic number.\n"); fclose(fp); return 0; } idx = (bam_index_t*)calloc(1, sizeof(bam_index_t)); - fread(&idx->n, 4, 1, fp); + if (fread(&idx->n, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_load] failed to read n_ref field.\n"); + fclose(fp); + return 0; + } if (bam_is_be) bam_swap_endian_4p(&idx->n); idx->index = (khash_t(i)**)calloc(idx->n, sizeof(void*)); idx->index2 = (bam_lidx_t*)calloc(idx->n, sizeof(bam_lidx_t)); @@ -347,18 +403,34 @@ static bam_index_t *bam_index_load_core(FILE *fp) bam_binlist_t *p; index = idx->index[i] = kh_init(i); // load binning index - fread(&size, 4, 1, fp); + if (fread(&size, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_load] read error while loading n_bin.\n"); + fclose(fp); + return 0; + } if (bam_is_be) bam_swap_endian_4p(&size); for (j = 0; j < (int)size; ++j) { - fread(&key, 4, 1, fp); + if (fread(&key, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_load] read error while loading bin.\n"); + fclose(fp); + return 0; + } if (bam_is_be) bam_swap_endian_4p(&key); k = kh_put(i, index, key, &ret); p = &kh_value(index, k); - fread(&p->n, 4, 1, fp); + if (fread(&p->n, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_load] read error while loading n_chunk.\n"); + fclose(fp); + return 0; + } if (bam_is_be) bam_swap_endian_4p(&p->n); p->m = p->n; p->list = (pair64_t*)malloc(p->m * 16); - fread(p->list, 16, p->n, fp); + if (fread(p->list, 16, p->n, fp) < p->n) { + fprintf(stderr, "[bam_index_load] read error while loading chunk pairs.\n"); + fclose(fp); + return 0; + } if (bam_is_be) { int x; for (x = 0; x < p->n; ++x) { @@ -368,11 +440,20 @@ static bam_index_t *bam_index_load_core(FILE *fp) } } // load linear index - fread(&index2->n, 4, 1, fp); + if (fread(&index2->n, 4, 1, fp) < 1) { + fprintf(stderr, "[bam_index_load] read error while loading n_intv.\n"); + fclose(fp); + return 0; + } if (bam_is_be) bam_swap_endian_4p(&index2->n); index2->m = index2->n; index2->offset = (uint64_t*)calloc(index2->m, 8); - fread(index2->offset, index2->n, 8, fp); + ret = fread(index2->offset, index2->n, 8, fp); + if ((ret > 0) && (ret < 8)) { + fprintf(stderr, "[bam_index_load] read error while loading ioffset.\n"); + fclose(fp); + return 0; + } if (bam_is_be) for (j = 0; j < index2->n; ++j) bam_swap_endian_8p(&index2->offset[j]); } @@ -438,7 +519,10 @@ static void download_from_remote(const char *url) } buf = (uint8_t*)calloc(buf_size, 1); while ((l = knet_read(fp_remote, buf, buf_size)) != 0) - fwrite(buf, 1, l, fp); + if (fwrite(buf, 1, l, fp) < l) { + fprintf(stderr, "[download_from_remote] fail to write to destination file.\n"); + break; + } free(buf); fclose(fp); knet_close(fp_remote);