github
/
OpenID-Connect-Java-Spring-Server
mirror of https://github.com/mitreid-connect/OpenID-Connect-Java-Spring-Server
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Wrote functions to export authorization/authentication credentials to file

pull/650/head
arielak 2014-04-23 13:01:20 -04:00
parent 1df4ae1cf0
commit 953b213d04
2 changed files with 49 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
openid-connect-server/src/main/java/org/mitre/oauth2/repository/impl/JpaAuthenticationHolderRepository.java
View File

@ -16,6 +16,7 @@
******************************************************************************/ ******************************************************************************/
package org.mitre.oauth2.repository.impl; package org.mitre.oauth2.repository.impl;
import java.util.Collection;
import javax.persistence.EntityManager; import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext; import javax.persistence.PersistenceContext;
import javax.persistence.TypedQuery; import javax.persistence.TypedQuery;
@ -34,6 +35,12 @@ public class JpaAuthenticationHolderRepository implements AuthenticationHolderRe
@PersistenceContext @PersistenceContext
private EntityManager manager; private EntityManager manager;
@Override
public Collection<AuthenticationHolderEntity> getAll() {
TypedQuery<AuthenticationHolderEntity> query = manager.createNamedQuery("AuthenticationHolderEntity.getAll", AuthenticationHolderEntity.class);
return query.getResultList();
}
@Override @Override
public AuthenticationHolderEntity getById(Long id) { public AuthenticationHolderEntity getById(Long id) {
return manager.find(AuthenticationHolderEntity.class, id); return manager.find(AuthenticationHolderEntity.class, id);

45
openid-connect-server/src/main/java/org/mitre/openid/connect/service/impl/MITREidDataService_1_0.java
View File

@ -41,6 +41,8 @@ import org.mitre.oauth2.model.OAuth2RefreshTokenEntity;
import org.mitre.oauth2.repository.AuthenticationHolderRepository; import org.mitre.oauth2.repository.AuthenticationHolderRepository;
import org.mitre.oauth2.repository.OAuth2ClientRepository; import org.mitre.oauth2.repository.OAuth2ClientRepository;
import org.mitre.oauth2.repository.OAuth2TokenRepository; import org.mitre.oauth2.repository.OAuth2TokenRepository;
import org.mitre.openid.connect.model.ApprovedSite;
import org.mitre.openid.connect.model.WhitelistedSite;
import org.mitre.openid.connect.repository.ApprovedSiteRepository; import org.mitre.openid.connect.repository.ApprovedSiteRepository;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.GrantedAuthority; import org.springframework.security.core.GrantedAuthority;
@ -107,7 +109,6 @@ public class MITREidDataService_1_0 implements MITREidDataService {
writeClients(writer); writeClients(writer);
writer.endArray(); writer.endArray();
writer.name(GRANTS); writer.name(GRANTS);
writer.beginArray(); writer.beginArray();
writeGrants(writer); writeGrants(writer);
@ -123,7 +124,6 @@ public class MITREidDataService_1_0 implements MITREidDataService {
writeAccessTokens(writer); writeAccessTokens(writer);
writer.endArray(); writer.endArray();
writer.name(REFRESHTOKENS); writer.name(REFRESHTOKENS);
writer.beginArray(); writer.beginArray();
writeRefreshTokens(writer); writeRefreshTokens(writer);
@ -191,6 +191,7 @@ public class MITREidDataService_1_0 implements MITREidDataService {
logger.info("Done writing authentication holders"); logger.info("Done writing authentication holders");
} }
//used by writeAuthenticationHolders
private void writeAuthorizationRequest(AuthorizationRequest authReq, JsonWriter writer) throws IOException { private void writeAuthorizationRequest(AuthorizationRequest authReq, JsonWriter writer) throws IOException {
writer.beginObject(); writer.beginObject();
Map<String, String> authParams = authReq.getAuthorizationParameters(); Map<String, String> authParams = authReq.getAuthorizationParameters();
@ -254,7 +255,43 @@ public class MITREidDataService_1_0 implements MITREidDataService {
* @param writer * @param writer
*/ */
private void writeGrants(JsonWriter writer) { private void writeGrants(JsonWriter writer) {
approvedSiteRepo.getAll(); for (ApprovedSite site : approvedSiteRepo.getAll()) {
try {
writer.beginObject();
writer.name("id").value(site.getId());
writer.name("accessDate").value(site.getAccessDate().toString());
writer.name("clientId").value(site.getClientId());
writer.name("creationDate").value(site.getCreationDate().toString());
writer.name("timeoutDate").value(site.getTimeoutDate().toString());
writer.name("userId").value(site.getUserId());
writer.name("allowedScopes");
writer.beginArray();
for (String s : site.getAllowedScopes()) {
writer.value(s);
}
writer.endArray();
if(site.getIsWhitelisted()) {
WhitelistedSite wlSite = site.getWhitelistedSite();
writer.name("whitelistedSite");
writer.beginObject();
writer.name("id").value(wlSite.getId());
writer.name("clientId").value(wlSite.getClientId());
writer.name("creatorUserId").value(wlSite.getCreatorUserId());
writer.name("allowedScopes");
writer.beginArray();
for(String s : wlSite.getAllowedScopes()) {
writer.value(s);
}
writer.endArray();
writer.endObject();
}
writer.endObject();
logger.debug("Wrote grant {}", site.getId());
} catch (IOException ex) {
logger.error("Unable to write grant {}", site.getId(), ex);
}
}
logger.info("Done writing grants");
} }
/** /**
@ -265,7 +302,9 @@ public class MITREidDataService_1_0 implements MITREidDataService {
for(ClientDetailsEntity client : clientRepo.getAllClients()) { for(ClientDetailsEntity client : clientRepo.getAllClients()) {
String clientStr = gson.toJson(client); String clientStr = gson.toJson(client);
try { try {
//writer.beginObject();
writer.value(clientStr); writer.value(clientStr);
//writer.endObject();
logger.debug("Wrote client {}", client.getId()); logger.debug("Wrote client {}", client.getId());
} catch (IOException ex) { } catch (IOException ex) {
logger.error("Unable to write client {}", client.getId(), ex); logger.error("Unable to write client {}", client.getId(), ex);