| /* Process Queue: File handling tasks */ |
| |
| /* |
| * This file is part of gapk (GSM Audio Pocket Knife). |
| * |
| * gapk is free software: you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation, either version 3 of the License, or |
| * (at your option) any later version. |
| * |
| * gapk is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with gapk. If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include <errno.h> |
| #include <stdint.h> |
| #include <stdbool.h> |
| #include <talloc.h> |
| |
| #include <osmocom/gapk/logging.h> |
| #include <osmocom/gapk/codecs.h> |
| #include <osmocom/gapk/formats.h> |
| #include <osmocom/gapk/procqueue.h> |
| |
| |
| struct pq_state_file { |
| FILE *fh; |
| unsigned int blk_len; |
| bool loop_input; |
| }; |
| |
| |
| static int |
| pq_cb_file_input(void *_state, uint8_t *out, const uint8_t *in, unsigned int in_len) |
| { |
| struct pq_state_file *state = _state; |
| int rv; |
| rv = fread(out, state->blk_len, 1, state->fh); |
| if (rv == 0 && state->loop_input && feof(state->fh)) { |
| /* FIXME: this only works for files without a header |
| * (e.g. .gsm files) but not for those with header (like |
| * .amr files). gs->opts.fmt_in->header is not |
| * accessible here :/ */ |
| fseek(state->fh, 0, SEEK_SET); |
| rv = fread(out, state->blk_len, 1, state->fh); |
| } |
| if (rv <= 0) |
| return -1; |
| return rv * state->blk_len; |
| } |
| |
| static int |
| pq_cb_file_output(void *_state, uint8_t *out, const uint8_t *in, unsigned int in_len) |
| { |
| struct pq_state_file *state = _state; |
| int rv; |
| rv = fwrite(in, in_len, 1, state->fh); |
| return rv == 1 ? 0 : -1; |
| } |
| |
| static void |
| pq_cb_file_exit(void *_state) |
| { |
| talloc_free(_state); |
| } |
| |
| static int |
| pq_queue_file_op(struct osmo_gapk_pq *pq, FILE *fh, unsigned int blk_len, int in_out_n, bool loop_input) |
| { |
| struct osmo_gapk_pq_item *item; |
| struct pq_state_file *state; |
| |
| state = talloc_zero(pq, struct pq_state_file); |
| if (!state) |
| return -ENOMEM; |
| |
| state->fh = fh; |
| state->blk_len = blk_len; |
| state->loop_input = loop_input; |
| |
| item = osmo_gapk_pq_add_item(pq); |
| if (!item) { |
| talloc_free(state); |
| return -ENOMEM; |
| } |
| |
| item->type = in_out_n ? |
| OSMO_GAPK_ITEM_TYPE_SOURCE : OSMO_GAPK_ITEM_TYPE_SINK; |
| item->cat_name = in_out_n ? |
| OSMO_GAPK_CAT_NAME_SOURCE : OSMO_GAPK_CAT_NAME_SINK; |
| item->sub_name = "file"; |
| |
| item->len_in = in_out_n ? 0 : blk_len; |
| item->len_out = in_out_n ? blk_len : 0; |
| item->state = state; |
| item->proc = in_out_n ? pq_cb_file_input : pq_cb_file_output; |
| item->wait = NULL; |
| item->exit = pq_cb_file_exit; |
| |
| /* Change state's talloc context from pq to item */ |
| talloc_steal(item, state); |
| |
| return 0; |
| } |
| |
| |
| /*! Add file input to given processing queue |
| * This usually only makes sense as first item in the queue |
| * \param pq Processing Queue to add the input file to |
| * \param[in] src caller-fopen()ed input file |
| * \param[in] blk_len block length to be read from file |
| * \returns 0 on success; negative on error */ |
| int |
| osmo_gapk_pq_queue_file_input(struct osmo_gapk_pq *pq, FILE *src, unsigned int blk_len, bool loop) |
| { |
| LOGPGAPK(LOGL_DEBUG, "PQ '%s': Adding file input (blk_len=%u)\n", |
| pq->name, blk_len); |
| return pq_queue_file_op(pq, src, blk_len, 1, loop); |
| } |
| |
| /*! Add file output to given processing queue |
| * This usually only makes sense as first item in the queue |
| * \param pq Processing Queue to add the output file to |
| * \param[in] dst caller-fopen()ed output file |
| * \param[in] blk_len block length to be written to file |
| * \returns 0 on success; negative on error */ |
| int |
| osmo_gapk_pq_queue_file_output(struct osmo_gapk_pq *pq, FILE *dst, unsigned int blk_len) |
| { |
| LOGPGAPK(LOGL_DEBUG, "PQ '%s': Adding file output (blk_len=%u)\n", |
| pq->name, blk_len); |
| return pq_queue_file_op(pq, dst, blk_len, 0, false); |
| } |