1 files changed, 128 insertions, 0 deletions
diff --git a/libaudio/arec.c b/libaudio/arec.c
new file mode 100644
index 0000000..b1e9eda
--- /dev/null
+++ b/libaudio/arec.c
@@ -0,0 +1,128 @@
+/*
+** Copyright 2010, The Android Open-Source Project
+**
+** Licensed under the Apache License, Version 2.0 (the "License");
+** you may not use this file except in compliance with the License.
+** You may obtain a copy of the License at
+**
+**     http://www.apache.org/licenses/LICENSE-2.0
+**
+** Unless required by applicable law or agreed to in writing, software
+** distributed under the License is distributed on an "AS IS" BASIS,
+** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+** See the License for the specific language governing permissions and
+** limitations under the License.
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <string.h>
+
+#include "alsa_audio.h"
+
+#define ID_RIFF 0x46464952
+#define ID_WAVE 0x45564157
+#define ID_FMT  0x20746d66
+#define ID_DATA 0x61746164
+
+#define FORMAT_PCM 1
+
+struct wav_header {
+    uint32_t riff_id;
+    uint32_t riff_sz;
+    uint32_t riff_fmt;
+    uint32_t fmt_id;
+    uint32_t fmt_sz;
+    uint16_t audio_format;
+    uint16_t num_channels;
+    uint32_t sample_rate;
+    uint32_t byte_rate;       /* sample_rate * num_channels * bps / 8 */
+    uint16_t block_align;     /* num_channels * bps / 8 */
+    uint16_t bits_per_sample;
+    uint32_t data_id;
+    uint32_t data_sz;
+};
+
+int record_file(unsigned rate, unsigned channels, int fd, unsigned count)
+{
+    struct pcm *pcm;
+    unsigned avail, xfer, bufsize;
+    char *data, *next;
+    int r;
+
+    pcm = pcm_open(PCM_IN|PCM_MONO);
+    if (!pcm_ready(pcm)) {
+        pcm_close(pcm);
+        goto fail;
+    }
+
+    bufsize = pcm_buffer_size(pcm);
+    
+    data = malloc(bufsize);
+    if (!data) {
+        fprintf(stderr,"could not allocate %d bytes\n", count);
+        return -1;
+    }
+
+    while (!pcm_read(pcm, data, bufsize)) {
+        if (write(fd, data, bufsize) != bufsize) {
+            fprintf(stderr,"could not write %d bytes\n", bufsize);
+            return -1;
+        }
+    }
+
+    close(fd);
+    pcm_close(pcm);
+    return 0;
+    
+fail:
+    fprintf(stderr,"pcm error: %s\n", pcm_error(pcm));
+    return -1;
+}
+
+int rec_wav(const char *fn)
+{
+    struct wav_header hdr;
+    unsigned rate, channels;
+    int fd;
+    fd = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0664);
+    if (fd < 0) {
+        fprintf(stderr, "arec: cannot open '%s'\n", fn);
+        return -1;
+    }
+
+    hdr.riff_id = ID_RIFF;
+    hdr.riff_fmt = ID_WAVE;
+    hdr.fmt_id = ID_FMT;
+    hdr.audio_format = FORMAT_PCM;
+    hdr.fmt_sz = 16;
+    hdr.bits_per_sample = 16;
+    hdr.num_channels = 1;
+    hdr.data_sz = 0;
+    hdr.sample_rate = 44100;
+
+    if (write(fd, &hdr, sizeof(hdr)) != sizeof(hdr)) {
+        fprintf(stderr, "arec: cannot write header\n");
+        return -1;
+    }
+    fprintf(stderr,"arec: %d ch, %d hz, %d bit, %s\n",
+            hdr.num_channels, hdr.sample_rate, hdr.bits_per_sample,
+            hdr.audio_format == FORMAT_PCM ? "PCM" : "unknown");
+    
+
+    return record_file(hdr.sample_rate, hdr.num_channels, fd, hdr.data_sz);
+}
+
+int main(int argc, char **argv)
+{
+    if (argc != 2) {
+        fprintf(stderr,"usage: arec <file>\n");
+        return -1;
+    }
+
+    return rec_wav(argv[1]);
+}
+