]> sigrok.org Git - libsigrok.git/blob - output/text/bits.c
output/text: Fix memory leak of internal state buffers.
[libsigrok.git] / output / text / bits.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2010-2012 Bert Vermeulen <bert@biot.com>
5  * Copyright (C) 2011 HÃ¥vard Espeland <gus@ping.uio.no>
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include <stdlib.h>
22 #include <string.h>
23 #include <glib.h>
24 #include "libsigrok.h"
25 #include "libsigrok-internal.h"
26 #include "text.h"
27
28 /* Message logging helpers with subsystem-specific prefix string. */
29 #define LOG_PREFIX "output/bits: "
30 #define sr_log(l, s, args...) sr_log(l, LOG_PREFIX s, ## args)
31 #define sr_spew(s, args...) sr_spew(LOG_PREFIX s, ## args)
32 #define sr_dbg(s, args...) sr_dbg(LOG_PREFIX s, ## args)
33 #define sr_info(s, args...) sr_info(LOG_PREFIX s, ## args)
34 #define sr_warn(s, args...) sr_warn(LOG_PREFIX s, ## args)
35 #define sr_err(s, args...) sr_err(LOG_PREFIX s, ## args)
36
37 SR_PRIV int init_bits(struct sr_output *o)
38 {
39         return init(o, DEFAULT_BPL_BITS, MODE_BITS);
40 }
41
42 SR_PRIV int data_bits(struct sr_output *o, const uint8_t *data_in,
43                       uint64_t length_in, uint8_t **data_out,
44                       uint64_t *length_out)
45 {
46         struct context *ctx;
47         unsigned int outsize, offset, p;
48         int max_linelen;
49         const uint8_t *sample;
50         uint8_t *outbuf, c;
51
52         ctx = o->internal;
53         max_linelen = SR_MAX_PROBENAME_LEN + 3 + ctx->samples_per_line
54                         + ctx->samples_per_line / 8;
55         /*
56          * Calculate space needed for probes. Set aside 512 bytes for
57          * extra output, e.g. trigger.
58          */
59         outsize = 512 + (1 + (length_in / ctx->unitsize) / ctx->samples_per_line)
60             * (ctx->num_enabled_probes * max_linelen);
61
62         if (!(outbuf = g_try_malloc0(outsize + 1))) {
63                 sr_err("%s: outbuf malloc failed", __func__);
64                 return SR_ERR_MALLOC;
65         }
66
67         outbuf[0] = '\0';
68         if (ctx->header) {
69                 /* The header is still here, this must be the first packet. */
70                 strncpy((char *)outbuf, ctx->header, outsize);
71                 g_free(ctx->header);
72                 ctx->header = NULL;
73         }
74
75         if (length_in >= ctx->unitsize) {
76                 for (offset = 0; offset <= length_in - ctx->unitsize;
77                      offset += ctx->unitsize) {
78                         sample = data_in + offset;
79                         for (p = 0; p < ctx->num_enabled_probes; p++) {
80                                 c = (sample[p / 8] & ((uint8_t) 1 << (p % 8))) ? '1' : '0';
81                                 ctx->linebuf[p * ctx->linebuf_len +
82                                              ctx->line_offset] = c;
83                         }
84                         ctx->line_offset++;
85                         ctx->spl_cnt++;
86
87                         /* Add a space every 8th bit. */
88                         if ((ctx->spl_cnt & 7) == 0) {
89                                 for (p = 0; p < ctx->num_enabled_probes; p++)
90                                         ctx->linebuf[p * ctx->linebuf_len +
91                                                      ctx->line_offset] = ' ';
92                                 ctx->line_offset++;
93                         }
94
95                         /* End of line. */
96                         if (ctx->spl_cnt >= ctx->samples_per_line) {
97                                 flush_linebufs(ctx, outbuf);
98                                 ctx->line_offset = ctx->spl_cnt = 0;
99                                 ctx->mark_trigger = -1;
100                         }
101                 }
102         } else {
103                 sr_info("Short buffer (length_in=%" PRIu64 ").", length_in);
104         }
105
106         *data_out = outbuf;
107         *length_out = strlen((const char *)outbuf);
108
109         return SR_OK;
110 }
111
112 SR_PRIV struct sr_output_format output_text_bits = {
113         .id = "bits",
114         .description = "Bits",
115         .df_type = SR_DF_LOGIC,
116         .init = init_bits,
117         .data = data_bits,
118         .event = event,
119         .cleanup = text_cleanup,
120 };