Marshalling.java

package org.objectweb.security.context;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;

public class Marshalling {

    public static final int SEC_CTX_ID = 101;

    private Marshalling() {

    }

    public static byte[] marshallSecurityContext(SecurityContext ctx) {
        byte [] result = null;
        try {
            if (ctx != null) {
                // build a new byte from ctx
                ArrayList cResult = new ArrayList();
                // add the pname
                addBytes(cResult, ctx.getPrincipalName());
                // add the roles
                String [] roles = ctx.getRoles();
                int nbRoles = 0;
                if (roles != null) {
                    nbRoles = roles.length;
                }
                addBytes(cResult, new Integer(nbRoles).toString());
                for (int i = 0; i < nbRoles; i++) {
                    addBytes(cResult, roles[i]);
                }

                // add runas
                addBytes(cResult, new Integer(ctx.getRunAsRoleStack().size()).toString());
                Iterator iRunAs = ctx.getRunAsRoleStack().iterator();
                while (iRunAs.hasNext()) {
                    addBytes(cResult, (String) iRunAs.next());
                }

                // add runas principal stack
                addBytes(cResult, new Integer(ctx.getRunAsPrincipalStack().size()).toString());
                iRunAs = ctx.getRunAsPrincipalStack().iterator();
                while (iRunAs.hasNext()) {
                    addBytes(cResult, (String) iRunAs.next());
                }

                // add runas roles of principal mapping
                addBytes(cResult, new Integer(ctx.getRunAsPrincipalRolesStack().size()).toString());
                iRunAs = ctx.getRunAsPrincipalRolesStack().iterator();
                while (iRunAs.hasNext()) {
                    addBytes(cResult, (String[]) iRunAs.next());
                }

                // result casting
                result = new byte[cResult.size()];
                for (int i = 0; i < cResult.size(); i++) {
                    result[i] = ((Byte) cResult.get(i)).byteValue();
                }
            }

        } catch (java.io.UnsupportedEncodingException uee) {
            uee.printStackTrace();
        }
        return result;
    }

    public static SecurityContext unmarshallSecurityContext(byte [] byteCtx) {
        SecurityContext result = null;
        if ((byteCtx != null) && (byteCtx.length > 0)) {
            try {
                Iterator istrs = getBytes2Strings(byteCtx).iterator();
                String principalName = (String) istrs.next();
                int nbRoles = new Integer((String) istrs.next()).intValue();
                ArrayList roles = null;
                if (nbRoles > 0) {
                    roles = new ArrayList();
                }
                while (nbRoles > 0) {
                    roles.add(istrs.next());
                    nbRoles--;
                }

                // unmarshall runAs role
                int nbRunas = new Integer((String) istrs.next()).intValue();
                ArrayList runas = null;
                if (nbRunas > 0) {
                    runas = new ArrayList();
                }
                while (nbRunas > 0) {
                    runas.add(istrs.next());
                    nbRunas--;
                }

                // unmarshall runAs principal
                int nbRunasPrincipal = new Integer((String) istrs.next()).intValue();
                ArrayList runasPrincipal = null;
                if (nbRunasPrincipal > 0) {
                    runasPrincipal = new ArrayList();
                }
                while (nbRunasPrincipal > 0) {
                    runasPrincipal.add(istrs.next());
                    nbRunasPrincipal--;
                }

                // unmarshall runAs principal to roles mapping
                int nbRunasRoles = new Integer((String) istrs.next()).intValue();
                ArrayList runasRoles = null;
                if (nbRunasRoles > 0) {
                    int arrayLength = new Integer((String) istrs.next()).intValue();

                    if (nbRunasRoles > 0) {
                        runasRoles = new ArrayList();
                    }
                    while (nbRunasRoles > 0) {
                        String[] st = new String[arrayLength];
                        for (int j = 0; j < arrayLength; j++) {
                            st[j] = (String) istrs.next();
                        }
                        runasRoles.add(st);
                        nbRunasRoles--;
                    }
                }
                result = new SecurityContext(principalName, roles, runas, runasPrincipal, runasRoles);

            } catch (java.io.UnsupportedEncodingException uee) {
                uee.printStackTrace();
            }
        }
        return result;
    }

    private static void addBytes(Collection c, String toAdd) throws java.io.UnsupportedEncodingException {
        byte [] b = toAdd.getBytes("UTF8");
        c.add(new Byte((byte) b.length));
        for (int i = 0; i < b.length; i++) {
            c.add(new Byte(b[i]));
        }
    }

    private static void addBytes(Collection c, String[] toAdd) throws java.io.UnsupportedEncodingException {
        // add length of the array
        addBytes(c, new Integer(toAdd.length).toString());
        // add each string of the array
        for (int n = 0; n < toAdd.length; n++) {
            byte [] b = toAdd[n].getBytes("UTF8");
            c.add(new Byte((byte) b.length));
            for (int i = 0; i < b.length; i++) {
                c.add(new Byte(b[i]));
            }
        }
    }

    private static ArrayList getBytes2Strings(byte[] bytes) throws java.io.UnsupportedEncodingException {
        ArrayList strs = new ArrayList();
        int index = 0;
        while (bytes.length > index) {
            int rSize = bytes[index];
            index++;
            byte[] rName = new byte [rSize];
            for (int j = 0; j < rSize; j++) {
                rName[j] = bytes[index];
                index++;
            }
            strs.add(new String(rName, "UTF8"));
        }
        return strs;
    }

}

---