ref: 35b98e0f4f5d021690beb4f274e637e327063e62
dir: /src/stio.c/
#include "st.h" #include "st_i.h" #include "stconfig.h" #ifdef HAVE_STRING_H #include <string.h> #endif #ifdef HAVE_ERRNO_H #include <errno.h> #endif #include <stdlib.h> #include <sys/types.h> /* for fstat() */ #include <sys/stat.h> /* for fstat() */ #ifdef _MSC_VER /* * __STDC__ is defined, so these symbols aren't created. */ #define S_IFMT _S_IFMT #define S_IFREG _S_IFREG #define fstat _fstat #endif /* Based from zlib's minigzip: */ #if defined(WIN32) || defined(__NT__) # include <fcntl.h> # include <io.h> # define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY) #else # define SET_BINARY_MODE(file) #endif static int is_seekable(ft_t ft) { struct stat st; fstat(fileno(ft->fp), &st); return ((st.st_mode & S_IFMT) == S_IFREG); } /* check that all settings have been given */ static int st_checkformat(ft_t ft) { ft->st_errno = ST_SUCCESS; if (ft->info.rate == 0) { st_fail_errno(ft,ST_EFMT,"sampling rate was not specified"); return ST_EOF; } if (ft->info.size == -1) { st_fail_errno(ft,ST_EFMT,"data size was not specified"); return ST_EOF; } if (ft->info.encoding == -1) { st_fail_errno(ft,ST_EFMT,"data encoding was not specified"); return ST_EOF; } if ((ft->info.size <= 0) || (ft->info.size > ST_INFO_SIZE_MAX)) { st_fail_errno(ft,ST_EFMT,"data size %i is invalid"); return ST_EOF; } if (ft->info.encoding <= 0 || ft->info.encoding > ST_ENCODING_MAX) { st_fail_errno(ft,ST_EFMT,"data encoding %i is invalid"); return ST_EOF; } return ST_SUCCESS; } ft_t st_open_read(const char *path, const st_signalinfo_t *info, const char *filetype) { ft_t ft; ft = (ft_t)calloc(sizeof(struct st_soundstream), 1); if (!ft ) return NULL; ft->filename = strdup(path); /* Let auto effect do the work if user is not overriding. */ if (!filetype) ft->filetype = strdup("auto"); else ft->filetype = strdup(filetype); if (!ft->filename || !ft->filetype) goto input_error; if (st_gettype(ft) != ST_SUCCESS) { st_warn("Unknown input file format for '%s': %s", ft->filename, ft->st_errstr); goto input_error; } ft->info.size = -1; ft->info.encoding = -1; ft->info.channels = -1; if (info) ft->info = *info; /* FIXME: Remove ft->swap from code */ ft->swap = ft->info.swap; ft->mode = 'r'; if (!(ft->h->flags & ST_FILE_NOSTDIO)) { /* Open file handler based on input name. Used stdin file handler * if the filename is "-" */ if (!strcmp(ft->filename, "-")) { SET_BINARY_MODE(stdin); ft->fp = stdin; } else if ((ft->fp = fopen(ft->filename, "rb")) == NULL) { st_warn("Can't open input file '%s': %s", ft->filename, strerror(errno)); goto input_error; } /* See if this file is seekable or not */ ft->seekable = is_seekable(ft); } /* Read and write starters can change their formats. */ if ((*ft->h->startread)(ft) != ST_SUCCESS) { st_warn("Failed reading %s: %s", ft->filename, ft->st_errstr); goto input_error; } /* Go a head and assume 1 channel audio if nothing is detected. * This is because libst usually doesn't set this for mono file * formats (for historical reasons). */ if (ft->info.channels == -1) ft->info.channels = 1; if (st_checkformat(ft) ) { st_fail("bad input format for file %s: %s", ft->filename, ft->st_errstr); goto input_error; } return ft; input_error: if (ft->filename) free(ft->filename); if (ft->filetype) free(ft->filetype); free(ft); return NULL; } #if defined(DOS) || defined(WIN32) #define LASTCHAR '\\' #else #define LASTCHAR '/' #endif ft_t st_open_write_instr(const char *path, const st_signalinfo_t *info, const char *filetype, const char *comment, const st_instrinfo_t *instr, const st_loopinfo_t *loops) { ft_t ft; int i; ft = (ft_t)calloc(sizeof(struct st_soundstream), 1); if (!ft ) return NULL; ft->filename = strdup(path); /* Let auto effect do the work if user is not overriding. */ if (!filetype) { char *chop; int len; len = strlen(ft->filename); /* Use filename extension to determine audio type. * Search for the last '.' appearing in the filename, same * as for input files. */ chop = ft->filename + len; while (chop > ft->filename && *chop != LASTCHAR && *chop != '.') chop--; if (*chop == '.') { chop++; ft->filetype = strdup(chop); } } else ft->filetype = strdup(filetype); if (!ft->filename) { st_warn("Unknown output file format for 'NULL'."); goto output_error; } if (!ft->filetype || st_gettype(ft) != ST_SUCCESS) { st_warn("Unknown output file format for '%s': %s", ft->filename, ft->st_errstr); goto output_error; } ft->info.size = -1; ft->info.encoding = -1; ft->info.channels = -1; if (info) ft->info = *info; ft->mode = 'w'; if (!(ft->h->flags & ST_FILE_NOSTDIO)) { /* Open file handler based on output name. Used stdout file handler * if the filename is "-" */ if (!strcmp(ft->filename, "-")) { ft->fp = stdout; } else if ((ft->fp = fopen(ft->filename, "wb")) == NULL) { st_warn("Can't open output file '%s': %s", ft->filename, strerror(errno)); goto output_error; } /* stdout tends to be line-buffered. Override this */ /* to be Full Buffering. */ /* FIXME: Use buffer size from ft structure */ if (setvbuf (ft->fp, NULL, _IOFBF, sizeof(char)*ST_BUFSIZ)) { st_warn("Can't set write buffer"); goto output_error; } /* See if this file is seekable or not */ ft->seekable = is_seekable(ft); } if (ft->comment == NULL && comment != NULL) ft->comment = strdup(comment); else ft->comment = strdup("Processed by SoX"); if (loops) { for (i = 0; i < ST_MAX_NLOOPS; i++) { ft->loops[i] = loops[i]; } } /* leave SMPTE # alone since it's absolute */ if (instr) ft->instr = *instr; /* FIXME: Remove ft->swap from code */ ft->swap = ft->info.swap; /* Read and write starters can change their formats. */ if ((*ft->h->startwrite)(ft) != ST_SUCCESS) { st_warn("Failed writing %s: %s", ft->filename, ft->st_errstr); goto output_error; } if (st_checkformat(ft) ) { st_fail("bad output format for file %s: %s", ft->filename, ft->st_errstr); goto output_error; } return ft; output_error: if (ft->filename) free(ft->filename); if (ft->filetype) free(ft->filetype); free(ft); return NULL; } ft_t st_open_write(const char *path, const st_signalinfo_t *info, const char *filetype, const char *comment) { return st_open_write_instr(path, info, filetype, comment, NULL, NULL); } st_ssize_t st_read(ft_t ft, st_sample_t *buf, st_ssize_t len) { return (*ft->h->read)(ft, buf, len); } st_ssize_t st_write(ft_t ft, st_sample_t *buf, st_ssize_t len) { return (*ft->h->write)(ft, buf, len); } int st_close(ft_t ft) { int rc; if (ft->mode == 'r') rc = (*ft->h->stopread)(ft); else rc = (*ft->h->stopwrite)(ft); if (!(ft->h->flags & ST_FILE_NOSTDIO)) { fclose(ft->fp); } if (ft->filename) free(ft->filename); if (ft->filetype) free(ft->filetype); /* Currently, since startread() mallocs comments, stopread * is expected to also free it. */ if (ft->mode == 'w' && ft->comment) free(ft->comment); return rc; } int st_seek(ft_t ft, st_size_t offset, int whence) { /* One day, ST_SEEK_CUR and ST_SEEK_END should be impelemented */ if (whence != ST_SEEK_SET) return ST_EOF; /* FIXME: Should return this */ /* return ST_EINVAL; */ /* If file is a seekable file and this handler supports seeking, * the invoke handlers function. */ if (ft->seekable && (ft->h->flags & ST_FILE_SEEK)) return (*ft->h->seek)(ft, offset); else return ST_EOF; /* FIXME: Should return this */ /* return ST_EBADF; */ }