2 * Copyright 2000, International Business Machines Corporation and others.
5 * This software has been released under the terms of the IBM Public
6 * License. For details, see the LICENSE file in the top-level source
7 * directory or online at http://www.openafs.org/dl/license10.html
10 #include <afs/param.h>
12 #include <afs/afs_args.h>
24 #include <rx\rx_null.h>
26 #include <WINNT/syscfg.h>
30 #include "lanahelper.h"
32 extern int RXAFSCB_ExecuteRequest(struct rx_call *z_call);
33 extern int RXSTATS_ExecuteRequest(struct rx_call *z_call);
35 extern afs_int32 cryptall;
37 char AFSConfigKeyName[] =
38 "SYSTEM\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters";
42 char cm_rootVolumeName[64];
43 DWORD cm_rootVolumeNameLen;
44 cm_volume_t *cm_rootVolumep = NULL;
45 cm_cell_t *cm_rootCellp = NULL;
47 cm_scache_t *cm_rootSCachep;
48 char cm_mountRoot[1024];
49 DWORD cm_mountRootLen;
52 #ifdef AFS_FREELANCE_CLIENT
54 #endif /* freelance */
67 char cm_HostName[200];
70 char cm_CachePath[200];
71 DWORD cm_CachePathLen;
73 BOOL isGateway = FALSE;
75 BOOL reportSessionStartups = FALSE;
77 cm_initparams_v1 cm_initParams;
80 * AFSD Initialization Log
82 * This is distinct from the regular debug logging facility.
83 * Log items go directly to a file, not to an array in memory, so that even
84 * if AFSD crashes, the log can be inspected.
90 int cm_dnsEnabled = 1;
93 char cm_NetBiosName[32];
103 afsi_file = INVALID_HANDLE_VALUE;
106 strcpy(wd, getenv("TEMP"));
110 code = GetWindowsDirectory(wd, sizeof(wd));
111 if (code == 0) return;
113 strcat(wd, "\\afsd_init.log");
114 GetTimeFormat(LOCALE_SYSTEM_DEFAULT, 0, NULL, NULL, t, sizeof(t));
115 afsi_file = CreateFile(wd, GENERIC_WRITE, FILE_SHARE_READ, NULL,
116 OPEN_ALWAYS, FILE_FLAG_WRITE_THROUGH, NULL);
117 SetFilePointer(afsi_file, 0, NULL, FILE_END);
118 GetTimeFormat(LOCALE_SYSTEM_DEFAULT, 0, NULL, NULL, u, sizeof(u));
119 strcat(t, ": Create log file\n");
120 strcat(u, ": Created log file\n");
121 WriteFile(afsi_file, t, strlen(t), &zilch, NULL);
122 WriteFile(afsi_file, u, strlen(u), &zilch, NULL);
125 static int afsi_log_useTimestamp = 1;
128 afsi_log(char *pattern, ...)
130 char s[100], t[100], d[100], u[300];
133 va_start(ap, pattern);
135 vsprintf(s, pattern, ap);
136 if ( afsi_log_useTimestamp ) {
137 GetTimeFormat(LOCALE_SYSTEM_DEFAULT, 0, NULL, NULL, t, sizeof(t));
138 GetDateFormat(LOCALE_SYSTEM_DEFAULT, 0, NULL, NULL, d, sizeof(d));
139 sprintf(u, "%s %s: %s\n", d, t, s);
140 if (afsi_file != INVALID_HANDLE_VALUE)
141 WriteFile(afsi_file, u, strlen(u), &zilch, NULL);
146 if (afsi_file != INVALID_HANDLE_VALUE)
147 WriteFile(afsi_file, s, strlen(s), &zilch, NULL);
152 * Standard AFSD trace
155 void afsd_ForceTrace(BOOL flush)
161 if (!logReady) return;
163 len = GetTempPath(99, buf);
164 strcpy(&buf[len], "/afsd.log");
165 handle = CreateFile(buf, GENERIC_WRITE, FILE_SHARE_READ,
166 NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
167 if (handle == INVALID_HANDLE_VALUE) {
169 osi_panic("Cannot create log file", __FILE__, __LINE__);
171 osi_LogPrint(afsd_logp, handle);
173 FlushFileBuffers(handle);
178 * AFSD Initialization
181 int afsd_InitCM(char **reasonP)
190 long rx_mtu, rx_nojumbo;
191 char rootCellName[256];
192 struct rx_service *serverp;
193 static struct rx_securityClass *nullServerSecurityClassp;
200 /*int freelanceEnabled;*/
203 WSAStartup(0x0101, &WSAjunk);
205 /* setup osidebug server at RPC slot 1000 */
206 osi_LongToUID(1000, &debugID);
207 code = osi_InitDebug(&debugID);
208 afsi_log("osi_InitDebug code %d", code);
209 // osi_LockTypeSetDefault("stat"); /* comment this out for speed *
211 *reasonP = "unknown error";
216 gethostname(cm_HostName, sizeof(cm_HostName));
217 afsi_log("gethostname %s", cm_HostName);
218 thp = gethostbyname(cm_HostName);
219 memcpy(&cm_HostAddr, thp->h_addr_list[0], 4);
221 /* seed random number generator */
222 srand(ntohl(cm_HostAddr));
224 /* Look up configuration parameters in Registry */
225 code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSConfigKeyName,
226 0, KEY_QUERY_VALUE, &parmKey);
227 if (code != ERROR_SUCCESS) {
228 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM
229 | FORMAT_MESSAGE_ALLOCATE_BUFFER,
230 NULL, code, 0, (LPTSTR)&msgBuf, 0, NULL);
232 "Failure in configuration while opening Registry: %s",
234 osi_panic(buf, __FILE__, __LINE__);
237 dummyLen = sizeof(LANadapter);
238 code = RegQueryValueEx(parmKey, "LANadapter", NULL, NULL,
239 (BYTE *) &LANadapter, &dummyLen);
240 if (code == ERROR_SUCCESS) {
241 afsi_log("LAN adapter number %d", LANadapter);
242 if (LANadapter < 0 || LANadapter > MAX_LANA)
247 if ( LANadapter == -1 )
248 afsi_log("Default LAN adapter number");
250 dummyLen = sizeof(cacheSize);
251 code = RegQueryValueEx(parmKey, "CacheSize", NULL, NULL,
252 (BYTE *) &cacheSize, &dummyLen);
253 if (code == ERROR_SUCCESS)
254 afsi_log("Cache size %d", cacheSize);
256 cacheSize = CM_CONFIGDEFAULT_CACHESIZE;
257 afsi_log("Default cache size %d", cacheSize);
260 dummyLen = sizeof(logChunkSize);
261 code = RegQueryValueEx(parmKey, "ChunkSize", NULL, NULL,
262 (BYTE *) &logChunkSize, &dummyLen);
263 if (code == ERROR_SUCCESS) {
264 if (logChunkSize < 12 || logChunkSize > 30) {
265 afsi_log("Invalid chunk size %d, using default",
267 logChunkSize = CM_CONFIGDEFAULT_CHUNKSIZE;
269 afsi_log("Chunk size %d", logChunkSize);
271 logChunkSize = CM_CONFIGDEFAULT_CHUNKSIZE;
272 afsi_log("Default chunk size %d", logChunkSize);
274 cm_logChunkSize = logChunkSize;
275 cm_chunkSize = 1 << logChunkSize;
277 dummyLen = sizeof(numBkgD);
278 code = RegQueryValueEx(parmKey, "Daemons", NULL, NULL,
279 (BYTE *) &numBkgD, &dummyLen);
280 if (code == ERROR_SUCCESS)
281 afsi_log("%d background daemons", numBkgD);
283 numBkgD = CM_CONFIGDEFAULT_DAEMONS;
284 afsi_log("Defaulting to %d background daemons", numBkgD);
287 dummyLen = sizeof(numSvThreads);
288 code = RegQueryValueEx(parmKey, "ServerThreads", NULL, NULL,
289 (BYTE *) &numSvThreads, &dummyLen);
290 if (code == ERROR_SUCCESS)
291 afsi_log("%d server threads", numSvThreads);
293 numSvThreads = CM_CONFIGDEFAULT_SVTHREADS;
294 afsi_log("Defaulting to %d server threads", numSvThreads);
297 dummyLen = sizeof(stats);
298 code = RegQueryValueEx(parmKey, "Stats", NULL, NULL,
299 (BYTE *) &stats, &dummyLen);
300 if (code == ERROR_SUCCESS)
301 afsi_log("Status cache size %d", stats);
303 stats = CM_CONFIGDEFAULT_STATS;
304 afsi_log("Default status cache size %d", stats);
307 dummyLen = sizeof(ltt);
308 code = RegQueryValueEx(parmKey, "LogoffTokenTransfer", NULL, NULL,
309 (BYTE *) <t, &dummyLen);
310 if (code == ERROR_SUCCESS)
311 afsi_log("Logoff token transfer %s", (ltt ? "on" : "off"));
314 afsi_log("Logoff token transfer on by default");
316 smb_LogoffTokenTransfer = ltt;
319 dummyLen = sizeof(ltto);
320 code = RegQueryValueEx(parmKey, "LogoffTokenTransferTimeout",
321 NULL, NULL, (BYTE *) <to, &dummyLen);
322 if (code == ERROR_SUCCESS)
323 afsi_log("Logoff token tranfer timeout %d seconds",
327 afsi_log("Default logoff token transfer timeout 10 seconds");
330 smb_LogoffTransferTimeout = ltto;
332 dummyLen = sizeof(cm_rootVolumeName);
333 code = RegQueryValueEx(parmKey, "RootVolume", NULL, NULL,
334 cm_rootVolumeName, &dummyLen);
335 if (code == ERROR_SUCCESS)
336 afsi_log("Root volume %s", cm_rootVolumeName);
338 strcpy(cm_rootVolumeName, "root.afs");
339 afsi_log("Default root volume name root.afs");
342 cm_mountRootLen = sizeof(cm_mountRoot);
343 code = RegQueryValueEx(parmKey, "Mountroot", NULL, NULL,
344 cm_mountRoot, &cm_mountRootLen);
345 if (code == ERROR_SUCCESS) {
346 afsi_log("Mount root %s", cm_mountRoot);
347 cm_mountRootLen = strlen(cm_mountRoot);
349 strcpy(cm_mountRoot, "/afs");
354 dummyLen = sizeof(cm_CachePath);
355 code = RegQueryValueEx(parmKey, "CachePath", NULL, NULL,
356 cm_CachePath, &dummyLen);
357 if (code == ERROR_SUCCESS)
358 afsi_log("Cache path %s", cm_CachePath);
360 GetWindowsDirectory(cm_CachePath, sizeof(cm_CachePath));
361 cm_CachePath[2] = 0; /* get drive letter only */
362 strcat(cm_CachePath, "\\AFSCache");
363 afsi_log("Default cache path %s", cm_CachePath);
366 dummyLen = sizeof(traceOnPanic);
367 code = RegQueryValueEx(parmKey, "TrapOnPanic", NULL, NULL,
368 (BYTE *) &traceOnPanic, &dummyLen);
369 if (code == ERROR_SUCCESS)
370 afsi_log("Set to %s on panic",
371 traceOnPanic ? "trap" : "not trap");
377 dummyLen = sizeof(isGateway);
378 code = RegQueryValueEx(parmKey, "IsGateway", NULL, NULL,
379 (BYTE *) &isGateway, &dummyLen);
380 if (code == ERROR_SUCCESS)
381 afsi_log("Set for %s service",
382 isGateway ? "gateway" : "stand-alone");
388 dummyLen = sizeof(reportSessionStartups);
389 code = RegQueryValueEx(parmKey, "ReportSessionStartups", NULL, NULL,
390 (BYTE *) &reportSessionStartups, &dummyLen);
391 if (code == ERROR_SUCCESS)
392 afsi_log("Session startups %s be recorded in the Event Log",
393 reportSessionStartups ? "will" : "will not");
395 reportSessionStartups = 0;
399 dummyLen = sizeof(traceBufSize);
400 code = RegQueryValueEx(parmKey, "TraceBufferSize", NULL, NULL,
401 (BYTE *) &traceBufSize, &dummyLen);
402 if (code == ERROR_SUCCESS)
403 afsi_log("Trace Buffer size %d", traceBufSize);
405 traceBufSize = CM_CONFIGDEFAULT_TRACEBUFSIZE;
406 afsi_log("Default trace buffer size %d", traceBufSize);
409 dummyLen = sizeof(cm_sysName);
410 code = RegQueryValueEx(parmKey, "SysName", NULL, NULL,
411 cm_sysName, &dummyLen);
412 if (code == ERROR_SUCCESS)
413 afsi_log("Sys name %s", cm_sysName);
415 strcat(cm_sysName, "i386_nt40");
416 afsi_log("Default sys name %s", cm_sysName);
419 dummyLen = sizeof(cryptall);
420 code = RegQueryValueEx(parmKey, "SecurityLevel", NULL, NULL,
421 (BYTE *) &cryptall, &dummyLen);
422 if (code == ERROR_SUCCESS)
423 afsi_log("SecurityLevel is %s", cryptall?"crypt":"clear");
425 cryptall = rxkad_clear;
426 afsi_log("Default SecurityLevel is clear");
430 dummyLen = sizeof(cm_dnsEnabled);
431 code = RegQueryValueEx(parmKey, "UseDNS", NULL, NULL,
432 (BYTE *) &cm_dnsEnabled, &dummyLen);
433 if (code == ERROR_SUCCESS) {
434 afsi_log("DNS %s be used to find AFS cell servers",
435 cm_dnsEnabled ? "will" : "will not");
438 cm_dnsEnabled = 1; /* default on */
439 afsi_log("Default to use DNS to find AFS cell servers");
441 #else /* AFS_AFSDB_ENV */
442 afsi_log("AFS not built with DNS support to find AFS cell servers");
443 #endif /* AFS_AFSDB_ENV */
445 #ifdef AFS_FREELANCE_CLIENT
446 dummyLen = sizeof(cm_freelanceEnabled);
447 code = RegQueryValueEx(parmKey, "FreelanceClient", NULL, NULL,
448 (BYTE *) &cm_freelanceEnabled, &dummyLen);
449 if (code == ERROR_SUCCESS) {
450 afsi_log("Freelance client feature %s activated",
451 cm_freelanceEnabled ? "is" : "is not");
454 cm_freelanceEnabled = 0; /* default off */
456 #endif /* AFS_FREELANCE_CLIENT */
458 dummyLen = sizeof(cm_NetBiosName);
459 code = RegQueryValueEx(parmKey, "NetbiosName", NULL, NULL,
460 (BYTE *) &cm_NetBiosName, &dummyLen);
461 if (code == ERROR_SUCCESS) {
462 afsi_log("Explicit NetBios name is used %s", cm_NetBiosName);
465 cm_NetBiosName[0] = 0; /* default off */
468 dummyLen = sizeof(smb_hideDotFiles);
469 code = RegQueryValueEx(parmKey, "HideDotFiles", NULL, NULL,
470 (BYTE *) &smb_hideDotFiles, &dummyLen);
471 if (code != ERROR_SUCCESS) {
472 smb_hideDotFiles = 1; /* default on */
474 afsi_log("Dot files/dirs will %sbe marked hidden",
475 smb_hideDotFiles ? "" : "not ");
477 dummyLen = sizeof(smb_maxMpxRequests);
478 code = RegQueryValueEx(parmKey, "MaxMpxRequests", NULL, NULL,
479 (BYTE *) &smb_maxMpxRequests, &dummyLen);
480 if (code != ERROR_SUCCESS) {
481 smb_maxMpxRequests = 50;
483 afsi_log("Maximum number of multiplexed sessions is %d", smb_maxMpxRequests);
485 dummyLen = sizeof(smb_maxVCPerServer);
486 code = RegQueryValueEx(parmKey, "MaxVCPerServer", NULL, NULL,
487 (BYTE *) &smb_maxVCPerServer, &dummyLen);
488 if (code != ERROR_SUCCESS) {
489 smb_maxVCPerServer = 100;
491 afsi_log("Maximum number of VCs per server is %d", smb_maxVCPerServer);
493 dummyLen = sizeof(rx_nojumbo);
494 code = RegQueryValueEx(parmKey, "RxNoJumbo", NULL, NULL,
495 (BYTE *) &rx_nojumbo, &dummyLen);
496 if (code != ERROR_SUCCESS) {
500 afsi_log("RX Jumbograms are disabled");
502 dummyLen = sizeof(rx_mtu);
503 code = RegQueryValueEx(parmKey, "RxMaxMTU", NULL, NULL,
504 (BYTE *) &rx_mtu, &dummyLen);
505 if (code != ERROR_SUCCESS || !rx_mtu) {
509 afsi_log("RX maximum MTU is %d", rx_mtu);
511 RegCloseKey (parmKey);
513 /* setup early variables */
514 /* These both used to be configurable. */
516 buf_bufferSize = CM_CONFIGDEFAULT_BLOCKSIZE;
518 /* turn from 1024 byte units into memory blocks */
519 cacheBlocks = (cacheSize * 1024) / buf_bufferSize;
521 /* setup and enable debug log */
522 afsd_logp = osi_LogCreate("afsd", traceBufSize);
523 afsi_log("osi_LogCreate log addr %x", (int)afsd_logp);
524 osi_LogEnable(afsd_logp);
527 osi_Log0(afsd_logp, "Log init");
529 /* get network related info */
530 cm_noIPAddr = CM_MAXINTERFACE_ADDR;
531 code = syscfg_GetIFInfo(&cm_noIPAddr,
532 cm_IPAddr, cm_SubnetMask,
533 cm_NetMtu, cm_NetFlags);
535 if ( (cm_noIPAddr <= 0) || (code <= 0 ) )
536 afsi_log("syscfg_GetIFInfo error code %d", code);
538 afsi_log("First Network address %x SubnetMask %x",
539 cm_IPAddr[0], cm_SubnetMask[0]);
542 * Save client configuration for GetCacheConfig requests
544 cm_initParams.nChunkFiles = 0;
545 cm_initParams.nStatCaches = stats;
546 cm_initParams.nDataCaches = 0;
547 cm_initParams.nVolumeCaches = 0;
548 cm_initParams.firstChunkSize = cm_chunkSize;
549 cm_initParams.otherChunkSize = cm_chunkSize;
550 cm_initParams.cacheSize = cacheSize;
551 cm_initParams.setTime = 0;
552 cm_initParams.memCache = 0;
554 /* Set RX parameters before initializing RX */
559 rx_SetMaxMTU(rx_mtu);
561 /* initialize RX, and tell it to listen to port 7001, which is used for
562 * callback RPC messages.
564 code = rx_Init(htons(7001));
565 afsi_log("rx_Init code %x", code);
567 *reasonP = "afsd: failed to init rx client on port 7001";
571 /* Initialize the RPC server for session keys */
574 /* create an unauthenticated service #1 for callbacks */
575 nullServerSecurityClassp = rxnull_NewServerSecurityObject();
576 serverp = rx_NewService(0, 1, "AFS", &nullServerSecurityClassp, 1,
577 RXAFSCB_ExecuteRequest);
578 afsi_log("rx_NewService addr %x", (int)serverp);
579 if (serverp == NULL) {
580 *reasonP = "unknown error";
584 nullServerSecurityClassp = rxnull_NewServerSecurityObject();
585 serverp = rx_NewService(0, RX_STATS_SERVICE_ID, "rpcstats",
586 &nullServerSecurityClassp, 1, RXSTATS_ExecuteRequest);
587 afsi_log("rx_NewService addr %x", (int)serverp);
588 if (serverp == NULL) {
589 *reasonP = "unknown error";
593 /* start server threads, *not* donating this one to the pool */
595 afsi_log("rx_StartServer");
597 /* init user daemon, and other packages */
600 cm_InitACLCache(2*stats);
616 cm_InitSCache(stats);
618 code = cm_InitDCache(0, cacheBlocks);
619 afsi_log("cm_InitDCache code %x", code);
621 *reasonP = "error initializing cache";
626 #if !defined(_WIN32_WINNT) || _WIN32_WINNT < 0x500
627 if (cm_InitDNS(cm_dnsEnabled) == -1)
628 cm_dnsEnabled = 0; /* init failed, so deactivate */
629 afsi_log("cm_InitDNS %d", cm_dnsEnabled);
633 code = cm_GetRootCellName(rootCellName);
634 afsi_log("cm_GetRootCellName code %d, cm_freelanceEnabled= %d, rcn= %s",
635 code, cm_freelanceEnabled, (code ? "<none>" : rootCellName));
636 if (code != 0 && !cm_freelanceEnabled)
638 *reasonP = "can't find root cell name in afsd.ini";
641 else if (cm_freelanceEnabled)
644 if (code == 0 && !cm_freelanceEnabled)
646 cm_rootCellp = cm_GetCell(rootCellName, CM_FLAG_CREATE);
647 afsi_log("cm_GetCell addr %x", (int)cm_rootCellp);
648 if (cm_rootCellp == NULL)
650 *reasonP = "can't find root cell in afsdcell.ini";
655 #ifdef AFS_FREELANCE_CLIENT
656 if (cm_freelanceEnabled)
663 int afsd_InitDaemons(char **reasonP)
670 /* this should really be in an init daemon from here on down */
672 if (!cm_freelanceEnabled) {
673 code = cm_GetVolumeByName(cm_rootCellp, cm_rootVolumeName, cm_rootUserp,
674 &req, CM_FLAG_CREATE, &cm_rootVolumep);
675 afsi_log("cm_GetVolumeByName code %x root vol %x", code,
676 (code ? (cm_volume_t *)-1 : cm_rootVolumep));
678 *reasonP = "can't find root volume in root cell";
683 /* compute the root fid */
684 if (!cm_freelanceEnabled) {
685 cm_rootFid.cell = cm_rootCellp->cellID;
686 cm_rootFid.volume = cm_GetROVolumeID(cm_rootVolumep);
687 cm_rootFid.vnode = 1;
688 cm_rootFid.unique = 1;
691 cm_FakeRootFid(&cm_rootFid);
693 code = cm_GetSCache(&cm_rootFid, &cm_rootSCachep, cm_rootUserp, &req);
694 afsi_log("cm_GetSCache code %x scache %x", code,
695 (code ? (cm_scache_t *)-1 : cm_rootSCachep));
697 *reasonP = "unknown error";
701 cm_InitDaemon(numBkgD);
702 afsi_log("cm_InitDaemon");
707 int afsd_InitSMB(char **reasonP, void *aMBfunc)
711 lana_number_t lana = LANA_INVALID;
713 /* Do this last so that we don't handle requests before init is done.
714 * Here we initialize the SMB listener.
716 if (LANadapter == -1) {
717 DWORD noFindByName = 0;
722 /* Find the default LAN adapter to use. First look for
723 * the adapter named AFS; otherwise, unless we are doing
724 * gateway service, look for any valid loopback adapter.
726 code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSConfigKeyName,
727 0, KEY_QUERY_VALUE, &parmKey);
728 if (code == ERROR_SUCCESS) {
729 dummyLen = sizeof(LANadapter);
730 code = RegQueryValueEx(parmKey, "NoFindLanaByName", NULL, NULL,
731 (BYTE *) &noFindByName, &dummyLen);
732 RegCloseKey(parmKey);
736 lana = lana_FindLanaByName("AFS");
737 if (lana == LANA_INVALID && !isGateway)
738 lana = lana_FindLoopback();
739 if (lana != LANA_INVALID)
742 afsi_log("Lana %d", (int) lana);
743 /* If we are using a loopback adapter, we can use the preferred
744 * (but non-unique) server name; otherwise, we must fall back to
745 * the <machine>-AFS name.
747 if (LANadapter >= 0 && lana_IsLoopback((lana_number_t) LANadapter)) {
748 if ( cm_NetBiosName[0] )
749 strcpy(hostName, cm_NetBiosName);
751 strcpy(hostName, "AFS");
753 if (!cm_NetBiosName[0])
755 strcpy(hostName, cm_HostName);
756 ctemp = strchr(hostName, '.'); /* turn ntdfs.* into ntdfs */
757 if (ctemp) *ctemp = 0;
758 hostName[11] = 0; /* ensure that even after adding the -A, we
759 * leave one byte free for the netbios server
762 strcat(hostName, "-AFS");
764 strcpy(hostName, cm_NetBiosName);
768 smb_Init(afsd_logp, hostName, smb_UseV3, LANadapter, numSvThreads, aMBfunc);
769 afsi_log("smb_Init");
782 #pragma pack( push, before_imagehlp, 8 )
783 #include <imagehlp.h>
784 #pragma pack( pop, before_imagehlp )
786 #define MAXNAMELEN 1024
788 void afsd_printStack(HANDLE hThread, CONTEXT *c)
790 HANDLE hProcess = GetCurrentProcess();
794 char functionName[MAXNAMELEN];
796 IMAGEHLP_MODULE Module;
800 IMAGEHLP_SYMBOL *pSym;
802 afsi_log_useTimestamp = 0;
804 pSym = (IMAGEHLP_SYMBOL *) GlobalAlloc(0, sizeof (IMAGEHLP_SYMBOL) + MAXNAMELEN);
806 memset( &s, '\0', sizeof s );
807 if (!SymInitialize(hProcess, NULL, 1) )
809 afsi_log("SymInitialize(): GetLastError() = %lu\n", GetLastError() );
811 SymCleanup( hProcess );
817 symOptions = SymGetOptions();
818 symOptions |= SYMOPT_LOAD_LINES;
819 symOptions &= ~SYMOPT_UNDNAME;
820 SymSetOptions( symOptions );
823 * init STACKFRAME for first call
824 * Notes: AddrModeFlat is just an assumption. I hate VDM debugging.
825 * Notes: will have to be #ifdef-ed for Alphas; MIPSes are dead anyway,
828 #if defined (_ALPHA_) || defined (_MIPS_) || defined (_PPC_)
829 #error The STACKFRAME initialization in afsd_printStack() for this platform
830 #error must be properly configured
832 s.AddrPC.Offset = c->Eip;
833 s.AddrPC.Mode = AddrModeFlat;
834 s.AddrFrame.Offset = c->Ebp;
835 s.AddrFrame.Mode = AddrModeFlat;
838 memset( pSym, '\0', sizeof (IMAGEHLP_SYMBOL) + MAXNAMELEN );
839 pSym->SizeOfStruct = sizeof (IMAGEHLP_SYMBOL);
840 pSym->MaxNameLength = MAXNAMELEN;
842 memset( &Line, '\0', sizeof Line );
843 Line.SizeOfStruct = sizeof Line;
845 memset( &Module, '\0', sizeof Module );
846 Module.SizeOfStruct = sizeof Module;
850 afsi_log("\n--# FV EIP----- RetAddr- FramePtr StackPtr Symbol" );
852 for ( frameNum = 0; ; ++ frameNum )
855 * get next stack frame (StackWalk(), SymFunctionTableAccess(),
856 * SymGetModuleBase()). if this returns ERROR_INVALID_ADDRESS (487) or
857 * ERROR_NOACCESS (998), you can assume that either you are done, or
858 * that the stack is so hosed that the next deeper frame could not be
861 if ( ! StackWalk( IMAGE_FILE_MACHINE_I386, hProcess, hThread, &s, c,
862 NULL, SymFunctionTableAccess, SymGetModuleBase,
866 /* display its contents */
867 afsi_log("\n%3d %c%c %08lx %08lx %08lx %08lx ",
868 frameNum, s.Far? 'F': '.', s.Virtual? 'V': '.',
869 s.AddrPC.Offset, s.AddrReturn.Offset,
870 s.AddrFrame.Offset, s.AddrStack.Offset );
872 if ( s.AddrPC.Offset == 0 )
874 afsi_log("(-nosymbols- PC == 0)" );
878 /* show procedure info from a valid PC */
879 if (!SymGetSymFromAddr(hProcess, s.AddrPC.Offset, &offset, pSym))
881 if ( GetLastError() != ERROR_INVALID_ADDRESS )
883 afsi_log("SymGetSymFromAddr(): errno = %lu",
889 UnDecorateSymbolName(pSym->Name, functionName, MAXNAMELEN,
891 afsi_log("%s", functionName );
895 afsi_log(" %+ld bytes", (long) offset);
899 if (!SymGetLineFromAddr(hProcess, s.AddrPC.Offset, &offset, &Line))
901 if (GetLastError() != ERROR_INVALID_ADDRESS)
903 afsi_log("Error: SymGetLineFromAddr(): errno = %lu",
909 afsi_log(" Line: %s(%lu) %+ld bytes", Line.FileName,
910 Line.LineNumber, offset);
914 /* no return address means no deeper stackframe */
915 if (s.AddrReturn.Offset == 0)
922 if (GetLastError() != 0)
924 afsi_log("\nStackWalk(): errno = %lu\n", GetLastError());
927 SymCleanup(hProcess);
932 static DWORD *afsd_crtDbgBreakCurrent = NULL;
933 static DWORD afsd_crtDbgBreaks[256];
936 LONG __stdcall afsd_ExceptionFilter(EXCEPTION_POINTERS *ep)
940 BOOL allocRequestBrk = FALSE;
943 afsi_log("UnhandledException : code : 0x%x, address: 0x%x\n",
944 ep->ExceptionRecord->ExceptionCode,
945 ep->ExceptionRecord->ExceptionAddress);
948 if (afsd_crtDbgBreakCurrent &&
949 *afsd_crtDbgBreakCurrent == _CrtSetBreakAlloc(*afsd_crtDbgBreakCurrent))
951 allocRequestBrk = TRUE;
952 afsi_log("Breaking on alloc request # %d\n", *afsd_crtDbgBreakCurrent);
956 /* save context if we want to print the stack information */
957 context = *ep->ContextRecord;
959 afsd_printStack(GetCurrentThread(), &context);
961 if (ep->ExceptionRecord->ExceptionCode == EXCEPTION_BREAKPOINT)
963 afsi_log("\nEXCEPTION_BREAKPOINT - continue execition ...\n");
968 afsd_crtDbgBreakCurrent++;
969 _CrtSetBreakAlloc(*afsd_crtDbgBreakCurrent);
973 ep->ContextRecord->Eip++;
974 return EXCEPTION_CONTINUE_EXECUTION;
978 return EXCEPTION_CONTINUE_SEARCH;
982 void afsd_SetUnhandledExceptionFilter()
984 SetUnhandledExceptionFilter(afsd_ExceptionFilter);
988 void afsd_DbgBreakAllocInit()
990 memset(afsd_crtDbgBreaks, -1, sizeof(afsd_crtDbgBreaks));
991 afsd_crtDbgBreakCurrent = afsd_crtDbgBreaks;
994 void afsd_DbgBreakAdd(DWORD requestNumber)
997 for (i = 0; i < sizeof(afsd_crtDbgBreaks) - 1; i++)
999 if (afsd_crtDbgBreaks[i] == -1)
1004 afsd_crtDbgBreaks[i] = requestNumber;
1006 _CrtSetBreakAlloc(afsd_crtDbgBreaks[0]);