LyoKICogQ29weXJpZ2h0IKkgMjAwNiBJbnRlbCBDb3Jwb3JhdGlvbgogKgogKiBQZXJtaXNzaW9uIGlzIGhlcmVieSBncmFudGVkLCBmcmVlIG9mIGNoYXJnZSwgdG8gYW55IHBlcnNvbiBvYnRhaW5pbmcgYQogKiBjb3B5IG9mIHRoaXMgc29mdHdhcmUgYW5kIGFzc29jaWF0ZWQgZG9jdW1lbnRhdGlvbiBmaWxlcyAodGhlICJTb2Z0d2FyZSIpLAogKiB0byBkZWFsIGluIHRoZSBTb2Z0d2FyZSB3aXRob3V0IHJlc3RyaWN0aW9uLCBpbmNsdWRpbmcgd2l0aG91dCBsaW1pdGF0aW9uCiAqIHRoZSByaWdodHMgdG8gdXNlLCBjb3B5LCBtb2RpZnksIG1lcmdlLCBwdWJsaXNoLCBkaXN0cmlidXRlLCBzdWJsaWNlbnNlLAogKiBhbmQvb3Igc2VsbCBjb3BpZXMgb2YgdGhlIFNvZnR3YXJlLCBhbmQgdG8gcGVybWl0IHBlcnNvbnMgdG8gd2hvbSB0aGUKICogU29mdHdhcmUgaXMgZnVybmlzaGVkIHRvIGRvIHNvLCBzdWJqZWN0IHRvIHRoZSBmb2xsb3dpbmcgY29uZGl0aW9uczoKICoKICogVGhlIGFib3ZlIGNvcHlyaWdodCBub3RpY2UgYW5kIHRoaXMgcGVybWlzc2lvbiBub3RpY2UgKGluY2x1ZGluZyB0aGUgbmV4dAogKiBwYXJhZ3JhcGgpIHNoYWxsIGJlIGluY2x1ZGVkIGluIGFsbCBjb3BpZXMgb3Igc3Vic3RhbnRpYWwgcG9ydGlvbnMgb2YgdGhlCiAqIFNvZnR3YXJlLgogKgogKiBUSEUgU09GVFdBUkUgSVMgUFJPVklERUQgIkFTIElTIiwgV0lUSE9VVCBXQVJSQU5UWSBPRiBBTlkgS0lORCwgRVhQUkVTUyBPUgogKiBJTVBMSUVELCBJTkNMVURJTkcgQlVUIE5PVCBMSU1JVEVEIFRPIFRIRSBXQVJSQU5USUVTIE9GIE1FUkNIQU5UQUJJTElUWSwKICogRklUTkVTUyBGT1IgQSBQQVJUSUNVTEFSIFBVUlBPU0UgQU5EIE5PTklORlJJTkdFTUVOVC4gIElOIE5PIEVWRU5UIFNIQUxMCiAqIFRIRSBBVVRIT1JTIE9SIENPUFlSSUdIVCBIT0xERVJTIEJFIExJQUJMRSBGT1IgQU5ZIENMQUlNLCBEQU1BR0VTIE9SIE9USEVSCiAqIExJQUJJTElUWSwgV0hFVEhFUiBJTiBBTiBBQ1RJT04gT0YgQ09OVFJBQ1QsIFRPUlQgT1IgT1RIRVJXSVNFLCBBUklTSU5HIEZST00sCiAqIE9VVCBPRiBPUiBJTiBDT05ORUNUSU9OIFdJVEggVEhFIFNPRlRXQVJFIE9SIFRIRSBVU0UgT1IgT1RIRVIgREVBTElOR1MgSU4gVEhFCiAqIFNPRlRXQVJFLgogKgogKiBBdXRob3JzOgogKiAgICBFcmljIEFuaG9sdCA8ZXJpY0BhbmhvbHQubmV0PgogKgogKi8KI2luY2x1ZGUgImRybVAuaCIKI2luY2x1ZGUgImRybS5oIgojaW5jbHVkZSAiaTkxNV9kcm0uaCIKI2luY2x1ZGUgImk5MTVfZHJ2LmgiCiNpbmNsdWRlICJpbnRlbF9iaW9zLmgiCgoKc3RhdGljIHZvaWQgKgpmaW5kX3NlY3Rpb24oc3RydWN0IGJkYl9oZWFkZXIgKmJkYiwgaW50IHNlY3Rpb25faWQpCnsKCXU4ICpiYXNlID0gKHU4ICopYmRiOwoJaW50IGluZGV4ID0gMDsKCXUxNiB0b3RhbCwgY3VycmVudF9zaXplOwoJdTggY3VycmVudF9pZDsKCgkvKiBza2lwIHRvIGZpcnN0IHNlY3Rpb24gKi8KCWluZGV4ICs9IGJkYi0+aGVhZGVyX3NpemU7Cgl0b3RhbCA9IGJkYi0+YmRiX3NpemU7CgoJLyogd2FsayB0aGUgc2VjdGlvbnMgbG9va2luZyBmb3Igc2VjdGlvbl9pZCAqLwoJd2hpbGUgKGluZGV4IDwgdG90YWwpIHsKCQljdXJyZW50X2lkID0gKihiYXNlICsgaW5kZXgpOwoJCWluZGV4Kys7CgkJY3VycmVudF9zaXplID0gKigodTE2ICopKGJhc2UgKyBpbmRleCkpOwoJCWluZGV4ICs9IDI7CgkJaWYgKGN1cnJlbnRfaWQgPT0gc2VjdGlvbl9pZCkKCQkJcmV0dXJuIGJhc2UgKyBpbmRleDsKCQlpbmRleCArPSBjdXJyZW50X3NpemU7Cgl9CgoJcmV0dXJuIE5VTEw7Cn0KCnN0YXRpYyB2b2lkCmZpbGxfZGV0YWlsX3RpbWluZ19kYXRhKHN0cnVjdCBkcm1fZGlzcGxheV9tb2RlICpwYW5lbF9maXhlZF9tb2RlLAoJCQlzdHJ1Y3QgbHZkc19kdm9fdGltaW5nICpkdm9fdGltaW5nKQp7CglwYW5lbF9maXhlZF9tb2RlLT5oZGlzcGxheSA9IChkdm9fdGltaW5nLT5oYWN0aXZlX2hpIDw8IDgpIHwKCQlkdm9fdGltaW5nLT5oYWN0aXZlX2xvOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfc3RhcnQgPSBwYW5lbF9maXhlZF9tb2RlLT5oZGlzcGxheSArCgkJKChkdm9fdGltaW5nLT5oc3luY19vZmZfaGkgPDwgOCkgfCBkdm9fdGltaW5nLT5oc3luY19vZmZfbG8pOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfZW5kID0gcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfc3RhcnQgKwoJCWR2b190aW1pbmctPmhzeW5jX3B1bHNlX3dpZHRoOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+aGRpc3BsYXkgKwoJCSgoZHZvX3RpbWluZy0+aGJsYW5rX2hpIDw8IDgpIHwgZHZvX3RpbWluZy0+aGJsYW5rX2xvKTsKCglwYW5lbF9maXhlZF9tb2RlLT52ZGlzcGxheSA9IChkdm9fdGltaW5nLT52YWN0aXZlX2hpIDw8IDgpIHwKCQlkdm9fdGltaW5nLT52YWN0aXZlX2xvOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfc3RhcnQgPSBwYW5lbF9maXhlZF9tb2RlLT52ZGlzcGxheSArCgkJZHZvX3RpbWluZy0+dnN5bmNfb2ZmOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfZW5kID0gcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfc3RhcnQgKwoJCWR2b190aW1pbmctPnZzeW5jX3B1bHNlX3dpZHRoOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+dmRpc3BsYXkgKwoJCSgoZHZvX3RpbWluZy0+dmJsYW5rX2hpIDw8IDgpIHwgZHZvX3RpbWluZy0+dmJsYW5rX2xvKTsKCXBhbmVsX2ZpeGVkX21vZGUtPmNsb2NrID0gZHZvX3RpbWluZy0+Y2xvY2sgKiAxMDsKCXBhbmVsX2ZpeGVkX21vZGUtPnR5cGUgPSBEUk1fTU9ERV9UWVBFX1BSRUZFUlJFRDsKCgkvKiBTb21lIFZCVHMgaGF2ZSBib2d1cyBoL3Z0b3RhbCB2YWx1ZXMgKi8KCWlmIChwYW5lbF9maXhlZF9tb2RlLT5oc3luY19lbmQgPiBwYW5lbF9maXhlZF9tb2RlLT5odG90YWwpCgkJcGFuZWxfZml4ZWRfbW9kZS0+aHRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfZW5kICsgMTsKCWlmIChwYW5lbF9maXhlZF9tb2RlLT52c3luY19lbmQgPiBwYW5lbF9maXhlZF9tb2RlLT52dG90YWwpCgkJcGFuZWxfZml4ZWRfbW9kZS0+dnRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfZW5kICsgMTsKCglkcm1fbW9kZV9zZXRfbmFtZShwYW5lbF9maXhlZF9tb2RlKTsKfQoKLyogVHJ5IHRvIGZpbmQgaW50ZWdyYXRlZCBwYW5lbCBkYXRhICovCnN0YXRpYyB2b2lkCnBhcnNlX2xmcF9wYW5lbF9kYXRhKHN0cnVjdCBkcm1faTkxNV9wcml2YXRlICpkZXZfcHJpdiwKCQkJICAgIHN0cnVjdCBiZGJfaGVhZGVyICpiZGIpCnsKCXN0cnVjdCBiZGJfbHZkc19vcHRpb25zICpsdmRzX29wdGlvbnM7CglzdHJ1Y3QgYmRiX2x2ZHNfbGZwX2RhdGEgKmx2ZHNfbGZwX2RhdGE7CglzdHJ1Y3QgYmRiX2x2ZHNfbGZwX2RhdGFfZW50cnkgKmVudHJ5OwoJc3RydWN0IGx2ZHNfZHZvX3RpbWluZyAqZHZvX3RpbWluZzsKCXN0cnVjdCBkcm1fZGlzcGxheV9tb2RlICpwYW5lbF9maXhlZF9tb2RlOwoKCS8qIERlZmF1bHRzIGlmIHdlIGNhbid0IGZpbmQgVkJUIGluZm8gKi8KCWRldl9wcml2LT5sdmRzX2RpdGhlciA9IDA7CglkZXZfcHJpdi0+bHZkc192YnQgPSAwOwoKCWx2ZHNfb3B0aW9ucyA9IGZpbmRfc2VjdGlvbihiZGIsIEJEQl9MVkRTX09QVElPTlMpOwoJaWYgKCFsdmRzX29wdGlvbnMpCgkJcmV0dXJuOwoKCWRldl9wcml2LT5sdmRzX2RpdGhlciA9IGx2ZHNfb3B0aW9ucy0+cGl4ZWxfZGl0aGVyOwoJaWYgKGx2ZHNfb3B0aW9ucy0+cGFuZWxfdHlwZSA9PSAweGZmKQoJCXJldHVybjsKCglsdmRzX2xmcF9kYXRhID0gZmluZF9zZWN0aW9uKGJkYiwgQkRCX0xWRFNfTEZQX0RBVEEpOwoJaWYgKCFsdmRzX2xmcF9kYXRhKQoJCXJldHVybjsKCglkZXZfcHJpdi0+bHZkc192YnQgPSAxOwoKCWVudHJ5ID0gJmx2ZHNfbGZwX2RhdGEtPmRhdGFbbHZkc19vcHRpb25zLT5wYW5lbF90eXBlXTsKCWR2b190aW1pbmcgPSAmZW50cnktPmR2b190aW1pbmc7CgoJcGFuZWxfZml4ZWRfbW9kZSA9IGRybV9jYWxsb2MoMSwgc2l6ZW9mKCpwYW5lbF9maXhlZF9tb2RlKSwKCQkJCSAgICAgIERSTV9NRU1fRFJJVkVSKTsKCglmaWxsX2RldGFpbF90aW1pbmdfZGF0YShwYW5lbF9maXhlZF9tb2RlLCBkdm9fdGltaW5nKTsKCglkZXZfcHJpdi0+bGZwX2x2ZHNfdmJ0X21vZGUgPSBwYW5lbF9maXhlZF9tb2RlOwoKCURSTV9ERUJVRygiRm91bmQgcGFuZWwgbW9kZSBpbiBCSU9TIFZCVCB0YWJsZXM6XG4iKTsKCWRybV9tb2RlX2RlYnVnX3ByaW50bW9kZWxpbmUocGFuZWxfZml4ZWRfbW9kZSk7CgoJcmV0dXJuOwp9CgovKiBUcnkgdG8gZmluZCBzZHZvIHBhbmVsIGRhdGEgKi8Kc3RhdGljIHZvaWQKcGFyc2Vfc2R2b19wYW5lbF9kYXRhKHN0cnVjdCBkcm1faTkxNV9wcml2YXRlICpkZXZfcHJpdiwKCQkgICAgICBzdHJ1Y3QgYmRiX2hlYWRlciAqYmRiKQp7CglzdHJ1Y3QgYmRiX3Nkdm9fbHZkc19vcHRpb25zICpzZHZvX2x2ZHNfb3B0aW9uczsKCXN0cnVjdCBsdmRzX2R2b190aW1pbmcgKmR2b190aW1pbmc7CglzdHJ1Y3QgZHJtX2Rpc3BsYXlfbW9kZSAqcGFuZWxfZml4ZWRfbW9kZTsKCglkZXZfcHJpdi0+c2R2b19sdmRzX3ZidF9tb2RlID0gTlVMTDsKCglzZHZvX2x2ZHNfb3B0aW9ucyA9IGZpbmRfc2VjdGlvbihiZGIsIEJEQl9TRFZPX0xWRFNfT1BUSU9OUyk7CglpZiAoIXNkdm9fbHZkc19vcHRpb25zKQoJCXJldHVybjsKCglkdm9fdGltaW5nID0gZmluZF9zZWN0aW9uKGJkYiwgQkRCX1NEVk9fUEFORUxfRFREUyk7CglpZiAoIWR2b190aW1pbmcpCgkJcmV0dXJuOwoKCXBhbmVsX2ZpeGVkX21vZGUgPSBkcm1fY2FsbG9jKDEsIHNpemVvZigqcGFuZWxfZml4ZWRfbW9kZSksCgkJCQkgICAgICBEUk1fTUVNX0RSSVZFUik7CgoJaWYgKCFwYW5lbF9maXhlZF9tb2RlKQoJCXJldHVybjsKCglmaWxsX2RldGFpbF90aW1pbmdfZGF0YShwYW5lbF9maXhlZF9tb2RlLAoJCQlkdm9fdGltaW5nICsgc2R2b19sdmRzX29wdGlvbnMtPnBhbmVsX3R5cGUpOwoKCWRldl9wcml2LT5zZHZvX2x2ZHNfdmJ0X21vZGUgPSBwYW5lbF9maXhlZF9tb2RlOwoKCXJldHVybjsKfQoKc3RhdGljIHZvaWQKcGFyc2VfZ2VuZXJhbF9mZWF0dXJlcyhzdHJ1Y3QgZHJtX2k5MTVfcHJpdmF0ZSAqZGV2X3ByaXYsCgkJICAgICAgIHN0cnVjdCBiZGJfaGVhZGVyICpiZGIpCnsKCXN0cnVjdCBiZGJfZ2VuZXJhbF9mZWF0dXJlcyAqZ2VuZXJhbDsKCgkvKiBTZXQgc2Vuc2libGUgZGVmYXVsdHMgaW4gY2FzZSB3ZSBjYW4ndCBmaW5kIHRoZSBnZW5lcmFsIGJsb2NrICovCglkZXZfcHJpdi0+aW50X3R2X3N1cHBvcnQgPSAxOwoJZGV2X3ByaXYtPmludF9jcnRfc3VwcG9ydCA9IDE7CgoJZ2VuZXJhbCA9IGZpbmRfc2VjdGlvbihiZGIsIEJEQl9HRU5FUkFMX0ZFQVRVUkVTKTsKCWlmIChnZW5lcmFsKSB7CgkJZGV2X3ByaXYtPmludF90dl9zdXBwb3J0ID0gZ2VuZXJhbC0+aW50X3R2X3N1cHBvcnQ7CgkJZGV2X3ByaXYtPmludF9jcnRfc3VwcG9ydCA9IGdlbmVyYWwtPmludF9jcnRfc3VwcG9ydDsKCQlkZXZfcHJpdi0+bHZkc191c2Vfc3NjID0gZ2VuZXJhbC0+ZW5hYmxlX3NzYzsKCgkJaWYgKGRldl9wcml2LT5sdmRzX3VzZV9zc2MpIHsKCQkgIGlmIChJU19JODU1KGRldl9wcml2LT5kZXYpKQoJCSAgICBkZXZfcHJpdi0+bHZkc19zc2NfZnJlcSA9IGdlbmVyYWwtPnNzY19mcmVxID8gNjYgOiA0ODsKCQkgIGVsc2UKCQkgICAgZGV2X3ByaXYtPmx2ZHNfc3NjX2ZyZXEgPSBnZW5lcmFsLT5zc2NfZnJlcSA/IDEwMCA6IDk2OwoJCX0KCX0KfQoKLyoqCiAqIGludGVsX2luaXRfYmlvcyAtIGluaXRpYWxpemUgVkJJT1Mgc2V0dGluZ3MgJiBmaW5kIFZCVAogKiBAZGV2OiBEUk0gZGV2aWNlCiAqCiAqIExvYWRzIHRoZSBWaWRlbyBCSU9TIGFuZCBjaGVja3MgdGhhdCB0aGUgVkJUIGV4aXN0cy4gIFNldHMgc2NyYXRjaCByZWdpc3RlcnMKICogdG8gYXBwcm9wcmlhdGUgdmFsdWVzLgogKgogKiBWQlQgZXhpc3RlbmNlIGlzIGEgc2FuaXR5IGNoZWNrIHRoYXQgaXMgcmVsaWVkIG9uIGJ5IG90aGVyIGk4MzBfYmlvcy5jIGNvZGUuCiAqIE5vdGUgdGhhdCBpdCB3b3VsZCBiZSBiZXR0ZXIgdG8gdXNlIGEgQklPUyBjYWxsIHRvIGdldCB0aGUgVkJULCBhcyBCSU9TZXMgbWF5CiAqIGZlZWQgYW4gdXBkYXRlZCBWQlQgYmFjayB0aHJvdWdoIHRoYXQsIGNvbXBhcmVkIHRvIHdoYXQgd2UnbGwgZmV0Y2ggdXNpbmcKICogdGhpcyBtZXRob2Qgb2YgZ3JvcGluZyBhcm91bmQgaW4gdGhlIEJJT1MgZGF0YS4KICoKICogUmV0dXJucyAwIG9uIHN1Y2Nlc3MsIG5vbnplcm8gb24gZmFpbHVyZS4KICovCmJvb2wKaW50ZWxfaW5pdF9iaW9zKHN0cnVjdCBkcm1fZGV2aWNlICpkZXYpCnsKCXN0cnVjdCBkcm1faTkxNV9wcml2YXRlICpkZXZfcHJpdiA9IGRldi0+ZGV2X3ByaXZhdGU7CglzdHJ1Y3QgcGNpX2RldiAqcGRldiA9IGRldi0+cGRldjsKCXN0cnVjdCB2YnRfaGVhZGVyICp2YnQgPSBOVUxMOwoJc3RydWN0IGJkYl9oZWFkZXIgKmJkYjsKCXU4IF9faW9tZW0gKmJpb3M7CglzaXplX3Qgc2l6ZTsKCWludCBpOwoKCWJpb3MgPSBwY2lfbWFwX3JvbShwZGV2LCAmc2l6ZSk7CglpZiAoIWJpb3MpCgkJcmV0dXJuIC0xOwoKCS8qIFNjb3VyIG1lbW9yeSBsb29raW5nIGZvciB0aGUgVkJUIHNpZ25hdHVyZSAqLwoJZm9yIChpID0gMDsgaSArIDQgPCBzaXplOyBpKyspIHsKCQlpZiAoIW1lbWNtcChiaW9zICsgaSwgIiRWQlQiLCA0KSkgewoJCQl2YnQgPSAoc3RydWN0IHZidF9oZWFkZXIgKikoYmlvcyArIGkpOwoJCQlicmVhazsKCQl9Cgl9CgoJaWYgKCF2YnQpIHsKCQlEUk1fRVJST1IoIlZCVCBzaWduYXR1cmUgbWlzc2luZ1xuIik7CgkJcGNpX3VubWFwX3JvbShwZGV2LCBiaW9zKTsKCQlyZXR1cm4gLTE7Cgl9CgoJYmRiID0gKHN0cnVjdCBiZGJfaGVhZGVyICopKGJpb3MgKyBpICsgdmJ0LT5iZGJfb2Zmc2V0KTsKCgkvKiBHcmFiIHVzZWZ1bCBnZW5lcmFsIGRlZmluaXRpb25zICovCglwYXJzZV9nZW5lcmFsX2ZlYXR1cmVzKGRldl9wcml2LCBiZGIpOwoJcGFyc2VfbGZwX3BhbmVsX2RhdGEoZGV2X3ByaXYsIGJkYik7CglwYXJzZV9zZHZvX3BhbmVsX2RhdGEoZGV2X3ByaXYsIGJkYik7CgoJcGNpX3VubWFwX3JvbShwZGV2LCBiaW9zKTsKCglyZXR1cm4gMDsKfQo=