Subversion Repositories or1k

//####COPYRIGHTBEGIN####

//                                                                          

// ----------------------------------------------------------------------------

// Copyright (C) 1998, 1999, 2000 Red Hat, Inc.

//

// This program is part of the eCos host tools.

//

// 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., 

// 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

//

// ----------------------------------------------------------------------------

//                                                                          

//####COPYRIGHTEND####

//#####DESCRIPTIONBEGIN####

//

// Author(s):     sdf

// Contributors:  sdf

// Date:          1999-04-01

// Description:   Holds information related to target reset

// Usage:

//

//####DESCRIPTIONEND####

#include "eCosStd.h"

#include "eCosThreadUtils.h"

#include "eCosTrace.h"

#include "ResetAttributes.h"

const CResetAttributes CResetAttributes::NoReset;

CResetAttributes::CResetAttributes(LPCTSTR psz) :

  // Default values:

  m_nDelay(1000),

  m_nReadTimeout(10*1000),

  m_nBaud(38400)

{

  // Remove spaces

  while(*psz){

    if(!_istspace(*psz)){

      m_str+=*psz;

    }

    psz++;

  }

}

/*

LPCTSTR CResetAttributes::Image(int nErr)

{

  switch(nErr){

    case RESET_OK:

      return _T("RESET_OK");

      break;

    case RESET_ILLEGAL_DEVICE_CODE:

      return _T("Illegal device code");

      break;

    case RESET_NO_REPLY:

      return _T("No reply from reset unit");

      break;

    case RESET_BAD_CHECKSUM:

      return _T("Bad checksum");

      break;

    case RESET_BAD_ACK:

      return _T("Bad ack");

      break;

    default:

      return _T("Unknown reset error");

      break;

  }

}

*/

void CResetAttributes::SuckThreadFunc()

{

  m_strResetOutput=_T("");

  // Board has apparently been powered on.  Suck initial output.

  ResetLog(String::SFormat(_T("Reading board startup output from %s with timeout of %d seconds..."),(LPCTSTR)m_strAuxPort,m_nReadTimeout/1000));

  enum {BUFSIZE=512};

  TCHAR buf[1+BUFSIZE];

  memset(buf,0,BUFSIZE); // safety for string functions in IsValidReset

  do {

    unsigned int dwRead=0;

    // We are working in non-blocking mode

    if(m_Socket.Ok()){

      if(!m_Socket.Peek(dwRead)||!m_Socket.recv(buf,MIN(dwRead,BUFSIZE))){

        break;

      }

    } else if (!m_Serial.Read(buf,BUFSIZE,dwRead)){

      m_Serial.ClearError();

      continue;

    }

    if(dwRead>0){

      buf[dwRead]=_TCHAR('\0');

      // Remove unprintable characters

      String str;

      for(const TCHAR *t=buf;*t;t++){

        if(_istprint(*t)){

          str+=*t;

        }

      }

      if(m_pfnReset){

        ENTERCRITICAL;

        m_pfnReset(m_pfnResetparam,str);

        LEAVECRITICAL;

      }

      ResetLog(str);

      m_strResetOutput+=str;

      if(IsValidReset()){

        break;

      }

//    } else { // Nothing read

//      CeCosThreadUtils::Sleep(50);

    }

  } while (0==m_tResetOccurred || Now()-m_tResetOccurred<m_nReadTimeout);

  if(0==m_strResetOutput.size()){

    ResetLog(_T("No response from board"));

  } else {

    if(m_pfnReset){

      ENTERCRITICAL;

      m_pfnReset(m_pfnResetparam,_T("\n"));

      LEAVECRITICAL;

    }

    TRACE(_T("%s"),(LPCTSTR)m_strResetOutput);

  }

}

bool CResetAttributes::Reset(Action action,bool bCheckOutput)

{

  m_tResetOccurred=0;

  m_strResetOutput=_T("");

  bool rc=false;

  CeCosSocket sock;

  String strStatus;

  strStatus.Format(_T("Reset target using %s %s port=%s(%d) read timeout=%d delay=%d"),

    (LPCTSTR)m_strHostPort,(LPCTSTR)m_strControl,

    (LPCTSTR)m_strAuxPort, m_nBaud, m_nReadTimeout, m_nDelay);

  if(bCheckOutput){

    strStatus+=_T(" expect(");

    for(unsigned int i=0;i<m_arValidResetStrings.size();i++){

      if(i>0){

        strStatus+=_TCHAR(',');

      }

      strStatus+=m_arValidResetStrings[i];

    }

    strStatus+=_T(")");

  }

  ResetLog(strStatus);

  // Open up communication to port whence we read the board startup

  bool bThreadDone=false;

  bCheckOutput&=(m_strAuxPort.size()>0);

  if(bCheckOutput){

    TRACE(_T("Opening %s\n"),(LPCTSTR)m_strAuxPort);

    if(CeCosSocket::IsLegalHostPort(m_strAuxPort)){

      // tcp/ip port

      if(!m_Socket.Connect(m_strAuxPort,m_nReadTimeout)){

        ResetLog(String::SFormat(_T("Failed to open %s - %s"),(LPCTSTR)m_strAuxPort,(LPCTSTR)m_Socket.SocketErrString()));

        return false;

      }

    } else {

      // Comms device

      m_Serial.SetBlockingReads(false);

      if(m_Serial.Open(m_strAuxPort,m_nBaud)){

        m_Serial.Flush();

      } else {

        ResetLog(String::SFormat(_T("Failed to open comms port %s - %s"),(LPCTSTR)m_strAuxPort,(LPCTSTR)m_Serial.ErrString()));

        return false;

      }

    }

    CeCosThreadUtils::RunThread(SSuckThreadFunc,this,&bThreadDone,_T("SSuckThreadFunc"));

  } else {

    ResetLog(_T("[not checking output]"));

  }

  // This will be true if we need to talk to a reset server, false to talk down a local port

  bool bRemote=CeCosSocket::IsLegalHostPort(m_strHostPort);

  if(bRemote){

    if(sock.Connect(m_strHostPort,10*1000)){

      // Write the message to the socket

      int nDelay=(action==ON_OFF || action==OFF_ON)?m_nDelay:0;

      TRACE(_T("-Control=%s -Action=%d -Delay=%d"),(LPCTSTR)m_strControl,action,nDelay);

      if(sock.sendString(String::SFormat(_T("-Control=%s -Action=%d -Delay=%d"),(LPCTSTR)m_strControl,action,nDelay),_T("Reset control codes"), 10*1000)){

        // Wait for an acknowledgement

        String strResponse;

        if(sock.recvString(strResponse, _T("Response"), nDelay+20*1000)){

          rc=(0==strResponse.size());

          if(!rc && m_pfnReset){

            ResetLog(String::SFormat(_T("Reset server reports error '%s'"),(LPCTSTR)strResponse));

          }

        } else {

          ResetLog(String::SFormat(_T("Failed to read response from reset server %s - %s"),(LPCTSTR)m_strHostPort,(LPCTSTR)sock.SocketErrString()));

        }

      } else {

        ResetLog(String::SFormat(_T("Failed to contact reset server %s - %s"),(LPCTSTR)m_strHostPort,(LPCTSTR)sock.SocketErrString()));

      }

      m_tResetOccurred=Now();

      if(bCheckOutput){

        if(!rc){

          // force thread to time out

          m_tResetOccurred=Now()-m_nReadTimeout;

        }

        CeCosThreadUtils::WaitFor(bThreadDone); // do not apply a timeout - the thread has one

        rc=IsValidReset();

        ResetLog(rc?_T("Reset output valid"):_T("Reset output INVALID"));

      }

    } else {

      ResetLog(String::SFormat(_T("Failed to contact reset server %s - %s"),(LPCTSTR)m_strHostPort,(LPCTSTR)sock.SocketErrString()));

    }

  } else {

    // Sending something locally

    m_tResetOccurred=Now();

    unsigned int nWritten;

    m_Serial.Write((void *)(LPCTSTR)m_strControl,1,nWritten);

    if(bCheckOutput){

      CeCosThreadUtils::WaitFor(bThreadDone);  // do not apply a timeout - the thread has one

      rc=IsValidReset();

      ResetLog(rc?_T("Reset output valid"):_T("Reset output INVALID"));

    }

  }

  if(m_Socket.Ok()){

    m_Socket.Close();

  } else {

    m_Serial.Close();

  }

  return rc && bCheckOutput;

}

// We expect to be passed a string that starts with "xxx(yyy)"

// and the task is to extract xxx into strID and yyy into strArg

const TCHAR *CResetAttributes::GetIdAndArg (LPCTSTR psz,String &strID,String &strArg)

{

  const TCHAR *cEnd=_tcschr(psz,_TCHAR('('));

  if(cEnd){

    strID=String(psz,cEnd-psz);

    int nNest=0;

    for(const TCHAR *c=cEnd;*c;c++){

      if(_TCHAR('(')==*c){

        nNest++;

      } else if(_TCHAR(')')==*c){

        nNest--;

        if(0==nNest){

          strArg=String(cEnd+1,c-(cEnd+1));

          return c+1;

        }

      }

    }

    assert(false);

  }

  return 0;

}

// Do the reset

CResetAttributes::ResetResult CResetAttributes::Reset (LogFunc *pfnLog, void *pfnLogparam,bool bCheckOnly)

{

  m_pfnReset=pfnLog;

  m_pfnResetparam=pfnLogparam;

  // First we clean up the reset string so as to make subsequent parsing less complicated.

  // Spaces have already been removed in the ctor

  // Check paren matching:

  int nNest=0;

  for(const TCHAR *c=m_str;*c;c++){

    if(_TCHAR('(')==*c){

      nNest++;

    } else if(_TCHAR(')')==*c){

      nNest--;

      if(nNest<0){

        ResetLog(_T("Too many right parentheses"));

        return INVALID_STRING;

      }

    }

  }

  if(nNest>0){

    ResetLog(_T("Too many left parentheses"));

    return INVALID_STRING;

  }

  return Parse(m_str,bCheckOnly);

}

// This function parses the reset string, whose form is something like:

//     expect($T05) 3(off(ginga:5000,a1) delay(2000) on(ginga:5000,a1,com1,38400,10000))

// It is recursive (which is another reason elementary syntax checking was carried out above)

// and calls itself to perform repeats [e.g. 3(...)]

CResetAttributes::ResetResult CResetAttributes::Parse (LPCTSTR psz,bool bCheckOnly)

{

  enum {ARGSEP=_TCHAR(',')};

  bool bCheck=false;

  for(const TCHAR *c=psz;*c;){

    String strID,strArg;

    c=GetIdAndArg(c,strID,strArg);

    if(0==c){

      ResetLog(_T("Invalid reset string"));

      return INVALID_STRING;

    }

    if(isdigit(*(LPCTSTR)strID)){

      // Process a repeat-until-reset.  Syntax is n(resetstring)

      int nRepeat=_ttoi(strID);

      if(0==nRepeat){

        ResetLog(_T("Invalid reset string"));

        return INVALID_STRING;

      }

      if(bCheckOnly){

        return Parse(strArg,true);

      } else {

        while(nRepeat--){

          ResetResult r=Parse(strArg);

          if(RESET_OK==r||INVALID_STRING==r){

            return r;

          }

        }

      }

    } else if (_T("expect")==strID) {

      //   Expected string(s).  e.g. expect(str1,str2,...).

      strArg.Chop(m_arValidResetStrings,ARGSEP,true);

    } else if (_T("port")==strID) {

      // Port information.      e.g. port(com1,38400,1000)

      // This information will apply to all subsequent actions until overwritten.

      // Specifically args are:

      //   0. Port

      //   1. Baud

      //   2. Read timeout

      StringArray ar;

      int nArgs=strArg.Chop(ar,ARGSEP,true);

      if(nArgs>0){

        m_strAuxPort=ar[0];

      }

      if(nArgs>1){

        m_nBaud=_ttoi(ar[1]);

      }

      if(nArgs>2){

        m_nReadTimeout=_ttoi(ar[2]);

      }

    } else if (_T("off")==strID || _T("on")==strID || _T("on_off")==strID || _T("off_on")==strID) {

      // Action information.      e.g. off(ginga:500,A4,com1,38400,10000,1000)

      // Specifically args are:

      //   0. Reset host:port

      //   1. Control string

      //   2. Port

      //   3. Baud

      //   4. Read timeout

      //   5. Delay

      StringArray ar;

      int nArgs=strArg.Chop(ar,ARGSEP,true);

      if(nArgs>0){

        m_strHostPort=ar[0];

      }

      if(nArgs>1){

        m_strControl=ar[1];

      }

      if(nArgs>2){

        m_strAuxPort=ar[2];

      }

      if(nArgs>3){

        m_nBaud=_ttoi(ar[3]);

      }

      if(nArgs>4){

        m_nReadTimeout=_ttoi(ar[4]);

      }

      if(nArgs>5){

        m_nDelay=_ttoi(ar[5]);

      }

      if(0==m_strHostPort.size()){

        ResetLog(_T("Failed to specify reset host:port"));

        return INVALID_STRING;

      }

      Action action=ON; // prevent compiler warning

      if(_T("on")==strID){

        action=ON;

      } else if(_T("off")==strID){

        action=OFF;

      } else if(_T("on_off")==strID){

        action=ON_OFF;

      } else if(_T("off_on")==strID){

        action=OFF_ON;

      }

      if(!bCheckOnly && Reset(action,bCheck||action==ON_OFF||action==OFF_ON)){

        return RESET_OK;

      }

      bCheck ^= 1;

    } else if (_T("delay")==strID) {

      // Delay for a given time right now.      e.g. delay(1000)

      // Specifically args are:

      //   0. msec to delay

      TRACE(_T("CeCosThreadUtils::Sleep %d\n"),_ttoi(strArg));

      if(!bCheckOnly){

        CeCosThreadUtils::Sleep(_ttoi(strArg));

      }

    } else {

      ResetLog(String::SFormat(_T("Unrecognized command '%s'"),(LPCTSTR)strID));

      return INVALID_STRING;

    }

  }

  ResetLog(_T("Target reset not verified"));

  return NOT_RESET;

}

// Log some output to the reset log function.

void CResetAttributes::ResetLog(LPCTSTR psz)

{

  if(m_pfnReset){

    ENTERCRITICAL;

    m_pfnReset(m_pfnResetparam,String::SFormat(_T("%s >>> %s\n"),(LPCTSTR)CeCosTrace::Timestamp(),psz));

    TRACE(_T("%s"),psz);

    LEAVECRITICAL;

  }

}

bool CResetAttributes::IsValidReset()

{

  unsigned int n=0;

  ENTERCRITICAL;

  for(int i=m_arValidResetStrings.size()-1;i>=0;--i){

    if(_tcsstr(m_strResetOutput,m_arValidResetStrings[i])){

      n++;

    }

  }

  LEAVECRITICAL;

  return n==m_arValidResetStrings.size();

}