qo100websdr.c

/*
* Web based SDR Client for SDRplay and RTLsdr
* =============================================================
* Author: DJ0ABR
*
*   (c) DJ0ABR
*   www.dj0abr.de
*
*   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, write to the Free Software
*   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
* 
* q100websdr.c ... file containing main() and calling all other functions 
* 
*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <unistd.h>
#include <rtl-sdr.h>
#include "sdrplay.h"
#include "audio.h"
#include "wf_univ.h"
#include "websocketserver.h"
#include "setqrg.h"
#include "rtlsdr.h"
#include "fifo.h"
#include "ssbfft.h"
#include "wb_fft.h"
#include "cat.h"
#include "minitiouner.h"
#include "setup.h"
#include "plutodrv.h"

void stop_SDR();

int hwtype = 0; // 1=playSDR, 2=rtlsdr
int samplesPerPacket;
char htmldir[256] = { "." };
int stopped = 0;

void sighandler(int signum)
{
    stopped = 1;
	printf("signal %d, exit program\n",signum);
    stop_SDR();
    
    #ifdef WIDEBAND
    remove_udp_minitiouner();
    #endif
    
    #ifdef PLUTO
    pluto_shutdown();
    #endif
    
    save_config();
    
    exit(0);
}

void sighandler_mem(int signum)
{
    stopped = 1;
	printf("memory error, signal %d, exit program\n",signum);
    stop_SDR();
    
    save_config();
    
    exit(0);
}

// check if playSDReshail2 is already running
void isRunning()
{
    int num = 0;
    char s[256];
    sprintf(s,"ps -e | grep playSDReshail2");
    
    FILE *fp = popen(s,"r");
    if(fp)
    {
        // gets the output of the system command
        while (fgets(s, sizeof(s)-1, fp) != NULL) 
        {
            if(strstr(s,"playSDReshail2") && !strstr(s,"grep"))
            {
                if(++num == 2)
                {
                    printf("playSDReshail2 is already running, do not start twice !");
                    pclose(fp);
                    exit(0);
                }
            }
        }
        pclose(fp);
    }
}

// cleans white spaces from beginning, middle and end of a string
char *cleanString(char *str, int cleanspace)
{
static char hs[256];
char *hp = str;
int idx = 0;

    // putze den Anfang
    while(*hp == ' ' || *hp == ',' || *hp == '\n' || *hp == '\r' || *hp == '\'' || *hp == '\"')
    {
 hp++;
    }
    
    // putze die Mitte
    if(cleanspace)
    {
 while(*hp)
 {
     if(*hp != ' ' && *hp != ',' && *hp != '\n' && *hp != '\r' && *hp != '\'' && *hp != '\"')
  hs[idx++] = *hp;
     hp++;
 }
    }
    else
    {
 while(*hp)
     hs[idx++] = *hp++;
    }
    
    // putze das Ende
    hp = hs+idx-1;

    while(*hp == ' ' || *hp == ',' || *hp == '\n' || *hp == '\r' || *hp == '\'' || *hp == '\"')
    {
 *hp = 0;
 hp--;
    }
    

    hs[idx] = 0;

    return hs;
}

int sret;
void *pret;

void searchHTMLpath()
{
    if(htmldir[0] == '.')
    {
        // search for the apache woking directory
        sret = system("find  /srv -xdev -name htdocs  -print > pfad 2>/dev/null");
        sret = system("find  /var -xdev -name htdocs  -print >> pfad 2>/dev/null");
        sret = system("find  /var -xdev -name html  -print >> pfad 2>/dev/null");
        sret = system("find  /usr -xdev -name htdocs  -print >> pfad 2>/dev/null");
        // if the directory was found its name is in file: "pfad"
        FILE *fp=fopen("./pfad","r");
        if(fp)
        {
            char p[256];
            pret = fgets(p,255,fp);
            char *cp= cleanString(p,0);
            if(strlen(cp)>3)
            {
                strcpy(htmldir,cp);
                printf("Webserver Path: %s\n",htmldir);
            }
            else
            {
                printf("Path to apache webserver files not found\n");
                exit(0);
            }
            
            fclose(fp);
        }
        else
        {
            printf("Path to apache webserver files not found\n");
            exit(0);
        }
    }
}

void installHTMLfiles()
{
    // the HTML files are located in the html folder below the es folder
    // copy these files into the Apache HTML folder
    int i;
    if ((i=getuid()))
    {
        printf("\nYou are not root, HTML files are not automatically copied into the Apache folder! Start this program ONCE with root privileges\n");
        return;
    }
    
    char fn[512];
    snprintf(fn,sizeof(fn),"cp ./html/* %s",htmldir);
    printf("copy Web Site files to: %s",htmldir);
    FILE *fp = popen(fn,"r");
    if(fp)
    {
        // Liest die Ausgabe von cp
        while (fgets(fn, sizeof(fn)-1, fp) != NULL) 
        {
            //printf("Output: %s\n",fn);
        }
        pclose(fp);
    }
}

void calc_system_parameters()
{
    // calculate settings
    calc_setup();

    // load external configuration (if used)
    readAMSConfig();
    
    printf("\nSDR parameters:\n\n");
    printf("SDR base QRG:    %d Hz\n",tuned_frequency);
    printf("SDR sample rate: %d S/s\n",SDR_SAMPLE_RATE);
    printf("WF width:        %d Hz\n",WF_RANGE_HZ);
    printf("WF width:        %d pixel\n",WF_WIDTH);
    printf("1st downsampling:%d S/s\n",(WF_RANGE_HZ * 2));
    printf("1st decim. rate: %d\n",SDR_SAMPLE_RATE / (WF_RANGE_HZ * 2));
    printf("1st FFT resol.:  %d Hz\n",WF_RANGE_HZ / WF_WIDTH);
    printf("1st FTT smp/pass:%d\n",((WF_RANGE_HZ * 2) / (WF_RANGE_HZ / WF_WIDTH)));
}

void start_SDR()
{
    #ifdef EXTUDP
        hwtype = 0;
        return;
    #else
        #ifdef WIDEBAND
            #ifdef SDR_PLAY
            if(init_SDRplay())
                hwtype = 1;
            #endif
            #ifdef PLUTO
                if(init_pluto())
                {
                    hwtype = 3;
                }
            #endif
        #else
            // for NB transponder try RTLsdr first
            if(init_rtlsdr())
            {
                hwtype = 2;
                samplesPerPacket = SAMPLES_PER_PASS;
            }
            
            #ifdef PLUTO
                if(init_pluto())
                {
                    hwtype = 3;
                }
            #endif

            #ifdef SDR_PLAY
                if(hwtype == 0)
                {
                    if(init_SDRplay())
                        hwtype = 1;
                }
            #endif
        #endif
        

        if(hwtype == 0)
        {
            printf("no SDR hardware found.\n");
            save_config();
            exit(0);
        }
    #endif
}

void stop_SDR()
{
#ifndef EXTUDP
#ifdef SDR_PLAY
    if(hwtype == 1) remove_SDRplay();
#endif
    if(hwtype == 2) rtlsdr_close(0);
#endif    
}

int main(int argc, char *argv[])
{
    // check if it is already running, if yes then exit
    isRunning();
    
    // look for the apache HTML path
    searchHTMLpath();
    
    // Install or Update the html files
    installHTMLfiles();
    
    // signal handler, mainly used if the user presses Ctrl-C
    struct sigaction sigact;
    sigact.sa_handler = sighandler;
	sigemptyset(&sigact.sa_mask);
	sigact.sa_flags = 0;
	sigaction(SIGINT, &sigact, NULL);
	sigaction(SIGTERM, &sigact, NULL);
	sigaction(SIGQUIT, &sigact, NULL);
	//sigaction(SIGPIPE, &sigact, NULL); // signal 13
    
    // switch off signal 13 (broken pipe)
    // instead handle the return value of the write or send function
    signal(SIGPIPE, SIG_IGN);
    
    /*struct sigaction sigact_mem;
    sigact_mem.sa_handler = sighandler_mem;
	sigemptyset(&sigact_mem.sa_mask);
	sigact_mem.sa_flags = 0;
    sigaction(SIGSEGV, &sigact_mem, NULL);*/
    
    calc_system_parameters();
    
    #ifdef WIDEBAND
        init_fwb();
        init_udp_minitiouner();
    #else
        init_fssb();
    #endif // WIDEBAND
    
    // init the FIFO
    initpipe();
    
    // init the Websocket Server
    ws_init();
    
    // init audio output
    #ifdef SOUNDLOCAL
        init_soundprocessing();
    #endif
        
    // init CAT interface
    cat_init();
    
    printf("\nInitialisation complete, system running ... stop with Ctrl-C\n");
    
    // init SDRplay hardware
    // this MUST be the LAST initialisation because
    // the callback starts immediately after init_SDRplay
    start_SDR();

    // infinite loop, 
    // stop program with Ctrl-C
    while(1)
    {
        if(retune_setup)
        {
            
            // setup requested new settings of the tuner
            retune_setup = 0;
            
            save_config();
            calc_system_parameters();
            #ifndef EXTUDP
            printf("set pluto freq\n");
            re_set_freq();
            #endif
        }
        #ifndef EXTUDP
        set_frequency();
        #endif
        if(configrequest)
            sendConfigToBrowser();
        usleep(1000);
    }
    
    removepipe();
    save_config();
    
    return 0;
}