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;
 | |
| }
 |