/* BLURB gpl Coda File System Release 5 Copyright (c) 1987-1999 Carnegie Mellon University Additional copyrights listed below This code is distributed "AS IS" without warranty of any kind under the terms of the GNU General Public Licence Version 2, as shown in the file LICENSE. The technical and financial contributors to Coda are listed in the file CREDITS. Additional copyrights #*/ /* IBM COPYRIGHT NOTICE Copyright (C) 1986 International Business Machines Corporation All Rights Reserved This file contains some code identical to or derived from the 1986 version of the Andrew File System ("AFS"), which is owned by the IBM Corporation. This code is provided "AS IS" and IBM does not warrant that it is free of infringement of any intellectual rights of any third party. IBM disclaims liability of any kind for any damages whatsoever resulting directly or indirectly from use of this software or of any derivative work. Carnegie Mellon University has obtained permission to modify, distribute and sublicense this code, which is based on Version 2 of AFS and does not contain the features and enhancements that are part of Version 3 of AFS. Version 3 of AFS is commercially available and supported by Transarc Corporation, Pittsburgh, PA. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "sftp.h" #include "test.h" #ifndef INET_ADDRSTRLEN #define INET_ADDRSTRLEN 16 #endif #define SUBSYS_SRV 1001 #define STESTSTACK 0x25000 extern long RPC2_Perror; extern long RPC2_DebugLevel; extern long SFTP_DebugLevel; static char ShortText[200]; static char LongText[3000]; long FindKey(); /* To obtain keys from ClientIdent */ long NoteAuthFailure(); /* To note authentication failures */ static void PrintHostIdent(), PrintPortIdent(); void GetParms(long argc, char *argv[], SFTP_Initializer *sftpI); void FillStrings(void); void InitRPC(void); long WhatHappened(long X, char *Y); long ProcessPacket(RPC2_Handle cIn, RPC2_PacketBuffer *pIn, RPC2_PacketBuffer *pOut); void PrintStats(void); #define DEFAULTLWPS 1 /* The default number of LWPs */ #define MAXLWPS 32 /* The maximum number of LWPs */ int numLWPs = 0; /* Number of LWPs created */ int availableLWPs = 0; /* Number of LWPs serving requests */ int maxLWPs = MAXLWPS; /* Max number of LWPs to create */ PROCESS pids[MAXLWPS] = { NULL }; /* Pid of each LWP */ void HandleRequests(void *); /* Routine to serve requests */ long VerboseFlag; RPC2_PortIdent ThisPort; long VMMaxFileSize; /* length of VMFileBuf, initially 0 */ long VMCurrFileSize; /* number of useful bytes in VMFileBuf */ char *VMFileBuf; /* for FILEINVM transfers */ int main(argc, argv) long argc; char *argv[]; { SFTP_Initializer sftpi; LWP_Init(LWP_VERSION, LWP_MAX_PRIORITY-1, &pids[0]); #ifdef PROFILE InitProfiling(); #endif FillStrings(); SFTP_SetDefaults(&sftpi); sftpi.WindowSize = 32; sftpi.SendAhead = 8; sftpi.AckPoint = 8; GetParms(argc, argv, &sftpi); SFTP_Activate(&sftpi); SFTP_EnforceQuota = 1; InitRPC(); if ((maxLWPs < 1) || (maxLWPs > MAXLWPS)) { printf("Bad max number of LWPs (%d), must be between 1 and %d\n", maxLWPs, MAXLWPS); exit(-1); } HandleRequests((void *)numLWPs); numLWPs++; return 0; /* make compiler happy */ } /* * Routine to server requests. */ void HandleRequests(void *arg) { RPC2_PacketBuffer *InBuff, *OutBuff; #if REQFILTER RPC2_RequestFilter reqfilter; #endif RPC2_Handle cid; long i; #if REQFILTER reqfilter.FromWhom = ANY; reqfilter.OldOrNew = OLDORNEW; #endif availableLWPs++; while (1) { /* * Get a request */ if ((i = RPC2_GetRequest( #if REQFILTER &reqfilter, #else (RPC2_RequestFilter *)NULL, #endif &cid, &InBuff, (struct timeval *)NULL, FindKey, (long)RPC2_XOR, NoteAuthFailure)) != RPC2_SUCCESS) { (void) WhatHappened(i, "GetRequest"); exit(-1); } /* * Decrement number of available LWPs. If count reaches zero and we * haven't reached the max number, create a new one. */ if ((--availableLWPs <= 0) && (numLWPs < maxLWPs)) { i = LWP_CreateProcess(HandleRequests, STESTSTACK, LWP_NORMAL_PRIORITY, (void *)numLWPs, "server", &pids[numLWPs]); assert(i == LWP_SUCCESS); printf("New LWP %d (%p)\n", numLWPs, pids[numLWPs]); numLWPs++; } assert(RPC2_AllocBuffer(RPC2_MAXPACKETSIZE-500, &OutBuff) == RPC2_SUCCESS); /* 500 is a fudge factor */ (void) ProcessPacket(cid, InBuff, OutBuff); availableLWPs++; if (InBuff != NULL) (void) RPC2_FreeBuffer(&InBuff); if (OutBuff != NULL) (void) RPC2_FreeBuffer(&OutBuff); } } long FindKey(authenticationtype, ClientIdent, IdentKey, SessionKey) RPC2_Integer authenticationtype; RPC2_CountedBS *ClientIdent; RPC2_EncryptionKey IdentKey; RPC2_EncryptionKey SessionKey; { long x; fprintf(stderr, "*** In FindKey('%s', 0x%lx, 0x%lx) ***\n", ClientIdent ? (char *)ClientIdent->SeqBody : "", *(long *)IdentKey, *(long *)SessionKey); x = -1; if (!ClientIdent) return 0; if (strcmp((char *)ClientIdent->SeqBody, "satya") == 0) x =1; if (strcmp((char *)ClientIdent->SeqBody, "bovik") == 0) x = 2; if (strcmp((char *)ClientIdent->SeqBody, "guest") == 0) x = 3; switch((int) x) { case 1: (void) strcpy((char *)IdentKey, "bananas"); (void) strcpy((char *)SessionKey, "BANANAS"); break; case 2: (void) strcpy((char *)IdentKey, "harryqb"); (void) strcpy((char *)SessionKey, "HARRYQB"); break; case 3: (void) strcpy((char *)IdentKey, "unknown"); (void) strcpy((char *)SessionKey, "UNKNOWN"); break; default: return(-1); } return(0); } long NoteAuthFailure(authenticationtype, cIdent, eType, pHost, pPort) RPC2_Integer authenticationtype; RPC2_CountedBS *cIdent; RPC2_Integer eType; RPC2_HostIdent *pHost; RPC2_PortIdent *pPort; { printf("Authentication using e-type %d failed for %s from\n\t", eType, (char *)cIdent->SeqBody); PrintHostIdent(pHost, (FILE *)NULL); printf("\t"); PrintPortIdent(pPort, (FILE *)NULL); printf("\n"); return(RPC2_SUCCESS); } static void PrintHostIdent(hPtr, tFile) RPC2_HostIdent *hPtr; FILE *tFile; { char addr[RPC2_ADDRSTRLEN]; if (tFile == NULL) tFile = stdout; /* it's ok, call-by-value */ switch (hPtr->Tag) { case RPC2_HOSTBYADDRINFO: RPC2_formataddrinfo(hPtr->Value.AddrInfo, addr, sizeof(addr)); fprintf(tFile, "Host.Addrinfo = %s", addr); break; case RPC2_HOSTBYINETADDR: inet_ntop(AF_INET, &hPtr->Value.InetAddress, addr,sizeof(addr)); fprintf(tFile, "Host.InetAddress = %s", addr); break; case RPC2_HOSTBYNAME: fprintf(tFile, "Host.Name = \"%s\"", hPtr->Value.Name); break; default: fprintf(tFile, "Host = ??????\n"); } (void) fflush(tFile); } static void PrintPortIdent(pPtr, tFile) RPC2_PortIdent *pPtr; FILE *tFile; { if (tFile == NULL) tFile = stdout; /* it's ok, call-by-value */ switch (pPtr->Tag) { case RPC2_PORTBYINETNUMBER: fprintf(tFile, "Port.InetPortNumber = %u", (unsigned) ntohs(pPtr->Value.InetPortNumber)); break; case RPC2_PORTBYNAME: fprintf(tFile, "Port.Name = \"%s\"", pPtr->Value.Name); break; default: fprintf(tFile, "Port = ??????"); } (void) fflush(tFile); } long WhatHappened(long X, char *Y) { if(VerboseFlag || X) printf("%s: %s (%ld)\n", Y, RPC2_ErrorMsg(X), X); return(X); } long ProcessPacket(RPC2_Handle cIn, RPC2_PacketBuffer *pIn, RPC2_PacketBuffer *pOut) { int *iptr; long i, opcode, replylen; char *cptr; SE_Descriptor sed; RPC2_NewConnectionBody *newconnbody; int smax, sused; memset(&sed, 0, sizeof(SE_Descriptor)); opcode = pIn->Header.Opcode; switch((int) opcode) { case REMOTESTATS: PrintStats(); pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = 0; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; case BEGINREMOTEPROFILING: pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = 0; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); #ifdef PROFILE ProfilingOn(); #endif break; case ENDREMOTEPROFILING: { /* hack */ extern PROCESS rpc2_SocketListenerPID; #ifdef PROFILE ProfilingOff(); DoneProfiling(); #endif pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = 0; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); LWP_StackUsed(rpc2_SocketListenerPID, &smax, &sused); printf("SL stack used: %d of %d\n", sused, smax); printf("\tCreation: Spkts = %ld Mpkts = %ld Lpkts = %ld SLEs = %ld Conns = %ld\n", rpc2_PBSmallCreationCount, rpc2_PBMediumCreationCount, rpc2_PBLargeCreationCount, rpc2_SLCreationCount, rpc2_ConnCreationCount); printf("\nFree: Spkts = %ld Mpkts = %ld Lpkts = %ld SLEs = %ld Conns = %ld\n", rpc2_PBSmallFreeCount, rpc2_PBMediumFreeCount, rpc2_PBLargeFreeCount, rpc2_SLFreeCount, rpc2_ConnFreeCount); break; } case FETCHFILE: case STOREFILE: sed.Tag = SMARTFTP; sed.Value.SmartFTPD.Tag = FILEBYNAME; sed.Value.SmartFTPD.SeekOffset = 0; sed.Value.SmartFTPD.FileInfo.ByName.ProtectionBits = 0644; if (opcode == (long) STOREFILE) sed.Value.SmartFTPD.TransmissionDirection = CLIENTTOSERVER; else sed.Value.SmartFTPD.TransmissionDirection = SERVERTOCLIENT; replylen = ntohl(*((unsigned long *)(pIn->Body))); sed.Value.SmartFTPD.SeekOffset = ntohl(*((unsigned long *)(pIn->Body + sizeof(long)))); printf("SeekOffset = %ld\n", sed.Value.SmartFTPD.SeekOffset); sed.Value.SmartFTPD.ByteQuota = ntohl(*((unsigned long *)(pIn->Body + 2*sizeof(long)))); printf("ByteQuota = %ld\n", sed.Value.SmartFTPD.ByteQuota); sed.Value.SmartFTPD.hashmark = *(pIn->Body+3*sizeof(long)); (void) strcpy((char *)sed.Value.SmartFTPD.FileInfo.ByName.LocalFileName, (char *)pIn->Body+1+3*sizeof(long)); if (strcmp(sed.Value.SmartFTPD.FileInfo.ByName.LocalFileName, "-") == 0) { sed.Value.SmartFTPD.Tag = FILEBYFD; sed.Value.SmartFTPD.FileInfo.ByFD.fd = (opcode == FETCHFILE) ? fileno(stdin) : fileno(stdout); } else { if (strcmp(sed.Value.SmartFTPD.FileInfo.ByName.LocalFileName, "/dev/mem") == 0) {/* Has to be set each time: other modes may clobber fields */ sed.Value.SmartFTPD.Tag = FILEINVM; sed.Value.SmartFTPD.FileInfo.ByAddr.vmfile.SeqBody = (RPC2_ByteSeq)VMFileBuf; sed.Value.SmartFTPD.FileInfo.ByAddr.vmfile.MaxSeqLen = VMMaxFileSize; sed.Value.SmartFTPD.FileInfo.ByAddr.vmfile.SeqLen = VMCurrFileSize; } } if (VerboseFlag) SFTP_PrintSED(&sed, rpc2_tracefile); if ((i = RPC2_InitSideEffect(cIn, &sed)) == RPC2_SUCCESS) { if ((i = RPC2_CheckSideEffect(cIn, &sed, (long)SE_AWAITLOCALSTATUS)) != RPC2_SUCCESS) { printf("RPC2_CheckSideEffect()--> %s\n", RPC2_ErrorMsg(i));(void) fflush(stdout); /* if (i < RPC2_FLIMIT) break; */ /* switch */ } } else { printf("RPC2_InitSideEffect()--> %s\n", RPC2_ErrorMsg(i));(void) fflush(stdout); /* if (i < RPC2_FLIMIT)break; */ /* switch */ } if (sed.LocalStatus != SE_SUCCESS) { printf("sed.LocalStatus = %s\n", SE_ErrorMsg((long)sed.LocalStatus)); } pOut->Header.ReturnCode = (int)sed.LocalStatus; if ((opcode == (long) STOREFILE) && (sed.Value.SmartFTPD.Tag == FILEINVM)) { VMCurrFileSize = sed.Value.SmartFTPD.FileInfo.ByAddr.vmfile.SeqLen; printf("VMCurrFileSize = %ld\n", VMCurrFileSize); } if (VerboseFlag) SFTP_PrintSED(&sed, rpc2_tracefile); pOut->Header.BodyLength = replylen; /* should ensure not too long */ memset(pOut->Body, 0, replylen); i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; case ONEPING: /* reply size = request size */ pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = pIn->Header.BodyLength; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; case MANYPINGS: pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = pIn->Header.BodyLength; i = RPC2_SendResponse(cIn, pOut); if (i != RPC2_SUCCESS) (void) WhatHappened(i, "SendResponse"); break; case LENGTHTEST: iptr = (int *)pIn->Body; *iptr = (int) htonl((unsigned long) *iptr); printf("Length: %d\n", *iptr);(void) fflush(stdout); cptr = (char *)iptr; cptr += sizeof(RPC2_Integer); pOut->Body[0] = '\0'; (void) strncat((char *)pOut->Body, cptr, *iptr); pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = *iptr; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; case DELACKTEST: { /* non lwp-blocking sleep for one second */ struct timeval tv; tv.tv_sec = 1; tv.tv_usec = 0; IOMGR_Select(0, NULL, NULL, NULL, &tv); pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = pIn->Header.BodyLength; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); } break; case SETREMOTEVMFILESIZE: { iptr = (int *)pIn->Body; VMMaxFileSize = (int) ntohl((unsigned long)*iptr); printf("New VM file buffer size = %ld\n ", VMMaxFileSize); if (VMFileBuf) free(VMFileBuf); assert(VMFileBuf = (char *)malloc((unsigned)VMMaxFileSize)); pOut->Header.ReturnCode = RPC2_SUCCESS; pOut->Header.BodyLength = 0; i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; } case RPC2_NEWCONNECTION: /* new connection */ newconnbody = (RPC2_NewConnectionBody *)pIn->Body; printf("New connection %#x: SideEffectType = %u SecurityLevel = %u ClientIdent = \"%s\"\n", cIn, ntohl(newconnbody->SideEffectType), ntohl(newconnbody->SecurityLevel), (char *)&newconnbody->ClientIdent_SeqBody); i = RPC2_SUCCESS; (void) RPC2_Enable(cIn); break; default: /* unknown opcode */ pOut->Header.ReturnCode = RPC2_FAIL; pOut->Header.BodyLength = 1 + strlen("Get your act together"); (void) strcpy((char *)pOut->Body, "Get your act together"); i = WhatHappened(RPC2_SendResponse(cIn, pOut), "SendResponse"); break; } #ifdef RPC2DEBUG if (i != RPC2_SUCCESS) sftp_DumpTrace("stest.dump"); #endif return(i); } void GetParms(long argc, char *argv[], SFTP_Initializer *sftpI) { int i; for (i = 1; i < argc; i++) { if (strcmp(argv[i], "-x") == 0 && i < argc -1) {RPC2_DebugLevel = atoi(argv[++i]); continue;} if (strcmp(argv[i], "-lwpdebug") == 0 && i < argc -1) {lwp_debug = atoi(argv[++i]); continue;} if (strcmp(argv[i], "-sx") == 0 && i < argc -1) {SFTP_DebugLevel = atoi(argv[++i]); continue;} if (strcmp(argv[i], "-l") == 0 && i < argc -1) {maxLWPs = atoi(argv[++i]); continue;} if (strcmp(argv[i], "-v") == 0) {VerboseFlag = 1; continue;} if (strcmp(argv[i], "-p") == 0 && i < argc - 1) { ThisPort.Value.InetPortNumber = atoi(argv[++i]); ThisPort.Value.InetPortNumber = htons(ThisPort.Value.InetPortNumber); continue; } printf("Usage: stest [-x debuglevel] [-sx sftpdebuglevel] [-l maxlwps] [-v verboseflag] [-p port]\n"); exit(-1); } } void FillStrings(void) { int i, j; for (i = 'a'; i < 'z'+1; i++) { j = 6*(i-'a'); ShortText[j] = ShortText[j+1] = ShortText[j+2] = i; ShortText[j+3] = ShortText[j+4] = ShortText[j+5] = i; } LongText[0]=0; for (i=0; i < 10; i++) (void) strcat(LongText, ShortText); } void InitRPC(void) { RPC2_PortIdent *pp; RPC2_SubsysIdent subsysid; ThisPort.Tag = RPC2_PORTBYINETNUMBER; pp = (ThisPort.Value.InetPortNumber == 0) ? NULL : &ThisPort; if (WhatHappened(RPC2_Init(RPC2_VERSION, (RPC2_Options *)NULL, pp, (long) 6, (struct timeval *)NULL), "Init") != RPC2_SUCCESS) exit(-1); PrintPortIdent(&rpc2_LocalPort, (FILE *)NULL); printf("\n\n"); subsysid.Tag = RPC2_SUBSYSBYID; subsysid.Value.SubsysId = SUBSYS_SRV; (void) RPC2_Export(&subsysid); } void PrintStats(void) { printf("RPC2:\n"); printf("Packets Sent = %lu\tPacket Retries = %lu (of %lu)\tPackets Received = %lu\n", rpc2_Sent.Total, rpc2_Sent.Retries, rpc2_Sent.Retries + rpc2_Sent.Cancelled, rpc2_Recvd.Total); printf("Bytes sent = %lu\tBytes received = %lu\n", rpc2_Sent.Bytes, rpc2_Recvd.Bytes); printf("Received Packet Distribution:\n"); printf("\tRequests = %lu\tGoodRequests = %lu\n", rpc2_Recvd.Requests, rpc2_Recvd.GoodRequests); printf("\tReplies = %lu\tGoodReplies = %lu\n", rpc2_Recvd.Replies, rpc2_Recvd.GoodReplies); printf("\tBusies = %lu\tGoodBusies = %lu\n", rpc2_Recvd.Busies, rpc2_Recvd.GoodBusies); printf("SFTP:\n"); printf("Packets Sent = %lu\t\tStarts Sent = %lu\t\tDatas Sent = %lu\n", sftp_Sent.Total, sftp_Sent.Starts, sftp_Sent.Datas); printf("Data Retries Sent = %lu\t\tAcks Sent = %lu\t\tNaks Sent = %lu\n", sftp_Sent.DataRetries, sftp_Sent.Acks, sftp_Sent.Naks); printf("Busies Sent = %lu\t\t\tBytes Sent = %lu\n", sftp_Sent.Busies, sftp_Sent.Bytes); printf("Packets Received = %lu\t\tStarts Received = %lu\tDatas Received = %lu\n", sftp_Recvd.Total, sftp_Recvd.Starts, sftp_Recvd.Datas); printf("Data Retries Received = %lu\tAcks Received = %lu\tNaks Received = %lu\n", sftp_Recvd.DataRetries, sftp_Recvd.Acks, sftp_Recvd.Naks); printf("Busies Received = %lu\t\tBytes Received = %lu\n", sftp_Recvd.Busies, sftp_Recvd.Bytes); (void) fflush(stdout); }