org.littleshoot.commom.xmpp.SASLFacebookMechanism Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xmpp Show documentation
The newest version!
package org.littleshoot.commom.xmpp;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.Map;
import java.util.SortedMap;
import java.util.TreeMap;

import javax.security.auth.callback.CallbackHandler;
import javax.security.sasl.Sasl;

//import org.apache.harmony.javax.security.auth.callback.CallbackHandler;
import org.jivesoftware.smack.SASLAuthentication;
import org.jivesoftware.smack.XMPPException;
import org.jivesoftware.smack.sasl.SASLMechanism;
import org.jivesoftware.smack.util.Base64;


// ported from
// http://code.google.com/p/fbgc/source/browse/trunk/daemon/src/main/java/org/albino/mechanisms/FacebookConnectSASLMechanism.java

public class SASLFacebookMechanism extends SASLMechanism {
    
    // TODO: line lengths
    // TODO: string format
    
    public static String NAME = "X-FACEBOOK-PLATFORM";

    private String sessionKey = "";
    private String sessionSecret = "";
    private String apiKey = "";
 
    public SASLFacebookMechanism(SASLAuthentication arg0) {
        super(arg0);
    }

    protected void authenticate() throws IOException, XMPPException {
        //final AuthMechanism packet = new AuthMechanism(NAME, null);
        // Send the authentication to the server
        getSASLAuthentication().send(null);
        //getSASLAuthentication().send(packet);
    }

    public void authenticate(String apiKeyAndSessionKey, String host, String sessionSecret)
            throws IOException, XMPPException {
        if (apiKeyAndSessionKey == null || sessionSecret == null) {
            throw new IllegalStateException("Invalid parameters!");
        }
        
        String[] keyArray = apiKeyAndSessionKey.split("\\|");
        
        if (keyArray == null || keyArray.length != 2) {
            throw new IllegalStateException("Api key or session key is not present!");
        }
        
        this.apiKey = keyArray[0];
        this.sessionKey = keyArray[1];
        this.sessionSecret = sessionSecret;
        
        this.authenticationId = sessionKey;
        this.password = sessionSecret;
        this.hostname = host;

        String[] mechanisms = { "DIGEST-MD5" };
        Map props = new HashMap();
        sc = Sasl.createSaslClient(mechanisms, null, "xmpp", host, props, this);
        authenticate();
    }

    public void authenticate(String username, String host, CallbackHandler cbh)
            throws IOException, XMPPException {
        String[] mechanisms = { "DIGEST-MD5" };
        Map props = new HashMap();
        sc = Sasl.createSaslClient(mechanisms, null, "xmpp", host, props, cbh);
        authenticate();
    }

    protected String getName() {
        return NAME;
    }

    public void challengeReceived(String challenge) throws IOException {
        byte response[] = null;
        if (challenge != null) {
            String decodedResponse = new String(Base64.decode(challenge));
            Map parameters = getQueryMap(decodedResponse);
            Long callId = new GregorianCalendar().getTimeInMillis() / 1000;
            
            SortedMap params = new TreeMap();
            params.put("api_key", apiKey);
            params.put("call_id", callId.toString());
            params.put("method", parameters.get("method"));
            params.put("nonce", parameters.get("nonce"));
            params.put("session_key", sessionKey);
            params.put("v", "1.0");
            
            StringBuilder sigB = new StringBuilder();
            for (Map.Entry entry : params.entrySet()) {
                sigB.append(String.format("%s=%s", 
                    entry.getKey(), entry.getValue()));
            }
            sigB.append(sessionSecret);
            
            String sig;
            try {
                sig = MD5(sigB.toString());
            } catch (NoSuchAlgorithmException e) {
                throw new IllegalStateException(e);
            }
            
            StringBuilder composedResp = new StringBuilder();
            for (Map.Entry entry : params.entrySet()) {
                composedResp.append(String.format("%s=%s&", 
                    entry.getKey(), entry.getValue()));
            }
            composedResp.append("sig=" + sig);
            
            response = composedResp.toString().getBytes();
        }
        String authenticationText = response == null? "" : 
            Base64.encodeBytes(response, Base64.DONT_BREAK_LINES);
        //getSASLAuthentication().send(authenticationText);
        getSASLAuthentication().send(new Response(authenticationText));
    }

    private Map getQueryMap(String query) {
            String[] params = query.split("&");
            Map map = new HashMap();
            for (String param : params) {
                String name = param.split("=")[0];
                String value = param.split("=")[1];
                map.put(name, value);
            }
            return map;
    }
    
    private String convertToHex(byte[] data) {
        StringBuilder buf = new StringBuilder();
        for (int i = 0; i < data.length; i++) {
            int halfbyte = (data[i] >>> 4) & 0x0F;
            int two_halfs = 0;
            do {
                if ((0 <= halfbyte) && (halfbyte <= 9))
                    buf.append((char) ('0' + halfbyte));
                else
                    buf.append((char) ('a' + (halfbyte - 10)));
                halfbyte = data[i] & 0x0F;
            } while(two_halfs++ < 1);
        }
        return buf.toString();
    }

    private String MD5(String text) 
            throws NoSuchAlgorithmException, UnsupportedEncodingException  {
        MessageDigest md;
        md = MessageDigest.getInstance("MD5");
        byte[] md5hash = new byte[32];
        md.update(text.getBytes("iso-8859-1"), 0, text.length());
        md5hash = md.digest();
        return convertToHex(md5hash);
    }
}