-fix build system issues
[oweals/gnunet.git] / src / conversation / gnunet-helper-audio-record.c
index a3e760794f122ba93b87e0e589ecb84643e903e8..89031221343b2a729575259287e89f85645284ce 100644 (file)
      Boston, MA 02111-1307, USA.
 */
 /**
- * @file conversation/gnunet-helper-audio-playback.c
- * @brief constants for network protocols
+ * @file conversation/gnunet-helper-audio-record.c
+ * @brief program to record audio data from the microphone
  * @author Siomon Dieterle
  * @author Andreas Fuchs
+ * @author Christian Grothoff
  */
 #include "platform.h"
 #include "gnunet_util_lib.h"
 #include <opus/opus.h>
 #include <opus/opus_types.h>
 
+#define SAMPLING_RATE 48000
+
 
 /**
  * Specification for recording. May change in the future to spec negotiation.
  */
 static pa_sample_spec sample_spec = {
   .format = PA_SAMPLE_FLOAT32LE,
-  .rate = 48000,
+  .rate = SAMPLING_RATE,
   .channels = 1
 };
 
@@ -89,7 +92,7 @@ static unsigned char *opus_data;
 static float *pcm_buffer;
 
 /**
- * Length of the pcm data needed for one OPUS frame 
+ * Length of the pcm data needed for one OPUS frame
  */
 static int pcm_length;
 
@@ -106,7 +109,7 @@ static int max_payload_bytes = 1500;
 /**
  * Audio buffer
  */
-static void *transmit_buffer;
+static char *transmit_buffer;
 
 /**
  * Length of audio buffer
@@ -141,15 +144,21 @@ quit (int ret)
 static void
 packetizer ()
 {
+  static unsigned long long toff;
+  char *nbuf;
+  size_t new_size;
+  const char *ptr;
+  size_t off;
+  ssize_t ret;
+  int len; // FIXME: int?
+  size_t msg_size;
+
   while (transmit_buffer_length >= transmit_buffer_index + pcm_length)
   {
-    ssize_t ret;
-    int len; // FIXME: int?
-    size_t msg_size;
-
     memcpy (pcm_buffer,
-           (float *) transmit_buffer +
-           (transmit_buffer_index / sizeof (float)), pcm_length);
+           &transmit_buffer[transmit_buffer_index],
+           pcm_length);
+    transmit_buffer_index += pcm_length;
     len =
       opus_encode_float (enc, pcm_buffer, frame_size, opus_data,
                         max_payload_bytes);
@@ -162,25 +171,43 @@ packetizer ()
     audio_message->header.size = htons ((uint16_t) msg_size);
     memcpy (&audio_message[1], opus_data, len);
 
-    // FIXME: handle partial writes better...
-    if ((ret = write (1, audio_message, msg_size)) != msg_size)
+    toff += msg_size;
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+               "Sending %u bytes of audio data (total: %llu)\n",
+               (unsigned int) msg_size,
+               toff);
+    ptr = (const char *) audio_message;
+    off = 0;
+    while (off < msg_size)
     {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("write"));
-      return;
-    }    
-    transmit_buffer_index += pcm_length;
+      ret = write (1, &ptr[off], msg_size - off);
+      if (0 >= ret)
+      {
+       if (-1 == ret)
+         GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "write");
+       quit (2);
+      }
+      off += ret;
+    }
   }
 
-  int new_size = transmit_buffer_length - transmit_buffer_index;
+  new_size = transmit_buffer_length - transmit_buffer_index;
   if (0 != new_size)
   {
-    transmit_buffer = pa_xrealloc (transmit_buffer, new_size);
-    memcpy (transmit_buffer, transmit_buffer + transmit_buffer_index,
-           new_size);
-    
-    transmit_buffer_index = 0;
-    transmit_buffer_length = new_size;
+    nbuf = pa_xmalloc (new_size);
+    memmove (nbuf,
+            &transmit_buffer[transmit_buffer_index],
+            new_size);
+    pa_xfree (transmit_buffer);
+    transmit_buffer = nbuf;
   }
+  else
+  {
+    pa_xfree (transmit_buffer);
+    transmit_buffer = NULL;
+  }
+  transmit_buffer_index = 0;
+  transmit_buffer_length = new_size;
 }
 
 
@@ -188,31 +215,38 @@ packetizer ()
  * Pulseaudio callback when new data is available.
  */
 static void
-stream_read_callback (pa_stream * s, size_t length, void *userdata)
+stream_read_callback (pa_stream * s,
+                     size_t length,
+                     void *userdata)
 {
   const void *data;
 
-  GNUNET_assert (s);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+             "Got %u/%u bytes of PCM data\n",
+             length,
+             pcm_length);
+
+  GNUNET_assert (NULL != s);
   GNUNET_assert (length > 0);
   if (stdio_event)
     mainloop_api->io_enable (stdio_event, PA_IO_EVENT_OUTPUT);
 
   if (pa_stream_peek (s, (const void **) &data, &length) < 0)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("pa_stream_peek() failed: %s\n"),
-                 pa_strerror (pa_context_errno (context)));
-      quit (1);
-      return;
-    }
-
-  GNUNET_assert (data);
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("pa_stream_peek() failed: %s\n"),
+               pa_strerror (pa_context_errno (context)));
+    quit (1);
+    return;
+  }
+  GNUNET_assert (NULL != data);
   GNUNET_assert (length > 0);
-
-  if (transmit_buffer)
+  if (NULL != transmit_buffer)
   {
-    transmit_buffer =
-      pa_xrealloc (transmit_buffer, transmit_buffer_length + length);
-    memcpy ((uint8_t *) transmit_buffer + transmit_buffer_length, data,
+    transmit_buffer = pa_xrealloc (transmit_buffer,
+                                  transmit_buffer_length + length);
+    memcpy (&transmit_buffer[transmit_buffer_length],
+           data,
            length);
     transmit_buffer_length += length;
   }
@@ -232,12 +266,12 @@ stream_read_callback (pa_stream * s, size_t length, void *userdata)
  * Exit callback for SIGTERM and SIGINT
  */
 static void
-exit_signal_callback (pa_mainloop_api * m, 
-                     pa_signal_event * e, 
+exit_signal_callback (pa_mainloop_api * m,
+                     pa_signal_event * e,
                      int sig,
                      void *userdata)
 {
-  GNUNET_log (GNUNET_ERROR_TYPE_INFO, 
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO,
              _("Got signal, exiting.\n"));
   quit (1);
 }
@@ -249,61 +283,57 @@ exit_signal_callback (pa_mainloop_api * m,
 static void
 stream_state_callback (pa_stream * s, void *userdata)
 {
-  GNUNET_assert (s);
+  GNUNET_assert (NULL != s);
 
   switch (pa_stream_get_state (s))
+  {
+  case PA_STREAM_CREATING:
+  case PA_STREAM_TERMINATED:
+    break;
+  case PA_STREAM_READY:
     {
-    case PA_STREAM_CREATING:
-    case PA_STREAM_TERMINATED:
-      break;
-
-    case PA_STREAM_READY:
-      if (1)
-       {
-         const pa_buffer_attr *a;
-         char cmt[PA_CHANNEL_MAP_SNPRINT_MAX],
-           sst[PA_SAMPLE_SPEC_SNPRINT_MAX];
-
-         GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-                     _("Stream successfully created.\n"));
-
-         if (!(a = pa_stream_get_buffer_attr (s)))
-           {
-             GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                         _("pa_stream_get_buffer_attr() failed: %s\n"),
-                         pa_strerror (pa_context_errno
-                                      (pa_stream_get_context (s))));
-
-           }
-         else
-           {
-             GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-                         _("Buffer metrics: maxlength=%u, fragsize=%u\n"),
-                         a->maxlength, a->fragsize);
-           }
-
-         GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-                     _("Using sample spec '%s', channel map '%s'.\n"),
-                     pa_sample_spec_snprint (sst, sizeof (sst),
-                                             pa_stream_get_sample_spec (s)),
-                     pa_channel_map_snprint (cmt, sizeof (cmt),
-                                             pa_stream_get_channel_map (s)));
-
-         GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-                     _("Connected to device %s (%u, %ssuspended).\n"),
-                     pa_stream_get_device_name (s),
-                     pa_stream_get_device_index (s),
-                     pa_stream_is_suspended (s) ? "" : "not ");
-       }
-
-      break;
-
-    case PA_STREAM_FAILED:
-    default:
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Stream error: %s\n"),
-                 pa_strerror (pa_context_errno (pa_stream_get_context (s))));
-      quit (1);
+      const pa_buffer_attr *a;
+      char cmt[PA_CHANNEL_MAP_SNPRINT_MAX];
+      char sst[PA_SAMPLE_SPEC_SNPRINT_MAX];
+
+      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                 _("Stream successfully created.\n"));
+
+      if (!(a = pa_stream_get_buffer_attr (s)))
+      {
+       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                   _("pa_stream_get_buffer_attr() failed: %s\n"),
+                   pa_strerror (pa_context_errno
+                                (pa_stream_get_context (s))));
+
+      }
+      else
+      {
+       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                   _("Buffer metrics: maxlength=%u, fragsize=%u\n"),
+                   a->maxlength, a->fragsize);
+      }
+      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                 _("Using sample spec '%s', channel map '%s'.\n"),
+                 pa_sample_spec_snprint (sst, sizeof (sst),
+                                         pa_stream_get_sample_spec (s)),
+                 pa_channel_map_snprint (cmt, sizeof (cmt),
+                                         pa_stream_get_channel_map (s)));
+
+      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                 _("Connected to device %s (%u, %ssuspended).\n"),
+                 pa_stream_get_device_name (s),
+                 pa_stream_get_device_index (s),
+                 pa_stream_is_suspended (s) ? "" : "not ");
     }
+    break;
+  case PA_STREAM_FAILED:
+  default:
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("Stream error: %s\n"),
+               pa_strerror (pa_context_errno (pa_stream_get_context (s))));
+    quit (1);
+  }
 }
 
 
@@ -317,58 +347,53 @@ context_state_callback (pa_context * c,
   GNUNET_assert (c);
 
   switch (pa_context_get_state (c))
+  {
+  case PA_CONTEXT_CONNECTING:
+  case PA_CONTEXT_AUTHORIZING:
+  case PA_CONTEXT_SETTING_NAME:
+    break;
+  case PA_CONTEXT_READY:
+  {
+    int r;
+    pa_buffer_attr na;
+
+    GNUNET_assert (!stream_in);
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+               _("Connection established.\n"));
+    if (! (stream_in =
+          pa_stream_new (c, "GNUNET_VoIP recorder", &sample_spec, NULL)))
     {
-    case PA_CONTEXT_CONNECTING:
-    case PA_CONTEXT_AUTHORIZING:
-    case PA_CONTEXT_SETTING_NAME:
-      break;
-
-    case PA_CONTEXT_READY:
-      {
-       int r;
-
-       GNUNET_assert (c);
-       GNUNET_assert (!stream_in);
-
-       GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Connection established.\n"));
-
-       if (!
-           (stream_in =
-            pa_stream_new (c, "GNUNET_VoIP recorder", &sample_spec, NULL)))
-         {
-           GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                       _("pa_stream_new() failed: %s\n"),
-                       pa_strerror (pa_context_errno (c)));
-           goto fail;
-         }
-
-
-       pa_stream_set_state_callback (stream_in, stream_state_callback, NULL);
-       pa_stream_set_read_callback (stream_in, stream_read_callback, NULL);
-
-
-       if ((r = pa_stream_connect_record (stream_in, NULL, NULL, 0)) < 0)
-         {
-           GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                       _("pa_stream_connect_record() failed: %s\n"),
-                       pa_strerror (pa_context_errno (c)));
-           goto fail;
-         }
-
-       break;
-      }
-
-    case PA_CONTEXT_TERMINATED:
-      quit (0);
-      break;
-
-    case PA_CONTEXT_FAILED:
-    default:
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Connection failure: %s\n"),
+      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                 _("pa_stream_new() failed: %s\n"),
+                 pa_strerror (pa_context_errno (c)));
+      goto fail;
+    }
+    pa_stream_set_state_callback (stream_in, &stream_state_callback, NULL);
+    pa_stream_set_read_callback (stream_in, &stream_read_callback, NULL);
+    memset (&na, 0, sizeof (na));
+    na.maxlength = UINT32_MAX;
+    na.fragsize = pcm_length;
+    if ((r = pa_stream_connect_record (stream_in, NULL, &na,
+                                      PA_STREAM_ADJUST_LATENCY)) < 0)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                 _("pa_stream_connect_record() failed: %s\n"),
                  pa_strerror (pa_context_errno (c)));
       goto fail;
     }
 
+    break;
+  }
+  case PA_CONTEXT_TERMINATED:
+    quit (0);
+    break;
+  case PA_CONTEXT_FAILED:
+  default:
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("Connection failure: %s\n"),
+               pa_strerror (pa_context_errno (c)));
+    goto fail;
+  }
   return;
 
 fail:
@@ -386,46 +411,43 @@ pa_init ()
   int i;
 
   if (!pa_sample_spec_valid (&sample_spec))
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Wrong Spec\n"));
-    }
-
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("Wrong Spec\n"));
+  }
   /* set up main record loop */
-
   if (!(m = pa_mainloop_new ()))
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("pa_mainloop_new() failed.\n"));
-    }
-
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("pa_mainloop_new() failed.\n"));
+  }
   mainloop_api = pa_mainloop_get_api (m);
 
   /* listen to signals */
-
   r = pa_signal_init (mainloop_api);
   GNUNET_assert (r == 0);
-  pa_signal_new (SIGINT, exit_signal_callback, NULL);
-  pa_signal_new (SIGTERM, exit_signal_callback, NULL);
+  pa_signal_new (SIGINT, &exit_signal_callback, NULL);
+  pa_signal_new (SIGTERM, &exit_signal_callback, NULL);
 
   /* connect to the main pulseaudio context */
 
   if (!(context = pa_context_new (mainloop_api, "GNUNET VoIP")))
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("pa_context_new() failed.\n"));
-    }
-
-  pa_context_set_state_callback (context, context_state_callback, NULL);
-
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("pa_context_new() failed.\n"));
+  }
+  pa_context_set_state_callback (context, &context_state_callback, NULL);
   if (pa_context_connect (context, NULL, 0, NULL) < 0)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                 _("pa_context_connect() failed: %s\n"),
-                 pa_strerror (pa_context_errno (context)));
-    }
-
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("pa_context_connect() failed: %s\n"),
+               pa_strerror (pa_context_errno (context)));
+  }
   if (pa_mainloop_run (m, &i) < 0)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("pa_mainloop_run() failed.\n"));
-    }
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+               _("pa_mainloop_run() failed.\n"));
+  }
 }
 
 
@@ -435,20 +457,25 @@ pa_init ()
 static void
 opus_init ()
 {
-  opus_int32 sampling_rate = 48000;
   int channels = 1;
   int err;
 
-  frame_size = sampling_rate / 50;
+  frame_size = SAMPLING_RATE / 50;
   pcm_length = frame_size * channels * sizeof (float);
-  enc =
-    opus_encoder_create (sampling_rate, channels, OPUS_APPLICATION_VOIP,
-                        &err);
-  pcm_buffer = (float *) pa_xmalloc (pcm_length);
-  opus_data = (unsigned char *) calloc (max_payload_bytes, sizeof (char));
-
-  audio_message = pa_xmalloc (UINT16_MAX);
-  audio_message->header.type = htons (GNUNET_MESSAGE_TYPE_CONVERSATION_AUDIO);
+  pcm_buffer = pa_xmalloc (pcm_length);
+  opus_data = GNUNET_malloc (max_payload_bytes);
+  enc = opus_encoder_create (SAMPLING_RATE,
+                            channels,
+                            OPUS_APPLICATION_VOIP,
+                            &err);
+  opus_encoder_ctl (enc,
+                   OPUS_SET_PACKET_LOSS_PERC(1));
+  opus_encoder_ctl (enc,
+                   OPUS_SET_COMPLEXITY(10));
+  opus_encoder_ctl (enc,
+                   OPUS_SET_INBAND_FEC(1));
+  opus_encoder_ctl (enc,
+                   OPUS_SET_SIGNAL (OPUS_SIGNAL_VOICE));
 }
 
 
@@ -466,8 +493,11 @@ main (int argc, char *argv[])
                 GNUNET_log_setup ("gnunet-helper-audio-record",
                                   "WARNING",
                                   NULL));
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+             "Audio source starts\n");
+  audio_message = GNUNET_malloc (UINT16_MAX);
+  audio_message->header.type = htons (GNUNET_MESSAGE_TYPE_CONVERSATION_AUDIO);
   opus_init ();
   pa_init ();
-
   return 0;
 }