use of com.yahoo.athenz.common.server.log.AuditLogMsgBuilder in project athenz by yahoo.
the class AuditLogMsgBuilderTest method starter.
DefaultAuditLogMsgBuilder starter(final String whatApi) {
AuditLoggerFactory auditLoggerFactory = new DefaultAuditLoggerFactory();
AuditLogger logger = auditLoggerFactory.create();
AuditLogMsgBuilder msgBldr = logger.getMsgBuilder();
msgBldr.who(TOKEN_STR).when("now-timestamp").clientIp("12.12.12.12").whatApi(whatApi);
return (DefaultAuditLogMsgBuilder) msgBldr;
}
use of com.yahoo.athenz.common.server.log.AuditLogMsgBuilder in project athenz by yahoo.
the class AuditLoggerTest method testLogMsgBuilder.
@Test
public void testLogMsgBuilder() {
AuditLoggerFactory auditLoggerFactory = new DefaultAuditLoggerFactory();
AuditLogger logger = auditLoggerFactory.create();
AuditLogMsgBuilder msgBldr = logger.getMsgBuilder();
auditLogger.log(msgBldr);
}
use of com.yahoo.athenz.common.server.log.AuditLogMsgBuilder in project athenz by yahoo.
the class ZMSUtils method getAuditLogMsgBuilder.
/**
* Setup a new AuditLogMsgBuilder object with common values.
*/
public static AuditLogMsgBuilder getAuditLogMsgBuilder(ResourceContext ctx, AuditLogger auditLogger, String domainName, String auditRef, String caller, String method) {
AuditLogMsgBuilder msgBldr = auditLogger.getMsgBuilder();
// get the where - which means where this server is running
msgBldr.where(ZMSImpl.serverHostName);
msgBldr.whatDomain(domainName).why(auditRef).whatApi(caller).whatMethod(method);
if (ctx != null) {
Principal princ = ((RsrcCtxWrapper) ctx).principal();
if (princ != null) {
String unsignedCreds = princ.getUnsignedCredentials();
if (unsignedCreds == null) {
StringBuilder sb = new StringBuilder();
sb.append("who-name=").append(princ.getName());
sb.append(",who-domain=").append(princ.getDomain());
sb.append(",who-fullname=").append(princ.getFullName());
List<String> roles = princ.getRoles();
if (roles != null && roles.size() > 0) {
sb.append(",who-roles=").append(roles.toString());
}
unsignedCreds = sb.toString();
}
msgBldr.who(unsignedCreds);
}
// get the client IP
msgBldr.clientIp(ServletRequestUtil.getRemoteAddress(ctx.request()));
}
return msgBldr;
}
use of com.yahoo.athenz.common.server.log.AuditLogMsgBuilder in project athenz by yahoo.
the class ZTSImplTest method testGetRoleTokenInvalidDomainAuditLog.
@Test
public void testGetRoleTokenInvalidDomainAuditLog() {
HttpServletRequest servletRequest = Mockito.mock(HttpServletRequest.class);
Mockito.when(servletRequest.getRemoteAddr()).thenReturn("55.88.77.66");
Mockito.when(servletRequest.isSecure()).thenReturn(true);
final java.util.Set<String> aLogMsgs = new java.util.HashSet<String>();
AuditLogger alogger = new AuditLogger() {
public void log(String logMsg, String msgVersionTag) {
aLogMsgs.add(logMsg);
}
public void log(AuditLogMsgBuilder msgBldr) {
String msg = msgBldr.build();
aLogMsgs.add(msg);
}
@Override
public AuditLogMsgBuilder getMsgBuilder() {
return new DefaultAuditLogMsgBuilder();
}
};
ChangeLogStore structStore = new ZMSFileChangeLogStore("/tmp/zts_server_unit_tests/zts_root", privateKey, "0");
DataStore store = new DataStore(structStore, null);
ZTSImpl ztsImpl = new ZTSImpl(mockCloudStore, store);
ztsImpl.auditLogger = alogger;
ZTSImpl.serverHostName = "localhost";
SignedDomain signedDomain = createSignedDomain("coretech", "weather", "storage", true);
store.processDomain(signedDomain, false);
Principal principal = SimplePrincipal.create("user_domain", "user", "v=U1;d=user_domain;n=user;s=signature", 0, null);
ResourceContext context = createResourceContext(principal, servletRequest);
try {
ztsImpl.getRoleToken(context, "invalidDomain", null, Integer.valueOf(600), Integer.valueOf(1200), null);
fail();
} catch (ResourceException ex) {
assertEquals(ex.getCode(), 404);
}
for (String msg : aLogMsgs) {
assertTrue(msg.contains("ERROR=(No Such Domain)"));
assertTrue(msg.contains("CLIENT-IP=(55.88.77.66)"));
assertTrue(msg.contains("WHO=(who-name=user,who-domain=user_domain,who-fullname=user_domain.user)"));
break;
}
}
use of com.yahoo.athenz.common.server.log.AuditLogMsgBuilder in project athenz by yahoo.
the class ZTSImplTest method testGetRoleTokenAddrLoopbackNoXFFAuditLog.
@Test
public void testGetRoleTokenAddrLoopbackNoXFFAuditLog() {
HttpServletRequest servletRequest = Mockito.mock(HttpServletRequest.class);
Mockito.when(servletRequest.getRemoteAddr()).thenReturn("127.0.0.1");
Mockito.when(servletRequest.isSecure()).thenReturn(true);
final java.util.Set<String> aLogMsgs = new java.util.HashSet<String>();
AuditLogger alogger = new AuditLogger() {
public void log(String logMsg, String msgVersionTag) {
aLogMsgs.add(logMsg);
}
public void log(AuditLogMsgBuilder msgBldr) {
String msg = msgBldr.build();
aLogMsgs.add(msg);
}
@Override
public AuditLogMsgBuilder getMsgBuilder() {
return new DefaultAuditLogMsgBuilder();
}
};
ChangeLogStore structStore = new ZMSFileChangeLogStore("/tmp/zts_server_unit_tests/zts_root", privateKey, "0");
DataStore store = new DataStore(structStore, null);
ZTSImpl ztsImpl = new ZTSImpl(mockCloudStore, store);
ZTSImpl.serverHostName = "localhost";
ztsImpl.auditLogger = alogger;
SignedDomain signedDomain = createSignedDomain("coretech", "weather", "storage", true);
store.processDomain(signedDomain, false);
Principal principal = SimplePrincipal.create("user_domain", "user", "v=U1;d=user_domain;n=user;s=signature", 0, null);
ResourceContext context = createResourceContext(principal, servletRequest);
RoleToken roleToken = ztsImpl.getRoleToken(context, "coretech", null, Integer.valueOf(600), Integer.valueOf(1200), null);
com.yahoo.athenz.auth.token.RoleToken token = new com.yahoo.athenz.auth.token.RoleToken(roleToken.getToken());
assertNotNull(token);
String unsignToken = token.getUnsignedToken();
for (String msg : aLogMsgs) {
assertTrue(msg.contains("SUCCESS ROLETOKEN=(" + unsignToken));
assertTrue(msg.contains("i=127.0.0.1"));
assertTrue(msg.contains("CLIENT-IP=(127.0.0.1)"));
break;
}
}
Aggregations