MWMessagePack.php

Go to the documentation of this file.

<?php
class MWMessagePack {

    public static $bigendian = null;

    public static function pack( $value ) {
        if ( self::$bigendian === null ) {
            self::$bigendian = pack( 'S', 1 ) === pack( 'n', 1 );
        }

        switch ( gettype( $value ) ) {
        case 'NULL':
            return "\xC0";

        case 'boolean':
            return $value ? "\xC3" : "\xC2";

        case 'double':
        case 'float':
            return self::$bigendian
                ? "\xCB" . pack( 'd', $value )
                : "\xCB" . strrev( pack( 'd', $value ) );

        case 'string':
            $length = strlen( $value );
            if ( $length < 32 ) {
                return pack( 'Ca*', 0xA0 | $length, $value );
            } elseif ( $length <= 0xFFFF ) {
                return pack( 'Cna*', 0xDA, $length, $value );
            } elseif ( $length <= 0xFFFFFFFF ) {
                return pack( 'CNa*', 0xDB, $length, $value );
            }
            throw new InvalidArgumentException( __METHOD__ . ": string too long (length: $length; max: 4294967295)" );

        case 'integer':
            if ( $value >= 0 ) {
                if ( $value <= 0x7F ) {
                    // positive fixnum
                    return chr( $value );
                }
                if ( $value <= 0xFF ) {
                    // uint8
                    return pack( 'CC', 0xCC, $value );
                }
                if ( $value <= 0xFFFF ) {
                    // uint16
                    return pack( 'Cn', 0xCD, $value );
                }
                if ( $value <= 0xFFFFFFFF ) {
                    // uint32
                    return pack( 'CN', 0xCE, $value );
                }
                if ( $value <= 0xFFFFFFFFFFFFFFFF ) {
                    // uint64
                    $hi = ( $value & 0xFFFFFFFF00000000 ) >> 32;
                    $lo = $value & 0xFFFFFFFF;
                    return self::$bigendian
                        ? pack( 'CNN', 0xCF, $lo, $hi )
                        : pack( 'CNN', 0xCF, $hi, $lo );
                }
            } else {
                if ( $value >= -32 ) {
                    // negative fixnum
                    return pack( 'c', $value );
                }
                if ( $value >= -0x80 ) {
                    // int8
                    return pack( 'Cc', 0xD0, $value );
                }
                if ( $value >= -0x8000 ) {
                    // int16
                    $p = pack( 's', $value );
                    return self::$bigendian
                        ? pack( 'Ca2', 0xD1, $p )
                        : pack( 'Ca2', 0xD1, strrev( $p ) );
                }
                if ( $value >= -0x80000000 ) {
                    // int32
                    $p = pack( 'l', $value );
                    return self::$bigendian
                        ? pack( 'Ca4', 0xD2, $p )
                        : pack( 'Ca4', 0xD2, strrev( $p ) );
                }
                if ( $value >= -0x8000000000000000 ) {
                    // int64
                    // pack() does not support 64-bit ints either so pack into two 32-bits
                    $p1 = pack( 'l', $value & 0xFFFFFFFF );
                    $p2 = pack( 'l', ( $value >> 32 ) & 0xFFFFFFFF );
                    return self::$bigendian
                        ? pack( 'Ca4a4', 0xD3, $p1, $p2 )
                        : pack( 'Ca4a4', 0xD3, strrev( $p2 ), strrev( $p1 ) );
                }
            }
            throw new InvalidArgumentException( __METHOD__ . ": invalid integer '$value'" );

        case 'array':
            $buffer = '';
            $length = count( $value );
            if ( $length > 0xFFFFFFFF ) {
                throw new InvalidArgumentException( __METHOD__ . ": array too long (length: $length, max: 4294967295)" );
            }

            $index = 0;
            foreach ( $value as $k => $v ) {
                if ( $index !== $k || $index === $length ) {
                    break;
                } else {
                    $index++;
                }
            }
            $associative = $index !== $length;

            if ( $associative ) {
                if ( $length < 16 ) {
                    $buffer .= pack( 'C', 0x80 | $length );
                } elseif ( $length <= 0xFFFF ) {
                    $buffer .= pack( 'Cn', 0xDE, $length );
                } else {
                    $buffer .= pack( 'CN', 0xDF, $length );
                }
                foreach ( $value as $k => $v ) {
                    $buffer .= self::pack( $k );
                    $buffer .= self::pack( $v );
                }
            } else {
                if ( $length < 16 ) {
                    $buffer .= pack( 'C', 0x90 | $length );
                } elseif ( $length <= 0xFFFF ) {
                    $buffer .= pack( 'Cn', 0xDC, $length );
                } else {
                    $buffer .= pack( 'CN', 0xDD, $length );
                }
                foreach ( $value as $v ) {
                    $buffer .= self::pack( $v );
                }
            }
            return $buffer;

        default:
            throw new InvalidArgumentException( __METHOD__ . ': unsupported type ' . gettype( $value ) );
        }
    }
}