void TraceSvcJrd::setAttachInfo(const string& /*svc_name*/, const string& user, const string& role, const string& /*pwd*/, bool /*trusted*/) { const unsigned char* bytes; unsigned int authBlockSize = m_svc.getAuthBlock(&bytes); if (authBlockSize) { m_authBlock.add(bytes, authBlockSize); m_user = ""; m_role = ""; m_admin = false; } else { m_user = user; m_role = role; m_admin = (m_user == DBA_USER_NAME) || (m_role == ADMIN_ROLE); } }
void fbtrace(UtilSvc* uSvc, TraceSvcIntf* traceSvc) { const char* const* end = uSvc->argv.end(); bool version = false, help = false; // search for "action" switch, set NULL into recognized argv const Switches actSwitches(trace_action_in_sw_table, FB_NELEM(trace_action_in_sw_table), false, true); const Switches::in_sw_tab_t* action_sw = NULL; const char** argv = uSvc->argv.begin(); for (++argv; argv < end; argv++) { if (!uSvc->isService() && strcmp(argv[0], "-?") == 0) { help = true; *argv = NULL; break; } const Switches::in_sw_tab_t* sw = actSwitches.findSwitch(*argv); if (sw) { if (sw->in_sw == IN_SW_TRACE_VERSION) { version = true; *argv = NULL; continue; } if (action_sw) usage(uSvc, isc_trace_conflict_acts, action_sw->in_sw_name, sw->in_sw_name); else action_sw = sw; *argv = NULL; } } if (version) { printMsg(1, SafeArg() << FB_VERSION); if (!action_sw) exit(FINI_OK); } if (!action_sw) { if (help) usage(uSvc, 0); else usage(uSvc, isc_trace_act_notfound); } // search for action's parameters, set NULL into recognized argv const Switches optSwitches(trace_option_in_sw_table, FB_NELEM(trace_option_in_sw_table), false, true); TraceSession session(*getDefaultMemoryPool()); argv = uSvc->argv.begin(); for (++argv; argv < end; argv++) { if (!*argv) continue; const Switches::in_sw_tab_t* sw = optSwitches.findSwitch(*argv); if (!sw) continue; *argv = NULL; switch (sw->in_sw) { case IN_SW_TRACE_CONFIG: switch (action_sw->in_sw) { case IN_SW_TRACE_STOP: case IN_SW_TRACE_SUSPEND: case IN_SW_TRACE_RESUME: case IN_SW_TRACE_LIST: usage(uSvc, isc_trace_param_act_notcompat, sw->in_sw_name, action_sw->in_sw_name); break; } if (!session.ses_config.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) session.ses_config = *argv; else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; case IN_SW_TRACE_NAME: switch (action_sw->in_sw) { case IN_SW_TRACE_STOP: case IN_SW_TRACE_SUSPEND: case IN_SW_TRACE_RESUME: case IN_SW_TRACE_LIST: usage(uSvc, isc_trace_param_act_notcompat, sw->in_sw_name, action_sw->in_sw_name); break; } if (!session.ses_name.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) session.ses_name = *argv; else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; case IN_SW_TRACE_ID: switch (action_sw->in_sw) { case IN_SW_TRACE_START: case IN_SW_TRACE_LIST: usage(uSvc, isc_trace_param_act_notcompat, sw->in_sw_name, action_sw->in_sw_name); break; } if (session.ses_id) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) { session.ses_id = atol(*argv); if (!session.ses_id) usage(uSvc, isc_trace_param_invalid, *argv, sw->in_sw_name); } else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; default: fb_assert(false); } *argv = NULL; } // search for authentication parameters const Switches authSwitches(trace_auth_in_sw_table, FB_NELEM(trace_auth_in_sw_table), false, true); string svc_name, user, pwd; bool adminRole = false; argv = uSvc->argv.begin(); for (++argv; argv < end; argv++) { if (!*argv) continue; const Switches::in_sw_tab_t* sw = authSwitches.findSwitch(*argv); if (!sw) { usage(uSvc, isc_trace_switch_unknown, *argv); } switch (sw->in_sw) { case IN_SW_TRACE_USERNAME: if (!user.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) user = *argv; else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; case IN_SW_TRACE_PASSWORD: if (!pwd.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) pwd = *argv; else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; case IN_SW_TRACE_FETCH_PWD: if (uSvc->isService()) usage(uSvc, isc_trace_switch_user_only, sw->in_sw_name); if (!pwd.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) { const PathName fileName(*argv); const char *s = NULL; switch (fb_utils::fetchPassword(fileName, s)) { case fb_utils::FETCH_PASS_OK: pwd = s; break; case fb_utils::FETCH_PASS_FILE_OPEN_ERROR: (Arg::Gds(isc_io_error) << Arg::Str("open") << Arg::Str(fileName) << Arg::Gds(isc_io_open_err) << Arg::OsError()).raise(); break; case fb_utils::FETCH_PASS_FILE_READ_ERROR: case fb_utils::FETCH_PASS_FILE_EMPTY: (Arg::Gds(isc_io_error) << Arg::Str("read") << Arg::Str(fileName) << Arg::Gds(isc_io_read_err) << Arg::OsError()).raise(); break; } } else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; case IN_SW_TRACE_TRUSTED_AUTH: if (uSvc->isService()) usage(uSvc, isc_trace_switch_user_only, sw->in_sw_name); adminRole = true; break; case IN_SW_TRACE_SERVICE_NAME: if (uSvc->isService()) continue; if (!svc_name.empty()) usage(uSvc, isc_trace_switch_once, sw->in_sw_name); argv++; if (argv < end && *argv) svc_name = *argv; else usage(uSvc, isc_trace_param_val_miss, sw->in_sw_name); break; default: fb_assert(false); } } // validate missed action's parameters and perform action if (!uSvc->isService() && svc_name.isEmpty()) { usage(uSvc, isc_trace_mandatory_switch_miss, "SERVICE"); } if (!session.ses_id) { switch (action_sw->in_sw) { case IN_SW_TRACE_STOP: case IN_SW_TRACE_SUSPEND: case IN_SW_TRACE_RESUME: usage(uSvc, isc_trace_switch_param_miss, "ID", action_sw->in_sw_name); break; } } if (session.ses_config.empty()) { if (action_sw->in_sw == IN_SW_TRACE_START) { usage(uSvc, isc_trace_switch_param_miss, "CONFIG", action_sw->in_sw_name); } } AuthReader::AuthBlock authBlock; const unsigned char* bytes; unsigned int authBlockSize = uSvc->getAuthBlock(&bytes); if (authBlockSize) { authBlock.add(bytes, authBlockSize); pwd = ""; user = ""; adminRole = false; } traceSvc->setAttachInfo(svc_name, user, pwd, authBlock, adminRole); switch (action_sw->in_sw) { case IN_SW_TRACE_START: traceSvc->startSession(session, true); break; case IN_SW_TRACE_STOP: traceSvc->stopSession(session.ses_id); break; case IN_SW_TRACE_SUSPEND: traceSvc->setActive(session.ses_id, false); break; case IN_SW_TRACE_RESUME: traceSvc->setActive(session.ses_id, true); break; case IN_SW_TRACE_LIST: traceSvc->listSessions(); break; default: fb_assert(false); } }