PierMesh
/
direwolf
mirror of https://github.com/wb2osz/direwolf.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
direwolf/gen_packets.c

730 lines
21 KiB
C
Raw Permalink Blame History

//
// This file is part of Dire Wolf, an amateur radio packet TNC.
//
// Copyright (C) 2011,2013 John Langner, WB2OSZ
//
// This program 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 2 of the License, or
// (at your option) any later version.
//
// This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
//
/*------------------------------------------------------------------
*
* Name: gen_packets.c
*
* Purpose: Test program for generating AFSK AX.25 frames.
*
* Description: Given messages are converted to audio and written
* to a .WAV type audio file.
*
*
* Bugs: Most options not implemented for second audio channel.
*
*------------------------------------------------------------------*/
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <string.h>
#include <assert.h>
#include "audio.h"
#include "ax25_pad.h"
#include "hdlc_send.h"
#include "gen_tone.h"
#include "textcolor.h"
static void usage (char **argv);
static int audio_file_open (char *fname, struct audio_s *pa);
static int audio_file_close (void);
static int g_add_noise = 0;
static float g_noise_level = 0;
int main(int argc, char **argv)
{
int c;
int digit_optind = 0;
int err;
unsigned char fbuf[AX25_MAX_PACKET_LEN+2];
int flen;
int packet_count = 0;
int i;
int chan;
/*
* Set up default values for the modem.
*/
struct audio_s modem;
modem.num_channels = DEFAULT_NUM_CHANNELS; /* -2 stereo */
modem.samples_per_sec = DEFAULT_SAMPLES_PER_SEC; /* -r option */
modem.bits_per_sample = DEFAULT_BITS_PER_SAMPLE; /* -8 for 8 instead of 16 bits */
for (chan = 0; chan < MAX_CHANS; chan++) {
modem.modem_type[chan] = AFSK; /* change with -g */
modem.mark_freq[chan] = DEFAULT_MARK_FREQ; /* -m option */
modem.space_freq[chan] = DEFAULT_SPACE_FREQ; /* -s option */
modem.baud[chan] = DEFAULT_BAUD; /* -b option */
}
/*
* Set up other default values.
*/
int amplitude = 50; /* -a option */
int leading_zeros = 12; /* -z option */
char output_file[256]; /* -o option */
FILE *input_fp = NULL; /* File or NULL for built-in message */
packet_t pp;
strcpy (output_file, "");
while (1) {
int this_option_optind = optind ? optind : 1;
int option_index = 0;
static struct option long_options[] = {
{"future1", 1, 0, 0},
{"future2", 0, 0, 0},
{"future3", 1, 0, 'c'},
{0, 0, 0, 0}
};
/* ':' following option character means arg is required. */
c = getopt_long(argc, argv, "gm:s:a:b:B:r:n:o:z:82",
long_options, &option_index);
if (c == -1)
break;
switch (c) {
case 0: /* possible future use */
text_color_set(DW_COLOR_INFO);
dw_printf("option %s", long_options[option_index].name);
if (optarg) {
dw_printf(" with arg %s", optarg);
}
dw_printf("\n");
break;
case 'b': /* -b for data Bit rate */
modem.baud[0] = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Data rate set to %d bits / second.\n", modem.baud[0]);
if (modem.baud[0] < 100 || modem.baud[0] > 10000) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable bit rate in range of 100 - 10000.\n");
exit (EXIT_FAILURE);
}
break;
case 'B': /* -B for data Bit rate */
/* 300 implies 1600/1800 AFSK. */
/* 1200 implies 1200/2200 AFSK. */
/* 9600 implies scrambled. */
modem.baud[0] = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Data rate set to %d bits / second.\n", modem.baud[0]);
if (modem.baud[0] < 100 || modem.baud[0] > 10000) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable bit rate in range of 100 - 10000.\n");
exit (EXIT_FAILURE);
}
switch (modem.baud[0]) {
case 300:
modem.mark_freq[0] = 1600;
modem.space_freq[0] = 1800;
break;
case 1200:
modem.mark_freq[0] = 1200;
modem.space_freq[0] = 2200;
break;
case 9600:
modem.modem_type[0] = SCRAMBLE;
text_color_set(DW_COLOR_INFO);
dw_printf ("Using scrambled baseband signal rather than AFSK.\n");
break;
}
break;
case 'g': /* -g for g3ruh scrambling */
modem.modem_type[0] = SCRAMBLE;
text_color_set(DW_COLOR_INFO);
dw_printf ("Using scrambled baseband signal rather than AFSK.\n");
break;
case 'm': /* -m for Mark freq */
modem.mark_freq[0] = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Mark frequency set to %d Hz.\n", modem.mark_freq[0]);
if (modem.mark_freq[0] < 300 || modem.mark_freq[0] > 3000) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable value in range of 300 - 3000.\n");
exit (EXIT_FAILURE);
}
break;
case 's': /* -s for Space freq */
modem.space_freq[0] = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Space frequency set to %d Hz.\n", modem.space_freq[0]);
if (modem.space_freq[0] < 300 || modem.space_freq[0] > 3000) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable value in range of 300 - 3000.\n");
exit (EXIT_FAILURE);
}
break;
case 'n': /* -n number of packets with increasing noise. */
packet_count = atoi(optarg);
g_add_noise = 1;
break;
case 'a': /* -a for amplitude */
amplitude = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Amplitude set to %d%%.\n", amplitude);
if (amplitude < 0 || amplitude > 100) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Amplitude must be in range of 0 to 100.\n");
exit (EXIT_FAILURE);
}
break;
case 'r': /* -r for audio sample Rate */
modem.samples_per_sec = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Audio sample rate set to %d samples / second.\n", modem.samples_per_sec);
if (modem.samples_per_sec < MIN_SAMPLES_PER_SEC || modem.samples_per_sec > MAX_SAMPLES_PER_SEC) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable audio sample rate in range of %d - %d.\n",
MIN_SAMPLES_PER_SEC, MAX_SAMPLES_PER_SEC);
exit (EXIT_FAILURE);
}
break;
case 'z': /* -z leading zeros before frame flag */
leading_zeros = atoi(optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Send %d zero bits before frame flag.\n", leading_zeros);
/* The demodulator needs a few for the clock recovery PLL. */
/* We don't want to be here all day either. */
/* We can't translast to time yet because the data bit rate */
/* could be changed later. */
if (leading_zeros < 8 || leading_zeros > 12000) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Use a more reasonable value.\n");
exit (EXIT_FAILURE);
}
break;
case '8': /* -8 for 8 bit samples */
modem.bits_per_sample = 8;
text_color_set(DW_COLOR_INFO);
dw_printf("8 bits per audio sample rather than 16.\n");
break;
case '2': /* -2 for 2 channels of sound */
modem.num_channels = 2;
text_color_set(DW_COLOR_INFO);
dw_printf("2 channels of sound rather than 1.\n");
break;
case 'o': /* -o for Output file */
strcpy (output_file, optarg);
text_color_set(DW_COLOR_INFO);
dw_printf ("Output file set to %s\n", output_file);
break;
case '?':
/* Unknown option message was already printed. */
usage (argv);
break;
default:
/* Should not be here. */
text_color_set(DW_COLOR_ERROR);
dw_printf("?? getopt returned character code 0%o ??\n", c);
usage (argv);
}
}
if (optind < argc) {
char str[400];
// dw_printf("non-option ARGV-elements: ");
// while (optind < argc)
// dw_printf("%s ", argv[optind++]);
//dw_printf("\n");
if (optind < argc - 1) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Warning: File(s) beyond the first are ignored.\n");
}
if (strcmp(argv[optind], "-") == 0) {
text_color_set(DW_COLOR_INFO);
dw_printf ("Reading from stdin ...\n");
input_fp = stdin;
}
else {
input_fp = fopen(argv[optind], "r");
if (input_fp == NULL) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Can't open %s for read.\n", argv[optind]);
exit (EXIT_FAILURE);
}
text_color_set(DW_COLOR_INFO);
dw_printf ("Reading from %s ...\n", argv[optind]);
}
while (fgets (str, sizeof(str), input_fp) != NULL) {
text_color_set(DW_COLOR_REC);
dw_printf ("%s", str);
}
if (input_fp != stdin) {
fclose (input_fp);
}
}
else {
text_color_set(DW_COLOR_INFO);
dw_printf ("built in message...\n");
}
if (strlen(output_file) == 0) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("ERROR: The -o ouput file option must be specified.\n");
usage (argv);
exit (1);
}
err = audio_file_open (output_file, &modem);
if (err < 0) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("ERROR - Can't open output file.\n");
exit (1);
}
gen_tone_init (&modem, amplitude);
assert (modem.bits_per_sample == 8 || modem.bits_per_sample == 16);
assert (modem.num_channels == 1 || modem.num_channels == 2);
assert (modem.samples_per_sec >= MIN_SAMPLES_PER_SEC && modem.samples_per_sec <= MAX_SAMPLES_PER_SEC);
if (packet_count > 0) {
/*
* Generate packets with increasing noise level.
* Would probably be better to record real noise from a radio but
* for now just use a random number generator.
*/
for (i = 1; i <= packet_count; i++) {
char stemp[80];
if (modem.modem_type[0] == SCRAMBLE) {
g_noise_level = 0.33 * (amplitude / 100.0) * ((float)i / packet_count);
}
else {
g_noise_level = (amplitude / 100.0) * ((float)i / packet_count);
}
sprintf (stemp, "WB2OSZ-1>APRS,W1AB-9,W1ABC-10,WB1ABC-15:,Hello, world! %04d", i);
pp = ax25_from_text (stemp, 1);
flen = ax25_pack (pp, fbuf);
for (c=0; c<modem.num_channels; c++)
{
hdlc_send_flags (c, 8, 0);
hdlc_send_frame (c, fbuf, flen);
hdlc_send_flags (c, 2, 1);
}
ax25_delete (pp);
}
}
else {
/*
* Builtin default 4 packets.
*/
pp = ax25_from_text ("WB2OSZ-1>APRS,W1AB-9,W1ABC-10,WB1ABC-15:,Hello, world!", 1);
flen = ax25_pack (pp, fbuf);
for (c=0; c<modem.num_channels; c++)
{
hdlc_send_flags (c, 8, 0);
hdlc_send_frame (c, fbuf, flen);
hdlc_send_flags (c, 2, 1);
}
ax25_delete (pp);
hdlc_send_flags (c, 8, 0);
pp = ax25_from_text ("WB2OSZ-1>APRS,W1AB-9*,W1ABC-10,WB1ABC-15:,Hello, world!", 1);
flen = ax25_pack (pp, fbuf);
for (c=0; c<modem.num_channels; c++)
{
hdlc_send_frame (c, fbuf, flen);
}
ax25_delete (pp);
pp = ax25_from_text ("WB2OSZ-1>APRS,W1AB-9,W1ABC-10*,WB1ABC-15:,Hello, world!", 1);
flen = ax25_pack (pp, fbuf);
for (c=0; c<modem.num_channels; c++)
{
hdlc_send_frame (c, fbuf, flen);
}
ax25_delete (pp);
pp = ax25_from_text ("WB2OSZ-1>APRS,W1AB-9,W1ABC-10,WB1ABC-15*:,Hello, world!", 1);
flen = ax25_pack (pp, fbuf);
for (c=0; c<modem.num_channels; c++)
{
hdlc_send_frame (c, fbuf, flen);
}
ax25_delete (pp);
hdlc_send_flags (c, 2, 1);
}
audio_file_close();
return EXIT_SUCCESS;
}
static void usage (char **argv)
{
text_color_set(DW_COLOR_ERROR);
dw_printf ("\n");
dw_printf ("Usage: xxx [options] [file]\n");
dw_printf ("Options:\n");
dw_printf (" -a <number> Signal amplitude in range of 0 - 100%%. Default 50.\n");
dw_printf (" -b <number> Bits / second for data. Default is %d.\n", DEFAULT_BAUD);
dw_printf (" -B <number> Bits / second for data. Proper modem selected for 300, 1200, 9600.\n");
dw_printf (" -g Scrambled baseband rather than AFSK.\n");
dw_printf (" -m <number> Mark frequency. Default is %d.\n", DEFAULT_MARK_FREQ);
dw_printf (" -s <number> Space frequency. Default is %d.\n", DEFAULT_SPACE_FREQ);
dw_printf (" -r <number> Audio sample Rate. Default is %d.\n", DEFAULT_SAMPLES_PER_SEC);
dw_printf (" -n <number> Generate specified number of frames with increasing noise.\n");
dw_printf (" -o <file> Send output to .wav file.\n");
dw_printf (" -8 8 bit audio rather than 16.\n");
dw_printf (" -2 2 channels of audio rather than 1.\n");
dw_printf (" -z <number> Number of leading zero bits before frame.\n");
dw_printf (" Default is 12 which is .01 seconds at 1200 bits/sec.\n");
dw_printf ("\n");
dw_printf ("An optional file may be specified to provide messages other than\n");
dw_printf ("the default built-n message. The format should correspond to ...\n");
dw_printf ("blah blah blah. For example,\n");
dw_printf (" WB2OSZ-1>APDW10,WIDE2-2:!4237.14NS07120.83W#\n");
dw_printf ("\n");
dw_printf ("Example: %s\n", argv[0]);
dw_printf ("\n");
dw_printf (" With all defaults, a built-in test message is generated\n");
dw_printf (" with standard Bell 202 tones used for packet radio on ordinary\n");
dw_printf (" VHF FM transceivers.\n");
dw_printf ("\n");
dw_printf ("Example: %s -g -b 9600\n", argv[0]);
dw_printf ("Shortcut: %s -B 9600\n", argv[0]);
dw_printf ("\n");
dw_printf (" 9600 baud mode.\n");
dw_printf ("\n");
dw_printf ("Example: %s -m 1600 -s 1800 -b 300\n", argv[0]);
dw_printf ("Shortcut: %s -B 300\n", argv[0]);
dw_printf ("\n");
dw_printf (" 200 Hz shift, 300 baud, suitable for HF SSB transceiver.\n");
dw_printf ("\n");
dw_printf ("Example: echo -n \"WB2OSZ>WORLD:Hello, world!\" | %s -a 25 -o x.wav -\n", argv[0]);
dw_printf ("\n");
dw_printf (" Read message from stdin and put quarter volume sound into the file x.wav.\n");
exit (EXIT_FAILURE);
}
/*------------------------------------------------------------------
*
* Name: audio_file_open
*
* Purpose: Open a .WAV format file for output.
*
* Inputs: fname - Name of .WAV file to create.
*
* pa - Address of structure of type audio_s.
*
* The fields that we care about are:
* num_channels
* samples_per_sec
* bits_per_sample
* If zero, reasonable defaults will be provided.
*
* Returns: 0 for success, -1 for failure.
*
*----------------------------------------------------------------*/
struct wav_header { /* .WAV file header. */
char riff[4]; /* "RIFF" */
int filesize; /* file length - 8 */
char wave[4]; /* "WAVE" */
char fmt[4]; /* "fmt " */
int fmtsize; /* 16. */
short wformattag; /* 1 for PCM. */
short nchannels; /* 1 for mono, 2 for stereo. */
int nsamplespersec; /* sampling freq, Hz. */
int navgbytespersec; /* = nblockalign * nsamplespersec. */
short nblockalign; /* = wbitspersample / 8 * nchannels. */
short wbitspersample; /* 16 or 8. */
char data[4]; /* "data" */
int datasize; /* number of bytes following. */
} ;
/* 8 bit samples are unsigned bytes */
/* in range of 0 .. 255. */
/* 16 bit samples are signed short */
/* in range of -32768 .. +32767. */
static FILE *out_fp = NULL;
static struct wav_header header;
static int byte_count; /* Number of data bytes written to file. */
/* Will be written to header when file is closed. */
static int audio_file_open (char *fname, struct audio_s *pa)
{
int n;
/*
* Fill in defaults for any missing values.
*/
if (pa -> num_channels == 0)
pa -> num_channels = DEFAULT_NUM_CHANNELS;
if (pa -> samples_per_sec == 0)
pa -> samples_per_sec = DEFAULT_SAMPLES_PER_SEC;
if (pa -> bits_per_sample == 0)
pa -> bits_per_sample = DEFAULT_BITS_PER_SAMPLE;
/*
* Write the file header. Don't know length yet.
*/
out_fp = fopen (fname, "wb");
if (out_fp == NULL) {
text_color_set(DW_COLOR_ERROR); dw_printf ("Couldn't open file for write: %s\n", fname);
perror ("");
return (-1);
}
memset (&header, 0, sizeof(header));
memcpy (header.riff, "RIFF", (size_t)4);
header.filesize = 0;
memcpy (header.wave, "WAVE", (size_t)4);
memcpy (header.fmt, "fmt ", (size_t)4);
header.fmtsize = 16; // Always 16.
header.wformattag = 1; // 1 for PCM.
header.nchannels = pa -> num_channels;
header.nsamplespersec = pa -> samples_per_sec;
header.wbitspersample = pa -> bits_per_sample;
header.nblockalign = header.wbitspersample / 8 * header.nchannels;
header.navgbytespersec = header.nblockalign * header.nsamplespersec;
memcpy (header.data, "data", (size_t)4);
header.datasize = 0;
assert (header.nchannels == 1 || header.nchannels == 2);
n = fwrite (&header, sizeof(header), (size_t)1, out_fp);
if (n != 1) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Couldn't write header to: %s\n", fname);
perror ("");
fclose (out_fp);
out_fp = NULL;
return (-1);
}
/*
* Number of bytes written will be filled in later.
*/
byte_count = 0;
return (0);
} /* end audio_open */
/*------------------------------------------------------------------
*
* Name: audio_put
*
* Purpose: Send one byte to the audio output file.
*
* Inputs: c - One byte in range of 0 - 255.
*
* Returns: Normally non-negative.
* -1 for any type of error.
*
* Description: The caller must deal with the details of mono/stereo
* and number of bytes per sample.
*
*----------------------------------------------------------------*/
int audio_put (int c)
{
static short sample16;
int s;
if (g_add_noise) {
if ((byte_count & 1) == 0) {
sample16 = c & 0xff; /* save lower byte. */
byte_count++;
return c;
}
else {
float r;
sample16 |= (c << 8) & 0xff00; /* insert upper byte. */
byte_count++;
s = sample16; // sign extend.
/* Add random noise to the signal. */
/* r should be in range of -1 .. +1. */
r = (rand() - RAND_MAX/2.0) / (RAND_MAX/2.0);
s += 5 * r * g_noise_level * 32767;
if (s > 32767) s = 32767;
if (s < -32767) s = -32767;
putc(s & 0xff, out_fp);
return (putc((s >> 8) & 0xff, out_fp));
}
}
else {
byte_count++;
return (putc(c, out_fp));
}
} /* end audio_put */
int audio_flush ()
{
return 0;
}
/*------------------------------------------------------------------
*
* Name: audio_file_close
*
* Purpose: Close the audio output file.
*
* Returns: Normally non-negative.
* -1 for any type of error.
*
*
* Description: Must go back to beginning of file and fill in the
* size of the data.
*
*----------------------------------------------------------------*/
static int audio_file_close (void)
{
int n;
text_color_set(DW_COLOR_DEBUG);
dw_printf ("audio_close()\n");
/*
* Go back and fix up lengths in header.
*/
header.filesize = byte_count + sizeof(header) - 8;
header.datasize = byte_count;
if (out_fp == NULL) {
return (-1);
}
fflush (out_fp);
fseek (out_fp, 0L, SEEK_SET);
n = fwrite (&header, sizeof(header), (size_t)1, out_fp);
if (n != 1) {
text_color_set(DW_COLOR_ERROR);
dw_printf ("Couldn't write header to audio file.\n");
perror (""); // TODO: remove perror.
fclose (out_fp);
out_fp = NULL;
return (-1);
}
fclose (out_fp);
out_fp = NULL;
return (0);
} /* end audio_close */
Reference in New Issue View Git Blame Copy Permalink