void YCancel(PFileVar fv, PYVar yv, PComVar cv)
{
// five cancels & five backspaces per spec
BYTE cancel[] = { CAN, CAN, CAN, CAN, CAN, BS, BS, BS, BS, BS };
YWrite(fv,yv,cv, (PCHAR)&cancel, sizeof(cancel));
yv->YMode = 0; // quit
}
void Y__write(int nArgs)
{
Symbol *keySymbols[1];
Symbol *stack= YGetKeywords(sp-nArgs+1, nArgs, wrtKeys, keySymbols);
IOStream *file= 0;
long address= 0;
int got_address= 0;
Symbol *object= 0;
Operand op;
StructDef *base;
char *type;
while (stack<=sp) {
if (!stack->ops) {
stack+= 2;
continue;
}
if (!file) {
file= YGetFile(stack);
if (file->history) file= file->history->child;
} else if (!got_address) {
got_address= 1;
address= YGetInteger(stack);
} else if (!object) {
object= stack;
} else {
object= 0;
}
stack++;
}
if (!object) YError("_write takes exactly three arguments");
sp->ops->FormOperand(object, &op);
if (!op.ops->isArray)
YError("third argument to _write must be array or scalar data");
if (YNotNil(keySymbols[0])) type= YGetString(keySymbols[0]);
else type= StructName(op.type.base);
if (!HashFind(&file->structTable, type, 0L))
YError("data type of third argument to _write undefined for this file");
base= file->structList[hashIndex];
if (op.type.base==&charStruct) {
/* special case type char, to have a way to do literal writes */
YcWrite(file, op.value, address, op.type.number);
} else {
YWrite(op.value, address, base, op.type.number, (Strider *)0);
}
PushDataBlock(RefNC(&nilDB));
}
void YSendNAKTimeout(PFileVar fv, PYVar yv, PComVar cv)
{
BYTE b;
int t;
/* flush comm buffer */
cv->InBuffCount = 0;
cv->InPtr = 0;
yv->NAKCount--;
if (yv->NAKCount<0)
{
if (yv->NAKMode==YnakC)
{
YSetOpt(fv,yv,XoptCheck);
yv->NAKMode = YnakC;
yv->NAKCount = 9;
}
else {
YCancel(fv,yv,cv);
return;
}
}
//if (yv->NAKMode!=YnakC)
if (1)
{
b = NAK;
if ((yv->PktNum==0) && (yv->PktNumOffset==0))
t = TimeOutInit;
else
t = yv->TOutLong;
}
else {
b = 'C';
t = TimeOutC;
}
YWrite(fv,yv,cv,&b,1);
yv->PktReadMode = XpktSOH;
FTSetTimeOut(fv,t);
}
// ファイル送信(local-to-remote)時に、YMODEMサーバからデータが送られてきたときに呼び出される。
BOOL YSendPacket(PFileVar fv, PYVar yv, PComVar cv)
{
// If current buffer is empty.
if (0 == yv->PktBufCount)
{
// Main read loop.
BOOL continue_read = TRUE;
while (continue_read)
{
BYTE isym = 0;
int is_success = YRead1Byte(fv, yv, cv, &isym);
if (0 == is_success) return TRUE;
// Analyze responce.
switch (isym)
{
case ACK:
// 1回目のEOT送信後のACK受信で、「1ファイル送信」の終わりとする。
// If we already send EOT, ACK means that client confirms it.
if (yv->SendEot)
{
// Reset the flag.
yv->SendEot = 0;
// 送信ファイルが残っていない場合は、「全てのファイルを転送終了」を通知する。
if (!GetNextFname(fv))
{
// If it is the last file.
yv->LastSendEot = 1;
break;
}
else
{
// Process with next file.
initialize_file_info(fv, yv);
}
}
// If client confirms that last (empty) packed was received.
// もう送信するファイルがない場合は、正常終了。
if (!fv->FileOpen)
{
fv->Success = TRUE;
yv->LastMessage = isym;
return FALSE;
}
// 次のブロックを送る
else if (yv->PktNumSent == (BYTE)(yv->PktNum + 1))
{
// ブロック0（ファイル情報）送信後は、ACK と 'C' を連続して受信することに
// なっているため、次の'C'を待つ。(2010.6.20 yutaka)
if ((yv->PktNum==0) && (yv->PktNumOffset==0))
{
// It is an ACK for file info, wait for 'C' by some reason (?).
// 送信済みフラグon
yv->SendFileInfo = 1;
continue_read = TRUE;
break;
}
yv->PktNum = yv->PktNumSent;
if (0 == yv->PktNum)
yv->PktNumOffset = yv->PktNumOffset + 256;
continue_read = FALSE;
}
break;
case NAK:
// 1回目のEOT送信後のNAK受信で、最後"EOT"を送る。
if (yv->SendEot)
{
yv->PktNum = yv->PktNumSent;
if (0 == yv->PktNum)
yv->PktNumOffset = yv->PktNumOffset + 256;
}
continue_read = FALSE;
break;
case CAN:
// 直前も CAN の場合はキャンセル
if (yv->LastMessage == CAN) {
fv->Success = FALSE; // failure
return FALSE;
}
break;
case 'C':
case 'G':
// 'C'を受け取ると、ブロックの送信を開始する。
if ((0 == yv->PktNum) && (0 == yv->PktNumOffset))
{
// ファイル情報送信後、ACK -> 'C' と受信したので、次のブロックを送信する。
if (yv->SendFileInfo)
{
yv->PktNum = yv->PktNumSent;
if (0 == yv->PktNum)
yv->PktNumOffset = yv->PktNumOffset + 256;
}
continue_read = FALSE;
}
else if (yv->LastSendEot)
{
continue_read = FALSE;
}
else
{
// TODO: analyze else branch.
}
break;
default:
assert(0);
break;
}
yv->LastMessage = isym;
}
// reset timeout timer
FTSetTimeOut(fv, TimeOutVeryLong);
#if 0
// 後続のサーバからのデータを読み捨てる。
do
{
lastrx = firstch;
i = YRead1Byte(fv,yv,cv,&b);
if (i != 0) {
firstch = b;
if (firstch == CAN && lastrx == CAN) {
// CAN(0x18)が連続してくると、ファイル送信の失敗と見なす。
// たとえば、サーバに同名のファイルが存在する場合など。
// (2010.3.23 yutaka)
fv->Success = FALSE; // failure
return FALSE;
}
}
} while (i != 0);
#endif
//================================
// Last packet case.
//================================
// オールゼロのブロックを送信して、もうファイルがないことを知らせる。
if (yv->LastSendEot)
{
WORD Check;
// Always 128 bytes for the last packet.
WORD last_packet_size = SOH_DATALEN;
int i;
// Clear the flag.
yv->LastSendEot = 0;
yv->__DataLen = last_packet_size;
yv->PktOut[0] = SOH;
yv->PktOut[1] = 0;
yv->PktOut[2] = ~0;
i = 0;
while (i < last_packet_size)
{
yv->PktOut[i+3] = 0x00;
i++;
}
Check = YCalcCheck(yv, yv->PktOut, last_packet_size);
// TODO: move checksum calculation to a function.
if (1 == yv->CheckLen)
{
yv->PktOut[last_packet_size + 3] = (BYTE)Check;
}
else
{
yv->PktOut[last_packet_size + 3] = HIBYTE(Check);
yv->PktOut[last_packet_size + 4] = LOBYTE(Check);
}
// TODO: remove magic number.
yv->PktBufCount = 3 + last_packet_size + yv->CheckLen;
}
//================================
// First or 256th packet case.
//================================
// Start a new sequence.
else if (yv->PktNumSent==yv->PktNum)
{
/* make a new packet */
BYTE *dataptr = &yv->PktOut[3];
int eot = 0; // End Of Transfer
WORD current_packet_size = yv->DataLen;
if (SOH_DATALEN == current_packet_size)
yv->PktOut[0] = SOH;
else
yv->PktOut[0] = STX;
yv->PktOut[1] = yv->PktNumSent;
yv->PktOut[2] = ~yv->PktNumSent;
// ブロック番号のカウントアップ。YMODEMでは"0"から開始する。
yv->PktNumSent++;
//================================
// First packet case.
//================================
// ブロック0
// ファイル情報の送信
if (yv->SendFileInfo == 0)
{
int ret, total;
size_t idx;
// TODO: remove magic number.
BYTE buf[1024 + 10];
// 128 bytes for the first packet.
current_packet_size = SOH_DATALEN;
yv->__DataLen = current_packet_size;
yv->PktOut[0] = SOH;
// Timestamp.
fv->FileMtime = GetFMtime(fv->FullName);
ret = _snprintf_s(buf, sizeof(buf), _TRUNCATE, "%s",
&(fv->FullName[fv->DirLen]));
// NULL-terminated string.
buf[ret] = 0x00;
total = ret + 1;
ret = _snprintf_s(&(buf[total]), sizeof(buf) - total, _TRUNCATE, "%lu %lo %o",
fv->FileSize, fv->FileMtime, 0644|_S_IFREG);
total += ret;
// if bloack0 is long, expand to 1024 bytes.
if (total > SOH_DATALEN) {
current_packet_size = STX_DATALEN;
yv->__DataLen = current_packet_size;
yv->PktOut[0] = STX;
}
// Padding.
idx = total;
while (idx <= current_packet_size)
{
buf[idx] = 0x00;
++idx;
}
// データコピー
memcpy(dataptr, buf, current_packet_size);
}
//================================
// 256th packet case.
//================================
else
{
BYTE fsym = 0;
size_t idx = 1;
yv->__DataLen = current_packet_size;
while ((idx <= current_packet_size) && fv->FileOpen &&
(1 == _lread(fv->FileHandle, &fsym, 1)))
{
// TODO: remove magic number.
yv->PktOut[2 + idx] = fsym;
++idx;
fv->ByteCount++;
}
// No bytes were read.
if (1 == idx)
{
// Close file handle.
if (fv->FileOpen)
{
_lclose(fv->FileHandle);
fv->FileHandle = 0;
fv->FileOpen = FALSE;
}
// Send EOT.
eot = 1;
}
else
{
// Padding.
while (idx <= current_packet_size)
{
// TODO: remove magic number.
yv->PktOut[2 + idx] = 0x1A;
++idx;
}
}
}
// データブロック
if (0 == eot)
{
// Add CRC if not End-of-Tranfer.
WORD Check = YCalcCheck(yv, yv->PktOut, current_packet_size);
// Checksum.
if (1 == yv->CheckLen)
yv->PktOut[current_packet_size + 3] = (BYTE)Check;
else {
yv->PktOut[current_packet_size + 3] = HIBYTE(Check);
yv->PktOut[current_packet_size + 4] = LOBYTE(Check);
}
// TODO: remove magic number.
yv->PktBufCount = 3 + current_packet_size + yv->CheckLen;
}
else
{
// EOT.
yv->PktOut[0] = EOT;
yv->PktBufCount = 1;
// EOTフラグon。次はNAKを期待する。
yv->SendEot = 1;
yv->LastSendEot = 0;
}
}
//================================
// TODO: Analyze resend case.
//================================
else
{
// Resend packet.
yv->PktBufCount = 3 + yv->__DataLen + yv->CheckLen;
}
// Reset counter.
yv->PktBufPtr = 0;
}
#if 0
/* a NAK or C could have arrived while we were buffering. Consume it. */
// 後続のサーバからのデータを読み捨てる。
do {
lastrx = firstch;
i = YRead1Byte(fv,yv,cv,&b);
if (i != 0) {
firstch = b;
if (firstch == CAN && lastrx == CAN) {
// CAN(0x18)が連続してくると、ファイル送信の失敗と見なす。
// たとえば、サーバに同名のファイルが存在する場合など。
// (2010.3.23 yutaka)
fv->Success = FALSE; // failure
return FALSE;
}
}
} while (i != 0);
#endif
// Write bytes to COM.
while (yv->PktBufCount > 0)
{
BYTE osym = yv->PktOut[yv->PktBufPtr];
int is_success = YWrite(fv, yv, cv, &osym, 1);
if (is_success > 0)
{
--yv->PktBufCount;
++yv->PktBufPtr;
}
else
break;
}
// Update dialog window.
if (0 == yv->PktBufCount)
{
if (0 == yv->PktNumSent)
{
SetDlgNum(fv->HWin, IDC_PROTOPKTNUM, yv->PktNumOffset + 256);
}
else
{
SetDlgNum(fv->HWin, IDC_PROTOPKTNUM, yv->PktNumOffset + yv->PktNumSent);
}
SetDlgNum(fv->HWin, IDC_PROTOBYTECOUNT, fv->ByteCount);
SetDlgPercent(fv->HWin, IDC_PROTOPERCENT, IDC_PROTOPROGRESS,
fv->ByteCount, fv->FileSize, &fv->ProgStat);
SetDlgTime(fv->HWin, IDC_PROTOELAPSEDTIME, fv->StartTime, fv->ByteCount);
}
return TRUE;
}
// YMODEMサーバからファイルを受信する際、ProtoParse()から呼び出される関数。
//
// +-------+-------+--------+---------+-----+
// |Header |Block# |1-Block#| Payload | CRC |
// +-------+-------+--------+---------+-----+
// 1 1 1 128/1024 2 byte
//
// return TRUE: ファイル受信中
// FALSE: 受信完了
BOOL YReadPacket(PFileVar fv, PYVar yv, PComVar cv)
{
BYTE b, d;
int i, c, nak;
BOOL GetPkt;
c = YRead1Byte(fv,yv,cv,&b);
GetPkt = FALSE;
while ((c>0) && (! GetPkt))
{
switch (yv->PktReadMode) {
case XpktSOH:
// SOH か STX かでブロック長が決まる。
if (b==SOH)
{
yv->PktIn[0] = b;
yv->PktReadMode = XpktBLK;
yv->__DataLen = SOH_DATALEN;
FTSetTimeOut(fv,yv->TOutShort);
}
else if (b==STX)
{
yv->PktIn[0] = b;
yv->PktReadMode = XpktBLK;
yv->__DataLen = STX_DATALEN;
FTSetTimeOut(fv,yv->TOutShort);
}
else if (b==EOT)
{
// EOTが来たら、1つのファイル受信が完了したことを示す。
if (fv->FileOpen) {
fv->FileOpen = 0;
_lclose(fv->FileHandle);
fv->FileHandle = -1;
if (fv->FileMtime > 0) {
SetFMtime(fv->FullName, fv->FileMtime);
}
// 1回目のEOTに対してNAKを返す
b = NAK;
YWrite(fv,yv,cv,&b, 1);
return TRUE;
}
initialize_file_info(fv, yv);
// EOTに対してACKを返す
b = ACK;
YWrite(fv,yv,cv,&b, 1);
// 次のファイル送信を促すため、'C'を送る。
YSendNAK(fv,yv,cv);
return TRUE;
}
else {
/* flush comm buffer */
cv->InBuffCount = 0;
cv->InPtr = 0;
return TRUE;
}
break;
case XpktBLK:
yv->PktIn[1] = b;
yv->PktReadMode = XpktBLK2;
FTSetTimeOut(fv,yv->TOutShort);
break;
case XpktBLK2:
nak = 1;
yv->PktIn[2] = b;
if ((b ^ yv->PktIn[1]) == 0xff) {
nak = 0;
if (yv->SendFileInfo) {
if (yv->PktIn[1] == (BYTE)(yv->PktNum + 1)) // 次のブロック番号か
nak = 0;
}
}
if (nak == 0)
{
yv->PktBufPtr = 3;
yv->PktBufCount = yv->__DataLen + yv->CheckLen;
yv->PktReadMode = XpktDATA;
FTSetTimeOut(fv,yv->TOutShort);
}
else
YSendNAK(fv,yv,cv);
break;
case XpktDATA:
yv->PktIn[yv->PktBufPtr] = b;
yv->PktBufPtr++;
yv->PktBufCount--;
GetPkt = yv->PktBufCount==0;
if (GetPkt)
{
FTSetTimeOut(fv,yv->TOutLong);
yv->PktReadMode = XpktSOH;
}
else
FTSetTimeOut(fv,yv->TOutShort);
break;
}
if (! GetPkt) c = YRead1Byte(fv,yv,cv,&b);
}
if (! GetPkt) return TRUE;
GetPkt = YCheckPacket(yv, yv->__DataLen);
if (! GetPkt)
{
YSendNAK(fv,yv,cv);
return TRUE;
}
// オールゼロならば、全ファイル受信の完了を示す。
if (yv->PktIn[1] == 0x00 && yv->PktIn[2] == 0xFF &&
yv->SendFileInfo == 0
) {
c = yv->__DataLen;
while ((c>0) && (yv->PktIn[2+c]==0x00))
c--;
if (c == 0) {
b = ACK;
YWrite(fv,yv,cv,&b, 1);
fv->Success = TRUE;
return FALSE;
}
}
d = yv->PktIn[1] - yv->PktNum;
if (d>1)
{
YCancel(fv,yv,cv);
return FALSE;
}
/* send ACK */
b = ACK;
YWrite(fv,yv,cv,&b, 1);
yv->NAKMode = YnakC;
yv->NAKCount = 10;
// 重複している場合は、何もしない。
if (yv->SendFileInfo &&
yv->PktIn[1] == (BYTE)(yv->PktNum)) {
return TRUE;
}
yv->PktNum = yv->PktIn[1];
// YMODEMの場合、block#0が「ファイル情報」となる。
if (d == 0 &&
yv->SendFileInfo == 0) {
long modtime;
long bytes_total;
int mode;
int ret;
BYTE *p;
char *name, *nameend;
p = (BYTE *)malloc(yv->__DataLen + 1);
memset(p, 0, yv->__DataLen + 1);
memcpy(p, &(yv->PktIn[3]), yv->__DataLen);
name = p;
strncpy_s(&(fv->FullName[fv->DirLen]),
sizeof(fv->FullName) - fv->DirLen, name,
_TRUNCATE);
if (!FTCreateFile(fv)) {
free(p);
return FALSE;
}
nameend = name + 1 + strlen(name);
if (*nameend) {
ret = sscanf(nameend, "%ld%lo%o", &bytes_total, &modtime, &mode);
if (ret >= 1) {
fv->FileSize = bytes_total;
yv->RecvFilesize = TRUE;
}
if (ret >= 2) {
fv->FileMtime = modtime;
}
}
SetDlgItemText(fv->HWin, IDC_PROTOFNAME, name);
yv->SendFileInfo = 1;
// 次のファイル送信を促すため、'C'を送る。
YSendNAK(fv,yv,cv);
free(p);
return TRUE;
}
if (yv->PktNum==0)
yv->PktNumOffset = yv->PktNumOffset + 256;
c = yv->__DataLen;
if (yv->TextFlag>0)
while ((c>0) && (yv->PktIn[2+c]==0x1A))
c--;
// 最終ブロックの余分なデータを除去する
if (yv->RecvFilesize && fv->ByteCount + c > fv->FileSize) {
c = fv->FileSize - fv->ByteCount;
}
if (yv->TextFlag>0)
for (i = 0 ; i <= c-1 ; i++)
{
b = yv->PktIn[3+i];
if ((b==LF) && (! yv->CRRecv))
_lwrite(fv->FileHandle,"\015",1);
if (yv->CRRecv && (b!=LF))
_lwrite(fv->FileHandle,"\012",1);
yv->CRRecv = b==CR;
_lwrite(fv->FileHandle,&b,1);
}
else
_lwrite(fv->FileHandle, &(yv->PktIn[3]), c);
fv->ByteCount = fv->ByteCount + c;
SetDlgNum(fv->HWin, IDC_PROTOPKTNUM, yv->PktNumOffset+yv->PktNum);
SetDlgNum(fv->HWin, IDC_PROTOBYTECOUNT, fv->ByteCount);
SetDlgTime(fv->HWin, IDC_PROTOELAPSEDTIME, fv->StartTime, fv->ByteCount);
FTSetTimeOut(fv,yv->TOutLong);
return TRUE;
}
void YInit
(PFileVar fv, PYVar yv, PComVar cv, PTTSet ts)
{
char inistr[MAX_PATH + 10];
if (yv->YMode == IdYSend) {
if (!GetNextFname(fv)) {
return;
}
}
fv->LogFlag = ((ts->LogFlag & LOG_Y)!=0);
if (fv->LogFlag)
fv->LogFile = _lcreat("YMODEM.LOG",0);
fv->LogState = 0;
fv->LogCount = 0;
SetWindowText(fv->HWin, fv->DlgCaption);
initialize_file_info(fv, yv);
if (cv->PortType==IdTCPIP)
{
yv->TOutShort = TimeOutVeryLong;
yv->TOutLong = TimeOutVeryLong;
}
else {
yv->TOutShort = TimeOutShort;
yv->TOutLong = TimeOutLong;
}
YSetOpt(fv,yv,yv->YOpt);
if (yv->YOpt == Yopt1K)
{
yv->NAKMode = YnakC;
yv->NAKCount = 10;
}
else {
yv->NAKMode = YnakG;
yv->NAKCount = 10;
}
if (fv->LogFlag) {
char buf[128];
time_t tm = time(NULL);
_snprintf_s(buf, sizeof(buf), _TRUNCATE, "YMODEM %s start: %s\n",
yv->YMode == IdYSend ? "Send" : "Recv",
ctime(&tm));
_lwrite(fv->LogFile, buf, strlen(buf));
}
switch (yv->YMode) {
case IdYSend:
yv->TextFlag = 0;
// ファイル送信開始前に、"rb ファイル名"を自動的に呼び出す。(2007.12.20 yutaka)
//strcpy(ts->YModemRcvCommand, "rb");
if (ts->YModemRcvCommand[0] != '\0') {
_snprintf_s(inistr, sizeof(inistr), _TRUNCATE, "%s\015",
ts->YModemRcvCommand);
YWrite(fv,yv,cv, inistr , strlen(inistr));
}
FTSetTimeOut(fv,TimeOutVeryLong);
break;
case IdYReceive:
#if 0 // for debug
strcpy(inistr, "sb -b svnrev.exe lrzsz-0.12.20.tar.gz\r\n");
// strcpy(inistr, "sb url3.txt url4.txt url5.txt\r\n");
YWrite(fv,yv,cv, inistr , strlen(inistr));
#endif
yv->TextFlag = 0;
YSendNAK(fv,yv,cv);
break;
}
}
