mirror of
https://github.com/eddyem/STM8_samples.git
synced 2025-12-06 10:45:12 +03:00
260 lines
6.5 KiB
C
260 lines
6.5 KiB
C
/*
|
|
* client.c - simple terminal client
|
|
*
|
|
* Copyright 2013 Edward V. Emelianoff <eddy@sao.ru>
|
|
*
|
|
* 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., 51 Franklin Street, Fifth Floor, Boston,
|
|
* MA 02110-1301, USA.
|
|
*/
|
|
#include <termios.h> // tcsetattr
|
|
#include <unistd.h> // tcsetattr, close, read, write
|
|
#include <sys/ioctl.h> // ioctl
|
|
#include <stdio.h> // printf, getchar, fopen, perror
|
|
#include <stdlib.h> // exit
|
|
#include <sys/stat.h> // read
|
|
#include <fcntl.h> // read
|
|
#include <signal.h> // signal
|
|
#include <time.h> // time
|
|
#include <string.h> // memcpy
|
|
#include <stdint.h> // int types
|
|
#include <sys/time.h> // gettimeofday
|
|
|
|
#define CMSPAR 010000000000
|
|
|
|
double t0; // start time
|
|
|
|
FILE *fout = NULL; // file for messages duplicating
|
|
char *comdev = "/dev/ttyUSB0";
|
|
int BAUD_RATE = B57600;
|
|
struct termio oldtty, tty; // TTY flags
|
|
struct termios oldt, newt; // terminal flags
|
|
int comfd; // TTY fd
|
|
|
|
/**
|
|
* function for different purposes that need to know time intervals
|
|
* @return double value: time in seconds
|
|
*/
|
|
double dtime(){
|
|
double t;
|
|
struct timeval tv;
|
|
gettimeofday(&tv, NULL);
|
|
t = tv.tv_sec + ((double)tv.tv_usec)/1e6;
|
|
return t;
|
|
}
|
|
|
|
/**
|
|
* Exit & return terminal to old state
|
|
* @param ex_stat - status (return code)
|
|
*/
|
|
void quit(int ex_stat){
|
|
tcsetattr(STDIN_FILENO, TCSANOW, &oldt); // return terminal to previous state
|
|
ioctl(comfd, TCSANOW, &oldtty ); // return TTY to previous state
|
|
close(comfd);
|
|
if(fout) fclose(fout);
|
|
printf("Exit! (%d)\n", ex_stat);
|
|
exit(ex_stat);
|
|
}
|
|
|
|
unsigned char crc(unsigned char data){
|
|
unsigned char crc = data & 1;
|
|
unsigned int i;
|
|
for(i = 1; i<8; i++) crc ^= (data >> i) & 1;
|
|
return crc;
|
|
}
|
|
|
|
/**
|
|
* Open & setup TTY, terminal
|
|
*/
|
|
void tty_init(){
|
|
// terminal without echo
|
|
tcgetattr(STDIN_FILENO, &oldt);
|
|
newt = oldt;
|
|
newt.c_lflag &= ~(ICANON | ECHO);
|
|
if(tcsetattr(STDIN_FILENO, TCSANOW, &newt) < 0) quit(-2);
|
|
printf("\nOpen port...\n");
|
|
if ((comfd = open(comdev,O_RDWR|O_NOCTTY|O_NONBLOCK)) < 0){
|
|
fprintf(stderr,"Can't use port %s\n",comdev);
|
|
quit(1);
|
|
}
|
|
printf(" OK\nGet current settings...\n");
|
|
if(ioctl(comfd,TCGETA,&oldtty) < 0) exit(-1); // Get settings
|
|
tty = oldtty;
|
|
tty.c_lflag = 0; // ~(ICANON | ECHO | ECHOE | ISIG)
|
|
tty.c_oflag = 0;
|
|
tty.c_iflag = 0;
|
|
tty.c_cflag = BAUD_RATE|CS8|CREAD|CLOCAL|PARENB; // we will emulate 9bit by PAR
|
|
tty.c_cc[VMIN] = 0; // non-canonical mode
|
|
tty.c_cc[VTIME] = 5;
|
|
if(ioctl(comfd,TCSETA,&tty) < 0) exit(-1); // set new mode
|
|
printf(" OK\n");
|
|
tcsetattr(STDIN_FILENO, TCSANOW, &newt);
|
|
}
|
|
|
|
/**
|
|
* Read character from console without echo
|
|
* @return char readed
|
|
*/
|
|
int read_console(){
|
|
int rb;
|
|
struct timeval tv;
|
|
int retval;
|
|
fd_set rfds;
|
|
FD_ZERO(&rfds);
|
|
FD_SET(STDIN_FILENO, &rfds);
|
|
tv.tv_sec = 0; tv.tv_usec = 10000;
|
|
retval = select(1, &rfds, NULL, NULL, &tv);
|
|
if(!retval) rb = 0;
|
|
else {
|
|
if(FD_ISSET(STDIN_FILENO, &rfds)) rb = getchar();
|
|
else rb = 0;
|
|
}
|
|
return rb;
|
|
}
|
|
|
|
/**
|
|
* getchar() without echo
|
|
* wait until at least one character pressed
|
|
* @return character readed
|
|
*/
|
|
int mygetchar(){ // аналог getchar() без необходимости жать Enter
|
|
int ret;
|
|
do ret = read_console();
|
|
while(ret == 0);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Read data from TTY
|
|
* @param buff (o) - buffer for data read
|
|
* @param length - buffer len
|
|
* @return amount of readed bytes
|
|
*/
|
|
size_t read_tty(uint8_t *buff, size_t length){
|
|
ssize_t L = 0;
|
|
fd_set rfds;
|
|
struct timeval tv;
|
|
int retval;
|
|
FD_ZERO(&rfds);
|
|
FD_SET(comfd, &rfds);
|
|
tv.tv_sec = 0; tv.tv_usec = 10000; // wait for 10ms
|
|
retval = select(comfd + 1, &rfds, NULL, NULL, &tv);
|
|
if (!retval) return 0;
|
|
if(FD_ISSET(comfd, &rfds)){
|
|
if((L = read(comfd, buff, length)) < 1) return 0;
|
|
}
|
|
return (size_t)L;
|
|
}
|
|
|
|
void help(){
|
|
printf("Use this commands:\n"
|
|
"H\tShow this help\n"
|
|
"D\tChange device number\n"
|
|
"q\tQuit\n"
|
|
);
|
|
}
|
|
|
|
#define dup_pr(...) do{printf(__VA_ARGS__); if(fout) fprintf(fout, __VA_ARGS__);}while(0)
|
|
|
|
/**
|
|
* Set/reset nineth bit and send command
|
|
* @param
|
|
* @return
|
|
*/
|
|
void send_with_9(unsigned char cmd, int nineth){
|
|
if(crc(cmd) ^ nineth) // (ODD CRC) XOR (set nineth) -> odd parity
|
|
tty.c_cflag |= PARODD; // odd parity
|
|
else
|
|
tty.c_cflag &= ~PARODD; // even parity
|
|
if(ioctl(comfd, TCSETA, &tty) < 0){
|
|
perror("Ioctl");
|
|
quit(2);
|
|
}
|
|
if(write(comfd, &cmd, 1) < 1){
|
|
perror("Can't write to port");
|
|
quit(2);
|
|
}
|
|
}
|
|
|
|
void con_sig(int rb){
|
|
if(rb < 1) return;
|
|
if(rb == 'q') quit(0); // q == exit
|
|
else if(rb == 'H'){ // this program help
|
|
help();
|
|
return;
|
|
}else if(rb == 'D'){ // change device
|
|
int N;
|
|
printf("device number: ");
|
|
tcsetattr(STDIN_FILENO, TCSANOW, &oldt); // return terminal to previous state
|
|
scanf("%d", &N);
|
|
tcsetattr(STDIN_FILENO, TCSANOW, &newt); // and again - no echo
|
|
if(N < 0 || N > 255){
|
|
printf("\nWRONG DEVNUM!\n");
|
|
return;
|
|
}
|
|
send_with_9((unsigned char)N, 1); // send marked byte with new device number
|
|
return;
|
|
}
|
|
send_with_9((unsigned char)rb, 0); // just send command with zero nineth bit
|
|
}
|
|
|
|
/**
|
|
* Get integer value from buffer
|
|
* @param buff (i) - buffer with int
|
|
* @param len - length of data in buffer (could be 2 or 4)
|
|
* @return
|
|
*/
|
|
uint32_t get_int(uint8_t *buff, size_t len){
|
|
if(len != 2 && len != 4){
|
|
fprintf(stdout, "Bad data length!\n");
|
|
return 0xffffffff;
|
|
}
|
|
uint32_t data = 0;
|
|
uint8_t *i8 = (uint8_t*) &data;
|
|
if(len == 2) memcpy(i8, buff, 2);
|
|
else memcpy(i8, buff, 4);
|
|
return data;
|
|
}
|
|
|
|
int main(int argc, char *argv[]){
|
|
int rb;
|
|
uint8_t buff[128];
|
|
size_t L;
|
|
if(argc == 2){
|
|
fout = fopen(argv[1], "a");
|
|
if(!fout){
|
|
perror("Can't open output file");
|
|
exit(-1);
|
|
}
|
|
setbuf(fout, NULL);
|
|
}
|
|
tty_init();
|
|
signal(SIGTERM, quit); // kill (-15)
|
|
signal(SIGINT, quit); // ctrl+C
|
|
signal(SIGQUIT, SIG_IGN); // ctrl+\ .
|
|
signal(SIGTSTP, SIG_IGN); // ctrl+Z
|
|
setbuf(stdout, NULL);
|
|
t0 = dtime();
|
|
while(1){
|
|
rb = read_console();
|
|
if(rb > 0) con_sig(rb);
|
|
L = read_tty(buff, 127);
|
|
if(L){
|
|
buff[L] = 0;
|
|
printf("%s", buff);
|
|
if(fout) fprintf(fout, "%zd\t%s\n", time(NULL), buff);
|
|
}
|
|
}
|
|
}
|