/**********************************************************************
 * Send.c: File transmission routines for xprzmodem.library;
 * Original Version 2.10, 12 February 1991, by Rick Huebner.
 * Based closely on Chuck Forsberg's sz.c example ZModem code,
 * but too pervasively modified to even think of detailing the changes.
 * Released to the Public Domain; do as you like with this code.
 *
 * Version 2.50, 15 November 1991, CRC-32 additions by William M. Perkins.
 * Version 2.51 29, January 1992, RX_timout fix by John Tillema
 * Version 2.52   6 March 1992, Very minor fix with compiled 020 library
 *               by William M. Perkins.
 *
 **********************************************************************/

/*#include <proto/all.h>
#include <exec/types.h>
#include <ctype.h>
#include <stdio.h>*/

#include <string.h>

#include "xproto.h"
#include "zmodem.h"
#include "xprzmodem.h"

#ifdef DEBUGLOG
extern void *DebugLog;
#endif

/* Proto für Inlines */

static short getzrxinit(struct Vars *v);
static void  sendbatch(struct Vars *v);
static short sendone(struct Vars *v);
static short sendname(struct Vars *v);
static short zsendfile(struct Vars *v, short blen);
static short zsendfdata(struct Vars *v);
static short getinsync(struct Vars *v);
static void  saybibi(struct Vars *v);


/**********************************************************
 *	long XProtocolSend(struct XPR_IO *xio)
 *
 * Main file transmission routine; called by comm program
 **********************************************************/
long XProtocolSend(struct XPR_IO *xio)
{
   struct Vars *v;
   short err;

   /* Perform common setup and initializations */
   if (! (v = setup(xio)) )
      return XPRS_FAILURE;

/*  was 600, set to 300 to fix so it uploads correctly */
   v->Rxtimeout = 300;
   v->Wantfcs32 = TRUE;
   v->Rxflags = 0;

   /* Transfer the files */  
   zmputs(v, "rz\r");
   stohdr(v, 0L);
   zshhdr(v, ZRQINIT);
   sendbuf(v);
   if (getzrxinit(v) == ERROR) {
	err=TRUE;
      upderr(v, "Senden abgebrochen oder Timeout!");
   }
   else
      sendbatch(v);

   /* Clean up and return */
   if (err = v->Errcnt) {
	err=TRUE;
      upderr(v, "Datei(en) wegen Übertragunsfehlern übersprungen!");
   }
   else
      updmsg(v, "Übertragung erfolgreich beendet.");
   FreeMem(v, (long) sizeof(struct Vars));
  
   return (err) ? XPRS_FAILURE : XPRS_SUCCESS;
   }	/* End of long XProtocolSend() */

/**********************************************************
 *	short getzrxinit(struct Vars *v)
 *
 * Negotiate with receiver to start a file transfer
 **********************************************************/
static short getzrxinit(struct Vars *v)
{
   short n;

   for (n = v->ErrorLimit; --n >= 0; )
   {
      /* Check for abort from comm program */
      if (xpr_chkabort())
         return ERROR;
      switch (zgethdr(v))
      {
      case ZCHALLENGE:        /* Echo receiver's challenge number */
         stohdr(v, v->Rxpos);
         zshhdr(v, ZACK);
         sendbuf(v);
         continue;
      case ZCOMMAND:          /* They didn't see our ZRQINIT; try again */
         stohdr(v, 0L);
         zshhdr(v, ZRQINIT);
         sendbuf(v);
         continue;
      case ZRINIT:            /* Receiver ready; get transfer parameters */
	 v->Rxflags = 0xFF & v->Rxhdr[ZF0];
	 v->Txfcs32 = (v->Wantfcs32 && (v->Rxflags & CANFC32));
         v->Rxbuflen = ((USHORT) v->Rxhdr[ZP1] << 8) | v->Rxhdr[ZP0];
#ifdef DEBUGLOG
         mysprintf(v->Msgbuf, "Txfcs32=%ld Rxbuflen=%ld Tframlen=%ld\n",
	    (long) v->Txfcs32, (long) v->Rxbuflen, (long) v->Tframlen);
         dlog(v, v->Msgbuf);
#endif
         /* Use shortest of the two side's max frame lengths */
         if (v->Tframlen && (! v->Rxbuflen || v->Tframlen < v->Rxbuflen))
            v->Rxbuflen = v->Tframlen;
#ifdef DEBUGLOG
         mysprintf(v->Msgbuf, "Rxbuflen=%ld\n", (long) v->Rxbuflen);
         dlog(v, v->Msgbuf);
#endif
         return OK;
      case ZCAN:
      case RCDO:
      case TIMEOUT:
         upderr(v, v->Msgbuf);
         return ERROR;
      case ZRQINIT:
         if (v->Rxhdr[ZF0] == ZCOMMAND)
	    continue;
         /* fallthrough... */
      default:
         zshhdr(v, ZNAK);
         sendbuf(v);
         continue;
         }
      }
   return ERROR;
   }	/* End of short getzrxinit() */

/**********************************************************
 *	void sendbatch(struct Vars *v)
 *
 * Send a batch of files
 **********************************************************/
static void sendbatch(struct Vars *v)
{
   UBYTE single, done = FALSE;
   long fstate;

     single = FALSE;
     fstate = xpr_ffirst(v->Filename, v->io.xpr_filename);
      if (! fstate)
      {
         upderr(v, "Keine Dateien gefunden.");
         return;
         }

   /* If using templates, keep getting names & sending until done */
   while (! done)
   {
      if (sendone(v) == ERROR)
	 return;
      if (single)
	 break;
      fstate = xpr_fnext(fstate, v->Filename, v->io.xpr_filename);
      done = ! fstate;
      }

   /* End batch and return; if we never got started, just cancel receiver */
   if (v->Filcnt)
      saybibi(v);
   else
      canit(v);
   }	/* End of void sendbatch() */

/**********************************************************
 *	short sendone(struct Vars *v)
 *
 * Send the file named in v->Filename
 **********************************************************/
static short sendone(struct Vars *v)
{
   struct SetupVars *sv;

#ifdef DEBUGLOG
   mysprintf(v->Msgbuf, "*** Sending %s\n", v->Filename);
   dlog(v, v->Msgbuf);
#endif

   /* Display name of file being sent for user */
   v->xpru.xpru_updatemask = XPRU_FILENAME;
   v->xpru.xpru_filename = v->Filename;
   xpr_update(&v->xpru);

   /* Set text/binary mode according to options before opening file */
   set_textmode(v);

   /* Open the file, if possible */
   if (! (v->File = bfopen(v, "r")))
   {
      ++v->Errcnt;
      upderr(v, "Kann Datei nicht öffnen (wird übersprungen)");
      return OK;      /* pass over it, there may be others */
      }
   ++v->Filcnt;
   getsystime(&v->Starttime);

   /* Kick off the file transfer */
   sv = (void *) v->io.xpr_data;
   switch (sendname(v))
   {
   case ERROR:
      ++v->Errcnt;
      return ERROR;
   case OK:
      bfclose(v);
      break;
      }
   return OK;
   }	/* End of short sendone() */

/**********************************************************
 *	short sendname(struct Vars *v)
 *
 * Build file info block consisting of file name, length,
 * time, and mode
 **********************************************************/
static short sendname(struct Vars *v)
{
   UBYTE *p, *q, buff [32];

   /* Initialize comm program transfer status display */
   v->Fsize = xpr_finfo(v->Filename, 1L);
   v->xpru.xpru_updatemask = XPRU_PROTOCOL | XPRU_FILESIZE | XPRU_MSG
      | XPRU_BLOCKS | XPRU_ERRORS | XPRU_TIMEOUTS | XPRU_BLOCKCHECK
      | XPRU_BYTES | XPRU_EXPECTTIME | XPRU_ELAPSEDTIME | XPRU_DATARATE;
   v->xpru.xpru_protocol = "ZModem";
   v->xpru.xpru_filesize = v->Fsize;
   v->xpru.xpru_msg = "Sende Datei...";
   v->xpru.xpru_blocks = v->xpru.xpru_errors = v->xpru.xpru_timeouts = 0;
   v->xpru.xpru_blockcheck = v->Crc32t ? "CRC-32" : "CRC-16";
   v->xpru.xpru_bytes = v->Strtpos = 0;
   update_rate(v);
   xpr_update(&v->xpru);

   /*sv = (void *) xpr_data;*/
   /* extract outgoing file name without directory path */
   for (p = v->Filename, q = v->Pktbuf ; *p; ++p, ++q)
     if ((*q = *p) == '/' || *q == ':')
       q = v->Pktbuf - 1;
      *q = '\0';
      p = ++q;

   /* Zero out remainder of file info packet */
   memset(p, 0, sizeof(v->Pktbuf) - (p - v->Pktbuf));

   /* Store file size, timestamp, and mode in info packet */
   /*
    * XPR spec doesn't provide a way to get the file timestamp or file mode,
    * so we'll just fake it with the current time and a dummy 0.
    */
   stcl_o(buff, getsystime(NULL) + UnixTimeOffset);
   /* amiga.lib mysprintf() can't do %lo format, so we do it the hard way */
   /* Yes, octal; ZModem was originally done on Unix, and they like octal there */
   mysprintf(p, "%ld %s 0", (v->Fsize < 0) ? 0L : v->Fsize,buff);

   /* Send filename packet */
   return zsendfile(v, (short) (p - v->Pktbuf + strlen(p) + 1));
   }	/* End of short sendname() */

/**********************************************************
 *	short zsendfile(struct Vars *v, short blen)
 *
 * Send the filename packet and see if receiver will accept
 * file
 **********************************************************/
static short zsendfile(struct Vars *v, short blen)
{
   short c;

   while (TRUE)
   {
      v->Txhdr[ZF0] = v->Lzconv; /* Text or Binary mode; from config string */
      v->Txhdr[ZF1] = LZMANAG;   /* Default file management mode */
      v->Txhdr[ZF2] = LZTRANS;   /* Default file transport mode */
      v->Txhdr[ZF3] = 0;
      zsbhdr(v, ZFILE);
      zsdata(v, blen, ZCRCW);
      sendbuf(v);
again:
      /* Check for abort from comm program */
      if (xpr_chkabort())
      {
         bfclose(v);
         return ERROR;
         }
      switch (c = zgethdr(v))
      {
      case ZRINIT:
         goto again;
      case ZCAN:
      case ZCRC:
      case RCDO:
      case TIMEOUT:
      case ZABORT:
      case ZFIN:
         upderr(v, v->Msgbuf);
         return ERROR;
      case ZSKIP:             /* Receiver doesn't want this one */
         upderr(v, "SKIP-Paket empfangen!");
         bfclose(v);
         return c;
      case ZRPOS:             /* Receiver wants it; this is starting position */
         bfseek(v, v->Rxpos);
         v->Strtpos = v->Txpos = v->Rxpos;
	       xpr_sflush();
         v->Modemcount = 0;
         return zsendfdata(v);
         }
      }
   }	/* End of short zsendfile() */

/**********************************************************
 *	short zsendfdata(struct Vars *v)
 *
 * Send the file data
 **********************************************************/

extern int zmodemmode;

static short zsendfdata(struct Vars *v)
{
   short c, e, blklen, goodbytes = 0;
   USHORT framelen, maxblklen, goodneeded = 512;

   /* Figure out max data packet size to send */
   maxblklen = KSIZE;
   if (v->Rxbuflen && maxblklen > v->Rxbuflen)
      maxblklen = v->Rxbuflen;
   blklen = (v->Baud < 1200) ? 256 : KSIZE;
   if (blklen > maxblklen)
      blklen = maxblklen;

	if( !zmodemmode )
	{
		maxblklen = min( 1024, maxblklen );
		blklen = min( 1024, blklen );
	}

#ifdef DEBUGLOG
   mysprintf(v->Msgbuf, "Rxbuflen=%ld blklen=%ld\n", (long) v->Rxbuflen,
      (long) blklen);
   dlog(v, v->Msgbuf);
#endif

   /* If an interruption happened, handle it; else keep sending data */
somemore:
   while (char_avail(v))
   {
      /* Check for another incoming packet while discarding line noise */
      switch (readock(v, 1))
      {
      case CAN:
      case RCDO:
      case ZPAD:
         break;
      default:
         continue;
         }
waitack:
#ifdef DEBUGLOG
      dlog(v, "--- At waitack\n");
#endif
      switch (c = getinsync(v))
      {
      default:
         upderr(v, "Übertraung abgebrochen.");
         bfclose(v);
         return ERROR;
      case ZSKIP:  /* Receiver changed its mind and wants to skip the file */
         return c;
      case ZACK:   /* ACK at end of frame; resume sending data */
         break;
      case ZRPOS:  /* An error; resend data from last good point */
         blklen >>= 2;
         if (blklen < MINBLOCK)
	    blklen = MINBLOCK;
         if (goodneeded < MAXGOODNEEDED)
	    goodneeded <<= 1;
         v->xpru.xpru_updatemask = XPRU_ERRORS;
         ++v->xpru.xpru_errors;
         xpr_update(&v->xpru);
         break;
      case ZRINIT:
         updmsg(v, "Übertragung erfolgreich beendet.");
         return OK;
         }
      }

   /* Transmit ZDATA frame header */
   framelen = v->Rxbuflen;
   stohdr(v, v->Txpos);
   zsbhdr(v, ZDATA);

   /* Keep sending data packets until finished or interrupted */
   do
   {
      /* Read next chunk of file data */
      c = bfread(v, v->Pktbuf, (long) blklen);

      /* Figure out how to handle this data packet */
      if (c < blklen)
         e = ZCRCE;  /* If end of file, this is last data packet */
      else if (v->Rxbuflen && (framelen -= c) <= 0)
         e = ZCRCW;  /* If end of frame, ask for ACK */
      else
         e = ZCRCG;  /* Else tell receiver to expect more data packets */

      zsdata(v, c, e);  /* Send the packet */
      sendbuf(v);

      /* Update comm program status display */
      v->xpru.xpru_updatemask = XPRU_BLOCKS | XPRU_BLOCKSIZE | XPRU_BYTES
         | XPRU_EXPECTTIME | XPRU_ELAPSEDTIME | XPRU_DATARATE
	 | XPRU_BLOCKCHECK;
      ++v->xpru.xpru_blocks;
      v->xpru.xpru_blocksize = c;
      v->xpru.xpru_blockcheck = v->Crc32t ? "CRC-32" : "CRC-16";
      v->xpru.xpru_bytes = v->Txpos += c;
      update_rate(v);
      xpr_update(&v->xpru);

      /*
       * If we've been sending smaller than normal packets, see if it's
       * time to bump the packet size up a notch yet
       */
      if (blklen < maxblklen && (goodbytes += c) >= goodneeded)
      {
         blklen <<= 1;
         if (blklen > maxblklen)
	    blklen = maxblklen;
         goodbytes = 0;
#ifdef DEBUGLOG
         mysprintf(v->Msgbuf, "Bumping packet size to %ld at %ld\n",
	    (long) blklen, v->Txpos);
         dlog(v, v->Msgbuf);
#endif
         }

      /* Give comm program its timeslice if it needs one */
      /* Check for abort from comm program */
      if (xpr_chkabort())
	 goto aborted;
      /* If this was last packet in frame, go wait for ACK from receiver */
      if (e == ZCRCW)
	 goto waitack;

      /*
       * Check if receiver trying to interrupt us; look for incoming packet
       * while discarding line noise
       */
      while (char_avail(v))
      {
         switch (readock(v, 1))
	 {
         case CAN:
         case RCDO:
         case ZPAD:
            /* Interruption detected; stop sending and process complaint */
#ifdef DEBUGLOG
            dlog(v, "--- Interrupted send\n");
#endif
            zsdata(v, 0, ZCRCE);
            sendbuf(v);
            goto waitack;
            }
         }
      }
   while (e == ZCRCG);  /* If no interruption, keep sending data packets */

   /* Done sending file data; send EOF and wait for receiver to acknowledge */
   while (TRUE)
   {
      updmsg(v, "Sende EOF.");
      stohdr(v, v->Txpos);
      zsbhdr(v, ZEOF);
      sendbuf(v);
      switch (c = getinsync(v))
      {
      case ZACK:
         continue;
      case ZRPOS:
         goto somemore;
      case ZRINIT:
         updmsg(v, "EOF bestätigt.");
         ++v->Starttime.tv_secs;
         update_rate(v);
         v->xpru.xpru_updatemask = XPRU_EXPECTTIME | XPRU_ELAPSEDTIME
	    | XPRU_DATARATE;
         xpr_update(&v->xpru);
         return OK;
      case ZSKIP:
         return c;
      default:
aborted:
	 upderr(v, "Übertragung abgebrochen!");
         bfclose(v);
         return ERROR;
         }
      }
   }	/* End of short zsendfdata() */

/**********************************************************
 *	short getinsync(struct Vars *v)
 *
 * Respond to receiver's complaint, get back in sync with
 * receiver
 **********************************************************/
static short getinsync(struct Vars *v)
{
   short c;

   while (TRUE)
   {
#ifdef DEBUGLOG
      dlog(v, "--- At getinsync\n");
#endif
      c = zgethdr(v);
      xpr_sflush();
      v->Modemcount = 0;
      switch (c)
      {
      case ZCAN:
      case ZABORT:
      case ZFIN:
      case RCDO:
      case TIMEOUT:
         upderr(v, v->Msgbuf);
         return ERROR;
      case ZRPOS:
         bfseek(v, v->Rxpos);
         v->Txpos = v->Rxpos;
         mysprintf(v->Msgbuf, "Zurück ab Offset %ld", v->Txpos);
         upderr(v, v->Msgbuf);
         return c;
      case ZSKIP:
         upderr(v, "SKIP-Paket empfangen!");
         /* fallthrough... */
      case ZRINIT:
         bfclose(v);
         /* fallthrough... */
      case ZACK:
         return c;
      default:
         zsbhdr(v, ZNAK);
         sendbuf(v);
         continue;
         }
      }
   }	/* End of short getinsync() */

/**********************************************************
 *	void saybibi(struct Vars *v)
 *
 * End of batch transmission; disengage cleanly from receiver
 **********************************************************/
static void saybibi(struct Vars *v)
{
   while (TRUE)
   {
      stohdr(v, 0L);
      zsbhdr(v, ZFIN);
      sendbuf(v);
      switch (zgethdr(v))
      {
      case ZFIN:
         sendline(v, 'O');
         sendline(v, 'O');
         sendbuf(v);
         /* fallthrough... */
      case ZCAN:
      case RCDO:
      case TIMEOUT:
         return;
         }
      }
   }	/* End of void saybibi() */
/* End of Send.c source */