wavcapture.c (4901B)
1 #include "qemu/osdep.h" 2 #include "qemu/qemu-print.h" 3 #include "qapi/error.h" 4 #include "qemu/error-report.h" 5 #include "audio.h" 6 7 typedef struct { 8 FILE *f; 9 int bytes; 10 char *path; 11 int freq; 12 int bits; 13 int nchannels; 14 CaptureVoiceOut *cap; 15 } WAVState; 16 17 /* VICE code: Store number as little endian. */ 18 static void le_store (uint8_t *buf, uint32_t val, int len) 19 { 20 int i; 21 for (i = 0; i < len; i++) { 22 buf[i] = (uint8_t) (val & 0xff); 23 val >>= 8; 24 } 25 } 26 27 static void wav_notify (void *opaque, audcnotification_e cmd) 28 { 29 (void) opaque; 30 (void) cmd; 31 } 32 33 static void wav_destroy (void *opaque) 34 { 35 WAVState *wav = opaque; 36 uint8_t rlen[4]; 37 uint8_t dlen[4]; 38 uint32_t datalen = wav->bytes; 39 uint32_t rifflen = datalen + 36; 40 41 if (wav->f) { 42 le_store (rlen, rifflen, 4); 43 le_store (dlen, datalen, 4); 44 45 if (fseek (wav->f, 4, SEEK_SET)) { 46 error_report("wav_destroy: rlen fseek failed: %s", 47 strerror(errno)); 48 goto doclose; 49 } 50 if (fwrite (rlen, 4, 1, wav->f) != 1) { 51 error_report("wav_destroy: rlen fwrite failed: %s", 52 strerror(errno)); 53 goto doclose; 54 } 55 if (fseek (wav->f, 32, SEEK_CUR)) { 56 error_report("wav_destroy: dlen fseek failed: %s", 57 strerror(errno)); 58 goto doclose; 59 } 60 if (fwrite (dlen, 1, 4, wav->f) != 4) { 61 error_report("wav_destroy: dlen fwrite failed: %s", 62 strerror(errno)); 63 goto doclose; 64 } 65 doclose: 66 if (fclose (wav->f)) { 67 error_report("wav_destroy: fclose failed: %s", strerror(errno)); 68 } 69 } 70 71 g_free (wav->path); 72 } 73 74 static void wav_capture(void *opaque, const void *buf, int size) 75 { 76 WAVState *wav = opaque; 77 78 if (fwrite (buf, size, 1, wav->f) != 1) { 79 error_report("wav_capture: fwrite error: %s", strerror(errno)); 80 } 81 wav->bytes += size; 82 } 83 84 static void wav_capture_destroy (void *opaque) 85 { 86 WAVState *wav = opaque; 87 88 AUD_del_capture (wav->cap, wav); 89 g_free (wav); 90 } 91 92 static void wav_capture_info (void *opaque) 93 { 94 WAVState *wav = opaque; 95 char *path = wav->path; 96 97 qemu_printf("Capturing audio(%d,%d,%d) to %s: %d bytes\n", 98 wav->freq, wav->bits, wav->nchannels, 99 path ? path : "<not available>", wav->bytes); 100 } 101 102 static struct capture_ops wav_capture_ops = { 103 .destroy = wav_capture_destroy, 104 .info = wav_capture_info 105 }; 106 107 int wav_start_capture(AudioState *state, CaptureState *s, const char *path, 108 int freq, int bits, int nchannels) 109 { 110 WAVState *wav; 111 uint8_t hdr[] = { 112 0x52, 0x49, 0x46, 0x46, 0x00, 0x00, 0x00, 0x00, 0x57, 0x41, 0x56, 113 0x45, 0x66, 0x6d, 0x74, 0x20, 0x10, 0x00, 0x00, 0x00, 0x01, 0x00, 114 0x02, 0x00, 0x44, 0xac, 0x00, 0x00, 0x10, 0xb1, 0x02, 0x00, 0x04, 115 0x00, 0x10, 0x00, 0x64, 0x61, 0x74, 0x61, 0x00, 0x00, 0x00, 0x00 116 }; 117 struct audsettings as; 118 struct audio_capture_ops ops; 119 int stereo, bits16, shift; 120 CaptureVoiceOut *cap; 121 122 if (bits != 8 && bits != 16) { 123 error_report("incorrect bit count %d, must be 8 or 16", bits); 124 return -1; 125 } 126 127 if (nchannels != 1 && nchannels != 2) { 128 error_report("incorrect channel count %d, must be 1 or 2", 129 nchannels); 130 return -1; 131 } 132 133 stereo = nchannels == 2; 134 bits16 = bits == 16; 135 136 as.freq = freq; 137 as.nchannels = 1 << stereo; 138 as.fmt = bits16 ? AUDIO_FORMAT_S16 : AUDIO_FORMAT_U8; 139 as.endianness = 0; 140 141 ops.notify = wav_notify; 142 ops.capture = wav_capture; 143 ops.destroy = wav_destroy; 144 145 wav = g_malloc0 (sizeof (*wav)); 146 147 shift = bits16 + stereo; 148 hdr[34] = bits16 ? 0x10 : 0x08; 149 150 le_store (hdr + 22, as.nchannels, 2); 151 le_store (hdr + 24, freq, 4); 152 le_store (hdr + 28, freq << shift, 4); 153 le_store (hdr + 32, 1 << shift, 2); 154 155 wav->f = fopen (path, "wb"); 156 if (!wav->f) { 157 error_report("Failed to open wave file `%s': %s", 158 path, strerror(errno)); 159 g_free (wav); 160 return -1; 161 } 162 163 wav->path = g_strdup (path); 164 wav->bits = bits; 165 wav->nchannels = nchannels; 166 wav->freq = freq; 167 168 if (fwrite (hdr, sizeof (hdr), 1, wav->f) != 1) { 169 error_report("Failed to write header: %s", strerror(errno)); 170 goto error_free; 171 } 172 173 cap = AUD_add_capture(state, &as, &ops, wav); 174 if (!cap) { 175 error_report("Failed to add audio capture"); 176 goto error_free; 177 } 178 179 wav->cap = cap; 180 s->opaque = wav; 181 s->ops = wav_capture_ops; 182 return 0; 183 184 error_free: 185 g_free (wav->path); 186 if (fclose (wav->f)) { 187 error_report("Failed to close wave file: %s", strerror(errno)); 188 } 189 g_free (wav); 190 return -1; 191 }