kopia lustrzana https://github.com/cyoung/stratux
				
				
				
			
		
			
				
	
	
		
			94 wiersze
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
			
		
		
	
	
			94 wiersze
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
//
 | 
						|
// Copyright 2015, Oliver Jowett <oliver@mutability.co.uk>
 | 
						|
//
 | 
						|
 | 
						|
// This file is free software: you may copy, redistribute 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 file 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/>.
 | 
						|
 | 
						|
#include <stdio.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <stdint.h>
 | 
						|
#include <string.h>
 | 
						|
#include <math.h>
 | 
						|
#include <unistd.h>
 | 
						|
 | 
						|
#include "uat.h"
 | 
						|
#include "fec/rs.h"
 | 
						|
 | 
						|
static void *rs_uplink;
 | 
						|
static void *rs_adsb_short;
 | 
						|
static void *rs_adsb_long;
 | 
						|
 | 
						|
#define UPLINK_POLY 0x187
 | 
						|
#define ADSB_POLY 0x187
 | 
						|
 | 
						|
void init_fec(void)
 | 
						|
{
 | 
						|
    rs_adsb_short = init_rs_char(8, /* gfpoly */ ADSB_POLY, /* fcr */ 120, /* prim */ 1, /* nroots */ 12, /* pad */ 225);
 | 
						|
    rs_adsb_long  = init_rs_char(8, /* gfpoly */ ADSB_POLY, /* fcr */ 120, /* prim */ 1, /* nroots */ 14, /* pad */ 207);
 | 
						|
    rs_uplink     = init_rs_char(8, /* gfpoly */ UPLINK_POLY, /* fcr */ 120, /* prim */ 1, /* nroots */ 20, /* pad */ 163);
 | 
						|
}
 | 
						|
 | 
						|
int correct_adsb_frame(uint8_t *to, int *rs_errors)
 | 
						|
{
 | 
						|
    // Try decoding as a Long UAT.
 | 
						|
    // We rely on decode_rs_char not modifying the data if there were
 | 
						|
    // uncorrectable errors.
 | 
						|
    int n_corrected = decode_rs_char(rs_adsb_long, to, NULL, 0);
 | 
						|
    if (n_corrected >= 0 && n_corrected <= 7 && (to[0]>>3) != 0) {
 | 
						|
        // Valid long frame.
 | 
						|
        *rs_errors = n_corrected;
 | 
						|
        return 2;
 | 
						|
    }
 | 
						|
 | 
						|
    // Retry as Basic UAT
 | 
						|
    n_corrected = decode_rs_char(rs_adsb_short, to, NULL, 0);
 | 
						|
    if (n_corrected >= 0 && n_corrected <= 6 && (to[0]>>3) == 0) {
 | 
						|
        // Valid short frame
 | 
						|
        *rs_errors = n_corrected;
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
 | 
						|
    // Failed.
 | 
						|
    *rs_errors = 9999;
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
int correct_uplink_frame(uint8_t *from, uint8_t *to, int *rs_errors)
 | 
						|
{
 | 
						|
    int block;
 | 
						|
    int total_corrected = 0;
 | 
						|
 | 
						|
    for (block = 0; block < UPLINK_FRAME_BLOCKS; ++block) {
 | 
						|
        int i, n_corrected;
 | 
						|
        uint8_t *blockdata = &to[block * UPLINK_BLOCK_DATA_BYTES];
 | 
						|
 | 
						|
        for (i = 0; i < UPLINK_BLOCK_BYTES; ++i)
 | 
						|
            blockdata[i] = from[i * UPLINK_FRAME_BLOCKS + block];
 | 
						|
 | 
						|
        // error-correct in place
 | 
						|
        n_corrected = decode_rs_char(rs_uplink, blockdata, NULL, 0);
 | 
						|
        if (n_corrected < 0 || n_corrected > 10) {
 | 
						|
            // Failed
 | 
						|
            *rs_errors = 9999;
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
 | 
						|
        total_corrected += n_corrected;
 | 
						|
        // next block (if there is one) will overwrite the ECC bytes.
 | 
						|
    }
 | 
						|
 | 
						|
    *rs_errors = total_corrected;
 | 
						|
    return 1;
 | 
						|
}
 |