1. 程式人生 > >linux下串列埠控制

linux下串列埠控制

/*  本程式符合GPL條約
 *  Beneboy 2003-5-16
*/
#include <stdio.h>              // printf
#include <fcntl.h>              // open
#include <string.h>             // bzero
#include <stdlib.h>             // exit
#include <sys/times.h>          // times
#include <sys/types.h>          // pid_t
#include <termios.h>  //termios, tcgetattr(), tcsetattr()
#include <unistd.h>
#include <sys/ioctl.h>          // ioctl
#include "MyCom.h"


#define TTY_DEV "/dev/ttyS" //埠路徑
//接收超時
#define TIMEOUT_SEC(buflen,baud) (buflen*20/baud+2)
#define TIMEOUT_USEC 0


/*******************************************
 * 獲得埠名稱
********************************************/
char *get_ptty(pportinfo_t pportinfo)
{
 char *ptty;

 switch(pportinfo->tty){
  case '0':{
   ptty = TTY_DEV"0";
  }break;
  case '1':{
   ptty = TTY_DEV"1";
  }break;
  case '2':{
   ptty = TTY_DEV"2";
  }break;
 }
 return(ptty);
}

/*******************************************
 * 波特率轉化轉換函式
********************************************/
int convbaud(unsigned long int baudrate)
{
 switch(baudrate){
  case 2400:
   return B2400;
  case 4800:
   return B4800;
  case 9600:
   return B9600;
  case 19200:
   return B19200;
  case 38400:
   return B38400;
  case 57600:
   return B57600;
  case 115200:
   return B115200;
  default:
   return B9600;
 }
}

/*******************************************
 * Setup comm attr
 * fdcom: 串列埠檔案描述符, pportinfo: 待設定的埠資訊s
 *
********************************************/
int PortSet(int fdcom, const pportinfo_t pportinfo)
{
 struct termios termios_old, termios_new;
 int  baudrate, tmp;
 char databit, stopbit, parity, fctl;

 bzero(&termios_old, sizeof(termios_old));
 bzero(&termios_new, sizeof(termios_new));
 cfmakeraw(&termios_new);
 tcgetattr(fdcom, &termios_old);   //get the serial port attributions

 /*------------設定埠屬性----------------*/

 //baudrates
 baudrate = convbaud(pportinfo -> baudrate);
 cfsetispeed(&termios_new, baudrate);  //填入串列埠輸入端波特率
 cfsetospeed(&termios_new, baudrate);  //填入串列埠輸出端波特率
 termios_new.c_cflag |= CLOCAL;   //控制模式, 保證程式不會成為埠的佔有者
 termios_new.c_cflag |= CREAD;   //控制模式, 使能埠讀取輸入的資料


 // 控制模式, flow control
 fctl = pportinfo-> fctl;
 switch(fctl){
  case '0':{
   termios_new.c_cflag &= ~CRTSCTS;  //no flow control
  }break;
  case '1':{
   termios_new.c_cflag |= CRTSCTS;   //hardware flow control
  }break;
  case '2':{
   termios_new.c_iflag |= IXON | IXOFF |IXANY; //software flow control
  }break;
 }

 //控制模式, data bits
 termios_new.c_cflag &= ~CSIZE;  //控制模式, 遮蔽字元大小位
 databit = pportinfo -> databit;
 switch(databit){
  case '5':
   termios_new.c_cflag |= CS5;
  case '6':
   termios_new.c_cflag |= CS6;
  case '7':
   termios_new.c_cflag |= CS7;
  default:
   termios_new.c_cflag |= CS8;
 }

 //控制模式 parity check
 parity = pportinfo -> parity;
 switch(parity){
  case '0':{
   termios_new.c_cflag &= ~PARENB;  //no parity check
  }break;
  case '1':{
   termios_new.c_cflag |= PARENB;  //odd check
   termios_new.c_cflag &= ~PARODD;
  }break;
  case '2':{
   termios_new.c_cflag |= PARENB;  //even check
   termios_new.c_cflag |= PARODD;
  }break;
 }

 //控制模式, stop bits
 stopbit = pportinfo -> stopbit;
 if(stopbit == '2'){
  termios_new.c_cflag |= CSTOPB; //2 stop bits
 }
 else{
  termios_new.c_cflag &= ~CSTOPB; //1 stop bits
 }

 //other attributions default
 termios_new.c_oflag &= ~OPOST;   //輸出模式, 原始資料輸出

 termios_new.c_cc[VMIN]  = 1;   //控制字元, 所要讀取字元的最小數量
 termios_new.c_cc[VTIME] = 1;   //控制字元, 讀取第一個字元的等待時間,  unit: (1/10)second

 tcflush(fdcom, TCIFLUSH);   //溢位的資料可以接收,但不讀
 tmp = tcsetattr(fdcom, TCSANOW, &termios_new); //設定新屬性, TCSANOW: 所由改變立即生效
 tcgetattr(fdcom, &termios_old);
 return(tmp);
}

/*******************************************
 * Open serial port
 * tty: 埠號 ttyS0, ttyS1, ....
 * 返回值為串列埠檔案描述符
********************************************/
int PortOpen(pportinfo_t pportinfo)
{
 int fdcom; //串列埠檔案描述符
 char *ptty;

 ptty = get_ptty(pportinfo);
 //fdcom = open(ptty, O_RDWR | O_NOCTTY | O_NONBLOCK | O_NDELAY);
 fdcom = open(ptty, O_RDWR | O_NOCTTY | O_NONBLOCK);

 return (fdcom);
}

/*******************************************
 * Close serial port
********************************************/
void PortClose(int fdcom)
{
 close(fdcom);
}

/********************************************
 * send data
 * fdcom: 串列埠描述符, data: 待發送資料, datalen: 資料長度
 * 返回實際傳送長度
*********************************************/
int PortSend(int fdcom, char *data, int datalen)
{
 int len = 0;

 len = write(fdcom, data, datalen); //實際寫入的長度
 if(len == datalen){
  return (len);
 }
 else{
  tcflush(fdcom, TCOFLUSH);
  return -1;
 }
}

/*******************************************
 * receive data
 * 返回實際讀入的位元組數
 *
********************************************/
int PortRecv(int fdcom, char *data, int datalen, int baudrate)
{
 int readlen, fs_sel;
 fd_set fs_read;
 struct timeval tv_timeout;

 FD_ZERO(&fs_read);
 FD_SET(fdcom, &fs_read);
 tv_timeout.tv_sec = TIMEOUT_SEC(datalen, baudrate);
 tv_timeout.tv_usec = TIMEOUT_USEC;

 fs_sel = select(fdcom+1, &fs_read, NULL, NULL, &tv_timeout);
 if(fs_sel){
  readlen = read(fdcom, data, datalen);
  return(readlen);
 }
 else{
  return(-1);
 }

 return (readlen);
}

//*************************Test*********************************
int main(int argc, char *argv[])
{
 int fdcom, i, SendLen, RecvLen;
 struct termios termios_cur;
 char RecvBuf[256];
        char writeBuf[256];
 portinfo_t portinfo ={
  '0',                          // print prompt after receiving
   9600,                       // baudrate: 9600
   '8',                          // databit: 8
   '0',                          // debug: off
   '0',                          // echo: off
   '2',                          // flow control: software
   '0',                          // default tty: COM1
   '0',                          // parity: none
   '1',                          // stopbit: 1
    0                           // reserved
 };

 if(argc != 2){
  printf("Usage: <type 0 -- send 1 -- receive>\n");
  printf("   eg:");
  printf("        MyPort 0");
  exit(-1);
 }

 fdcom = PortOpen(&portinfo);
 if(fdcom<0){
  printf("Error: open serial port error.\n");
  exit(1);
 }

 PortSet(fdcom, &portinfo);

        while(1)
 if(atoi(argv[1]) == 0)
        {
  //send data
  //for(i=0; i<1000; i++)
                {
                        //fgets(writeBuf,256,stdin);
                        scanf("%s",writeBuf);
   SendLen = PortSend(fdcom, writeBuf, strlen(writeBuf));
   if(SendLen>0)
                        {
    //printf("send data %s", writeBuf);
   }
   else
                        {
    printf("Error: send failed.\n");
   }

                        if(strncmp(writeBuf,"exit",4)==0)
                           break;
   sleep(1);
  }
  //PortClose(fdcom);
 }
 else
       {
  //for(;;)
                {
   RecvLen = PortRecv(fdcom, RecvBuf, 256, portinfo.baudrate);
   if(RecvLen>0)
                        {
    /*for(i=0; i<RecvLen; i++)
                                {
     printf("Receive data No %d is %x.\n", i, RecvBuf[i]);
    }//*/
                                RecvBuf[RecvLen] = '\0';
    printf("read data %s\n",RecvBuf);

                                if(strncmp(RecvBuf,"exit",4)==0)
                                    break;
   }
   else
                        {
    //printf("Error: receive error.\n");
   }
   sleep(1);
  }
 }
        PortClose(fdcom);
 return 0;
}