LyoKICogQ29weXJpZ2h0IKkgMjAwNiBJbnRlbCBDb3Jwb3JhdGlvbgogKgogKiBQZXJtaXNzaW9uIGlzIGhlcmVieSBncmFudGVkLCBmcmVlIG9mIGNoYXJnZSwgdG8gYW55IHBlcnNvbiBvYnRhaW5pbmcgYQogKiBjb3B5IG9mIHRoaXMgc29mdHdhcmUgYW5kIGFzc29jaWF0ZWQgZG9jdW1lbnRhdGlvbiBmaWxlcyAodGhlICJTb2Z0d2FyZSIpLAogKiB0byBkZWFsIGluIHRoZSBTb2Z0d2FyZSB3aXRob3V0IHJlc3RyaWN0aW9uLCBpbmNsdWRpbmcgd2l0aG91dCBsaW1pdGF0aW9uCiAqIHRoZSByaWdodHMgdG8gdXNlLCBjb3B5LCBtb2RpZnksIG1lcmdlLCBwdWJsaXNoLCBkaXN0cmlidXRlLCBzdWJsaWNlbnNlLAogKiBhbmQvb3Igc2VsbCBjb3BpZXMgb2YgdGhlIFNvZnR3YXJlLCBhbmQgdG8gcGVybWl0IHBlcnNvbnMgdG8gd2hvbSB0aGUKICogU29mdHdhcmUgaXMgZnVybmlzaGVkIHRvIGRvIHNvLCBzdWJqZWN0IHRvIHRoZSBmb2xsb3dpbmcgY29uZGl0aW9uczoKICoKICogVGhlIGFib3ZlIGNvcHlyaWdodCBub3RpY2UgYW5kIHRoaXMgcGVybWlzc2lvbiBub3RpY2UgKGluY2x1ZGluZyB0aGUgbmV4dAogKiBwYXJhZ3JhcGgpIHNoYWxsIGJlIGluY2x1ZGVkIGluIGFsbCBjb3BpZXMgb3Igc3Vic3RhbnRpYWwgcG9ydGlvbnMgb2YgdGhlCiAqIFNvZnR3YXJlLgogKgogKiBUSEUgU09GVFdBUkUgSVMgUFJPVklERUQgIkFTIElTIiwgV0lUSE9VVCBXQVJSQU5UWSBPRiBBTlkgS0lORCwgRVhQUkVTUyBPUgogKiBJTVBMSUVELCBJTkNMVURJTkcgQlVUIE5PVCBMSU1JVEVEIFRPIFRIRSBXQVJSQU5USUVTIE9GIE1FUkNIQU5UQUJJTElUWSwKICogRklUTkVTUyBGT1IgQSBQQVJUSUNVTEFSIFBVUlBPU0UgQU5EIE5PTklORlJJTkdFTUVOVC4gIElOIE5PIEVWRU5UIFNIQUxMCiAqIFRIRSBBVVRIT1JTIE9SIENPUFlSSUdIVCBIT0xERVJTIEJFIExJQUJMRSBGT1IgQU5ZIENMQUlNLCBEQU1BR0VTIE9SIE9USEVSCiAqIExJQUJJTElUWSwgV0hFVEhFUiBJTiBBTiBBQ1RJT04gT0YgQ09OVFJBQ1QsIFRPUlQgT1IgT1RIRVJXSVNFLCBBUklTSU5HIEZST00sCiAqIE9VVCBPRiBPUiBJTiBDT05ORUNUSU9OIFdJVEggVEhFIFNPRlRXQVJFIE9SIFRIRSBVU0UgT1IgT1RIRVIgREVBTElOR1MgSU4gVEhFCiAqIFNPRlRXQVJFLgogKgogKiBBdXRob3JzOgogKiAgICBFcmljIEFuaG9sdCA8ZXJpY0BhbmhvbHQubmV0PgogKgogKi8KI2luY2x1ZGUgImRybVAuaCIKI2luY2x1ZGUgImRybS5oIgojaW5jbHVkZSAiaTkxNV9kcm0uaCIKI2luY2x1ZGUgImk5MTVfZHJ2LmgiCiNpbmNsdWRlICJpbnRlbF9iaW9zLmgiCgoKc3RhdGljIHZvaWQgKgpmaW5kX3NlY3Rpb24oc3RydWN0IGJkYl9oZWFkZXIgKmJkYiwgaW50IHNlY3Rpb25faWQpCnsKCXU4ICpiYXNlID0gKHU4ICopYmRiOwoJaW50IGluZGV4ID0gMDsKCXUxNiB0b3RhbCwgY3VycmVudF9zaXplOwoJdTggY3VycmVudF9pZDsKCgkvKiBza2lwIHRvIGZpcnN0IHNlY3Rpb24gKi8KCWluZGV4ICs9IGJkYi0+aGVhZGVyX3NpemU7Cgl0b3RhbCA9IGJkYi0+YmRiX3NpemU7CgoJLyogd2FsayB0aGUgc2VjdGlvbnMgbG9va2luZyBmb3Igc2VjdGlvbl9pZCAqLwoJd2hpbGUgKGluZGV4IDwgdG90YWwpIHsKCQljdXJyZW50X2lkID0gKihiYXNlICsgaW5kZXgpOwoJCWluZGV4Kys7CgkJY3VycmVudF9zaXplID0gKigodTE2ICopKGJhc2UgKyBpbmRleCkpOwoJCWluZGV4ICs9IDI7CgkJaWYgKGN1cnJlbnRfaWQgPT0gc2VjdGlvbl9pZCkKCQkJcmV0dXJuIGJhc2UgKyBpbmRleDsKCQlpbmRleCArPSBjdXJyZW50X3NpemU7Cgl9CgoJcmV0dXJuIE5VTEw7Cn0KCi8qIFRyeSB0byBmaW5kIHBhbmVsIGRhdGEgKi8Kc3RhdGljIHZvaWQKcGFyc2VfcGFuZWxfZGF0YShzdHJ1Y3QgZHJtX2k5MTVfcHJpdmF0ZSAqZGV2X3ByaXYsIHN0cnVjdCBiZGJfaGVhZGVyICpiZGIpCnsKCXN0cnVjdCBiZGJfbHZkc19vcHRpb25zICpsdmRzX29wdGlvbnM7CglzdHJ1Y3QgYmRiX2x2ZHNfbGZwX2RhdGEgKmx2ZHNfbGZwX2RhdGE7CglzdHJ1Y3QgYmRiX2x2ZHNfbGZwX2RhdGFfZW50cnkgKmVudHJ5OwoJc3RydWN0IGx2ZHNfZHZvX3RpbWluZyAqZHZvX3RpbWluZzsKCXN0cnVjdCBkcm1fZGlzcGxheV9tb2RlICpwYW5lbF9maXhlZF9tb2RlOwoKCS8qIERlZmF1bHRzIGlmIHdlIGNhbid0IGZpbmQgVkJUIGluZm8gKi8KCWRldl9wcml2LT5sdmRzX2RpdGhlciA9IDA7CglkZXZfcHJpdi0+bHZkc192YnQgPSAwOwoKCWx2ZHNfb3B0aW9ucyA9IGZpbmRfc2VjdGlvbihiZGIsIEJEQl9MVkRTX09QVElPTlMpOwoJaWYgKCFsdmRzX29wdGlvbnMpCgkJcmV0dXJuOwoKCWRldl9wcml2LT5sdmRzX2RpdGhlciA9IGx2ZHNfb3B0aW9ucy0+cGl4ZWxfZGl0aGVyOwoJaWYgKGx2ZHNfb3B0aW9ucy0+cGFuZWxfdHlwZSA9PSAweGZmKQoJCXJldHVybjsKCglsdmRzX2xmcF9kYXRhID0gZmluZF9zZWN0aW9uKGJkYiwgQkRCX0xWRFNfTEZQX0RBVEEpOwoJaWYgKCFsdmRzX2xmcF9kYXRhKQoJCXJldHVybjsKCglkZXZfcHJpdi0+bHZkc192YnQgPSAxOwoKCWVudHJ5ID0gJmx2ZHNfbGZwX2RhdGEtPmRhdGFbbHZkc19vcHRpb25zLT5wYW5lbF90eXBlXTsKCWR2b190aW1pbmcgPSAmZW50cnktPmR2b190aW1pbmc7CgoJcGFuZWxfZml4ZWRfbW9kZSA9IGRybV9jYWxsb2MoMSwgc2l6ZW9mKCpwYW5lbF9maXhlZF9tb2RlKSwKCQkJCSAgICAgIERSTV9NRU1fRFJJVkVSKTsKCglwYW5lbF9maXhlZF9tb2RlLT5oZGlzcGxheSA9IChkdm9fdGltaW5nLT5oYWN0aXZlX2hpIDw8IDgpIHwKCQlkdm9fdGltaW5nLT5oYWN0aXZlX2xvOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfc3RhcnQgPSBwYW5lbF9maXhlZF9tb2RlLT5oZGlzcGxheSArCgkJKChkdm9fdGltaW5nLT5oc3luY19vZmZfaGkgPDwgOCkgfCBkdm9fdGltaW5nLT5oc3luY19vZmZfbG8pOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfZW5kID0gcGFuZWxfZml4ZWRfbW9kZS0+aHN5bmNfc3RhcnQgKwoJCWR2b190aW1pbmctPmhzeW5jX3B1bHNlX3dpZHRoOwoJcGFuZWxfZml4ZWRfbW9kZS0+aHRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+aGRpc3BsYXkgKwoJCSgoZHZvX3RpbWluZy0+aGJsYW5rX2hpIDw8IDgpIHwgZHZvX3RpbWluZy0+aGJsYW5rX2xvKTsKCglwYW5lbF9maXhlZF9tb2RlLT52ZGlzcGxheSA9IChkdm9fdGltaW5nLT52YWN0aXZlX2hpIDw8IDgpIHwKCQlkdm9fdGltaW5nLT52YWN0aXZlX2xvOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfc3RhcnQgPSBwYW5lbF9maXhlZF9tb2RlLT52ZGlzcGxheSArCgkJZHZvX3RpbWluZy0+dnN5bmNfb2ZmOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfZW5kID0gcGFuZWxfZml4ZWRfbW9kZS0+dnN5bmNfc3RhcnQgKwoJCWR2b190aW1pbmctPnZzeW5jX3B1bHNlX3dpZHRoOwoJcGFuZWxfZml4ZWRfbW9kZS0+dnRvdGFsID0gcGFuZWxfZml4ZWRfbW9kZS0+dmRpc3BsYXkgKwoJCSgoZHZvX3RpbWluZy0+dmJsYW5rX2hpIDw8IDgpIHwgZHZvX3RpbWluZy0+dmJsYW5rX2xvKTsKCXBhbmVsX2ZpeGVkX21vZGUtPmNsb2NrID0gZHZvX3RpbWluZy0+Y2xvY2sgKiAxMDsKCXBhbmVsX2ZpeGVkX21vZGUtPnR5cGUgPSBEUk1fTU9ERV9UWVBFX1BSRUZFUlJFRDsKCglkcm1fbW9kZV9zZXRfbmFtZShwYW5lbF9maXhlZF9tb2RlKTsKCglkZXZfcHJpdi0+dmJ0X21vZGUgPSBwYW5lbF9maXhlZF9tb2RlOwoKCURSTV9ERUJVRygiRm91bmQgcGFuZWwgbW9kZSBpbiBCSU9TIFZCVCB0YWJsZXM6XG4iKTsKCWRybV9tb2RlX2RlYnVnX3ByaW50bW9kZWxpbmUocGFuZWxfZml4ZWRfbW9kZSk7CgoJcmV0dXJuOwp9CgpzdGF0aWMgdm9pZApwYXJzZV9nZW5lcmFsX2ZlYXR1cmVzKHN0cnVjdCBkcm1faTkxNV9wcml2YXRlICpkZXZfcHJpdiwKCQkgICAgICAgc3RydWN0IGJkYl9oZWFkZXIgKmJkYikKewoJc3RydWN0IGJkYl9nZW5lcmFsX2ZlYXR1cmVzICpnZW5lcmFsOwoKCS8qIFNldCBzZW5zaWJsZSBkZWZhdWx0cyBpbiBjYXNlIHdlIGNhbid0IGZpbmQgdGhlIGdlbmVyYWwgYmxvY2sgKi8KCWRldl9wcml2LT5pbnRfdHZfc3VwcG9ydCA9IDE7CglkZXZfcHJpdi0+aW50X2NydF9zdXBwb3J0ID0gMTsKCglnZW5lcmFsID0gZmluZF9zZWN0aW9uKGJkYiwgQkRCX0dFTkVSQUxfRkVBVFVSRVMpOwoJaWYgKGdlbmVyYWwpIHsKCQlkZXZfcHJpdi0+aW50X3R2X3N1cHBvcnQgPSBnZW5lcmFsLT5pbnRfdHZfc3VwcG9ydDsKCQlkZXZfcHJpdi0+aW50X2NydF9zdXBwb3J0ID0gZ2VuZXJhbC0+aW50X2NydF9zdXBwb3J0OwoJfQp9CgovKioKICogaW50ZWxfaW5pdF9iaW9zIC0gaW5pdGlhbGl6ZSBWQklPUyBzZXR0aW5ncyAmIGZpbmQgVkJUCiAqIEBkZXY6IERSTSBkZXZpY2UKICoKICogTG9hZHMgdGhlIFZpZGVvIEJJT1MgYW5kIGNoZWNrcyB0aGF0IHRoZSBWQlQgZXhpc3RzLiAgU2V0cyBzY3JhdGNoIHJlZ2lzdGVycwogKiB0byBhcHByb3ByaWF0ZSB2YWx1ZXMuCiAqCiAqIFZCVCBleGlzdGVuY2UgaXMgYSBzYW5pdHkgY2hlY2sgdGhhdCBpcyByZWxpZWQgb24gYnkgb3RoZXIgaTgzMF9iaW9zLmMgY29kZS4KICogTm90ZSB0aGF0IGl0IHdvdWxkIGJlIGJldHRlciB0byB1c2UgYSBCSU9TIGNhbGwgdG8gZ2V0IHRoZSBWQlQsIGFzIEJJT1NlcyBtYXkKICogZmVlZCBhbiB1cGRhdGVkIFZCVCBiYWNrIHRocm91Z2ggdGhhdCwgY29tcGFyZWQgdG8gd2hhdCB3ZSdsbCBmZXRjaCB1c2luZwogKiB0aGlzIG1ldGhvZCBvZiBncm9waW5nIGFyb3VuZCBpbiB0aGUgQklPUyBkYXRhLgogKgogKiBSZXR1cm5zIDAgb24gc3VjY2Vzcywgbm9uemVybyBvbiBmYWlsdXJlLgogKi8KYm9vbAppbnRlbF9pbml0X2Jpb3Moc3RydWN0IGRybV9kZXZpY2UgKmRldikKewoJc3RydWN0IGRybV9pOTE1X3ByaXZhdGUgKmRldl9wcml2ID0gZGV2LT5kZXZfcHJpdmF0ZTsKCXN0cnVjdCBwY2lfZGV2ICpwZGV2ID0gZGV2LT5wZGV2OwoJc3RydWN0IHZidF9oZWFkZXIgKnZidCA9IE5VTEw7CglzdHJ1Y3QgYmRiX2hlYWRlciAqYmRiOwoJdTggX19pb21lbSAqYmlvczsKCXNpemVfdCBzaXplOwoJaW50IGk7CgoJYmlvcyA9IHBjaV9tYXBfcm9tKHBkZXYsICZzaXplKTsKCWlmICghYmlvcykKCQlyZXR1cm4gLTE7CgoJLyogU2NvdXIgbWVtb3J5IGxvb2tpbmcgZm9yIHRoZSBWQlQgc2lnbmF0dXJlICovCglmb3IgKGkgPSAwOyBpICsgNCA8IHNpemU7IGkrKykgewoJCWlmICghbWVtY21wKGJpb3MgKyBpLCAiJFZCVCIsIDQpKSB7CgkJCXZidCA9IChzdHJ1Y3QgdmJ0X2hlYWRlciAqKShiaW9zICsgaSk7CgkJCWJyZWFrOwoJCX0KCX0KCglpZiAoIXZidCkgewoJCURSTV9FUlJPUigiVkJUIHNpZ25hdHVyZSBtaXNzaW5nXG4iKTsKCQlwY2lfdW5tYXBfcm9tKHBkZXYsIGJpb3MpOwoJCXJldHVybiAtMTsKCX0KCgliZGIgPSAoc3RydWN0IGJkYl9oZWFkZXIgKikoYmlvcyArIGkgKyB2YnQtPmJkYl9vZmZzZXQpOwoKCS8qIEdyYWIgdXNlZnVsIGdlbmVyYWwgZGVmaW5pdGlvbnMgKi8KCXBhcnNlX2dlbmVyYWxfZmVhdHVyZXMoZGV2X3ByaXYsIGJkYik7CglwYXJzZV9wYW5lbF9kYXRhKGRldl9wcml2LCBiZGIpOwoKCXBjaV91bm1hcF9yb20ocGRldiwgYmlvcyk7CgoJcmV0dXJuIDA7Cn0K