LyoKICoKICogQ29weXJpZ2h0IChDKSAyMDAwICBSb2JlcnQgT2xzc29uLgogKiBTd2VkaXNoIFVuaXZlcnNpdHkgb2YgQWdyaWN1bHR1cmFsIFNjaWVuY2VzCiAqCiAqIFRoaXMgZmlsZSBpcyBwYXJ0IG9mIEdOVSBaZWJyYS4KICoKICogR05VIFplYnJhIGlzIGZyZWUgc29mdHdhcmU7IHlvdSBjYW4gcmVkaXN0cmlidXRlIGl0IGFuZC9vciBtb2RpZnkgaXQKICogdW5kZXIgdGhlIHRlcm1zIG9mIHRoZSBHTlUgR2VuZXJhbCBQdWJsaWMgTGljZW5zZSBhcyBwdWJsaXNoZWQgYnkgdGhlCiAqIEZyZWUgU29mdHdhcmUgRm91bmRhdGlvbjsgZWl0aGVyIHZlcnNpb24gMiwgb3IgKGF0IHlvdXIgb3B0aW9uKSBhbnkKICogbGF0ZXIgdmVyc2lvbi4KICoKICogR05VIFplYnJhIGlzIGRpc3RyaWJ1dGVkIGluIHRoZSBob3BlIHRoYXQgaXQgd2lsbCBiZSB1c2VmdWwsIGJ1dAogKiBXSVRIT1VUIEFOWSBXQVJSQU5UWTsgd2l0aG91dCBldmVuIHRoZSBpbXBsaWVkIHdhcnJhbnR5IG9mCiAqIE1FUkNIQU5UQUJJTElUWSBvciBGSVRORVNTIEZPUiBBIFBBUlRJQ1VMQVIgUFVSUE9TRS4gIFNlZSB0aGUgR05VCiAqIEdlbmVyYWwgUHVibGljIExpY2Vuc2UgZm9yIG1vcmUgZGV0YWlscy4KICoKICogWW91IHNob3VsZCBoYXZlIHJlY2VpdmVkIGEgY29weSBvZiB0aGUgR05VIEdlbmVyYWwgUHVibGljIExpY2Vuc2UKICogYWxvbmcgd2l0aCBHTlUgWmVicmE7IHNlZSB0aGUgZmlsZSBDT1BZSU5HLiAgSWYgbm90LCB3cml0ZSB0byB0aGUgRnJlZQogKiBTb2Z0d2FyZSBGb3VuZGF0aW9uLCBJbmMuLCA1OSBUZW1wbGUgUGxhY2UgLSBTdWl0ZSAzMzAsIEJvc3RvbiwgTUEKICogMDIxMTEtMTMwNywgVVNBLiAgCiAqLwoKLyogCiAqIFRoaXMgd29yayBpbmNsdWRlcyB3b3JrIHdpdGggdGhlIGZvbGxvd2luZyBjb3B5d3JpdGU6CiAqCiAqIENvcHlyaWdodCAoQykgMTk5NywgMjAwMCBLdW5paGlybyBJc2hpZ3VybwogKgogKi8KCi8qIAogKiBUaGFua3MgdG8gSmVucyBM5eVzIGF0IFN3ZWRpc2ggVW5pdmVyc2l0eSBvZiBBZ3JpY3VsdHVyYWwgU2NpZW5jZXMKICogZm9yIHJldmlld2luZyBhbmQgdGVzdHMuCiAqLwoKCiNpbmNsdWRlIDx6ZWJyYS5oPgoKI2lmZGVmIEhBVkVfSVJEUAoKI2luY2x1ZGUgImlmLmgiCiNpbmNsdWRlICJ2dHkuaCIKI2luY2x1ZGUgInNvY2t1bmlvbi5oIgojaW5jbHVkZSAic29ja29wdC5oIgojaW5jbHVkZSAicHJlZml4LmgiCiNpbmNsdWRlICJjb21tYW5kLmgiCiNpbmNsdWRlICJtZW1vcnkuaCIKI2luY2x1ZGUgInN0cmVhbS5oIgojaW5jbHVkZSAiaW9jdGwuaCIKI2luY2x1ZGUgImNvbm5lY3RlZC5oIgojaW5jbHVkZSAibG9nLmgiCiNpbmNsdWRlICJ6Y2xpZW50LmgiCiNpbmNsdWRlICJ0aHJlYWQuaCIKI2luY2x1ZGUgInByaXZzLmgiCiNpbmNsdWRlICJ6ZWJyYS9pbnRlcmZhY2UuaCIKI2luY2x1ZGUgInplYnJhL3J0YWR2LmgiCiNpbmNsdWRlICJ6ZWJyYS9yaWIuaCIKI2luY2x1ZGUgInplYnJhL3pzZXJ2LmgiCiNpbmNsdWRlICJ6ZWJyYS9yZWRpc3RyaWJ1dGUuaCIKI2luY2x1ZGUgInplYnJhL2lyZHAuaCIKI2luY2x1ZGUgPG5ldGluZXQvaXBfaWNtcC5oPgoKI2luY2x1ZGUgImNoZWNrc3VtLmgiCiNpbmNsdWRlICJpZi5oIgojaW5jbHVkZSAic29ja3VuaW9uLmgiCiNpbmNsdWRlICJsb2cuaCIKCi8qIEdMT0JBTCBWQVJTICovCgpleHRlcm4gc3RydWN0IHplYnJhX3ByaXZzX3QgenNlcnZfcHJpdnM7CgovKiBNYXN0ZXIgb2YgdGhyZWFkcy4gKi8KZXh0ZXJuIHN0cnVjdCB6ZWJyYV90IHplYnJhZDsKc3RydWN0IHRocmVhZCAqdF9pcmRwX3JhdzsKCi8qIFRpbWVyIGludGVydmFsIG9mIGlyZHAuICovCmludCBpcmRwX3RpbWVyX2ludGVydmFsID0gSVJEUF9ERUZBVUxUX0lOVEVSVkFMOwoKaW50CmlyZHBfc29ja19pbml0ICh2b2lkKQp7CiAgaW50IHJldCwgaTsKICBpbnQgc2F2ZV9lcnJubzsKICBpbnQgc29jazsKCiAgaWYgKCB6c2Vydl9wcml2cy5jaGFuZ2UgKFpQUklWU19SQUlTRSkgKQogICAgICAgemxvZ19lcnIgKCJpcmRwX3NvY2tfaW5pdDogY291bGQgbm90IHJhaXNlIHByaXZzLCAlcyIsCiAgICAgICAgICAgICAgICAgIHNhZmVfc3RyZXJyb3IgKGVycm5vKSApOwoKICBzb2NrID0gc29ja2V0IChBRl9JTkVULCBTT0NLX1JBVywgSVBQUk9UT19JQ01QKTsKICBzYXZlX2Vycm5vID0gZXJybm87CgogIGlmICggenNlcnZfcHJpdnMuY2hhbmdlIChaUFJJVlNfTE9XRVIpICkKICAgICAgIHpsb2dfZXJyICgiaXJkcF9zb2NrX2luaXQ6IGNvdWxkIG5vdCBsb3dlciBwcml2cywgJXMiLAogICAgICAgICAgICAgc2FmZV9zdHJlcnJvciAoZXJybm8pICk7CgogIGlmIChzb2NrIDwgMCkgewogICAgemxvZ193YXJuICgiSVJEUDogY2FuJ3QgY3JlYXRlIGlyZHAgc29ja2V0ICVzIiwgc2FmZV9zdHJlcnJvcihzYXZlX2Vycm5vKSk7CiAgICByZXR1cm4gc29jazsKICB9OwogIAogIGkgPSAxOwogIHJldCA9IHNldHNvY2tvcHQgKHNvY2ssIElQUFJPVE9fSVAsIElQX1RUTCwgCiAgICAgICAgICAgICAgICAgICAgICAgICh2b2lkICopICZpLCBzaXplb2YgKGkpKTsKICBpZiAocmV0IDwgMCkgewogICAgemxvZ193YXJuICgiSVJEUDogY2FuJ3QgZG8gaXJkcCBzb2Nrb3B0ICVzIiwgc2FmZV9zdHJlcnJvcihlcnJubykpOwogICAgY2xvc2Uoc29jayk7CiAgICByZXR1cm4gcmV0OwogIH07CiAgCiAgcmV0ID0gc2V0c29ja29wdF9pZmluZGV4IChBRl9JTkVULCBzb2NrLCAxKTsKICBpZiAocmV0IDwgMCkgewogICAgemxvZ193YXJuICgiSVJEUDogY2FuJ3QgZG8gaXJkcCBzb2Nrb3B0ICVzIiwgc2FmZV9zdHJlcnJvcihlcnJubykpOwogICAgY2xvc2Uoc29jayk7CiAgICByZXR1cm4gcmV0OwogIH07CgogIHRfaXJkcF9yYXcgPSB0aHJlYWRfYWRkX3JlYWQgKHplYnJhZC5tYXN0ZXIsIGlyZHBfcmVhZF9yYXcsIE5VTEwsIHNvY2spOyAKCiAgcmV0dXJuIHNvY2s7Cn0KCgpzdGF0aWMgaW50CmdldF9wcmVmKHN0cnVjdCBpcmRwX2ludGVyZmFjZSAqaXJkcCwgc3RydWN0IHByZWZpeCAqcCkKewogIHN0cnVjdCBsaXN0bm9kZSAqbm9kZTsKICBzdHJ1Y3QgQWR2ICphZHY7CgogIC8qIFVzZSBkZWZhdWx0IHByZWZlcmVuY2Ugb3IgdXNlIHRoZSBvdmVycmlkZSBwcmVmICovCiAgCiAgaWYoIGlyZHAtPkFkdlByZWZMaXN0ID09IE5VTEwgKQogICAgcmV0dXJuIGlyZHAtPlByZWZlcmVuY2U7CiAgCiAgZm9yIChBTExfTElTVF9FTEVNRU5UU19STyAoaXJkcC0+QWR2UHJlZkxpc3QsIG5vZGUsIGFkdikpCiAgICBpZiggcC0+dS5wcmVmaXg0LnNfYWRkciA9PSBhZHYtPmlwLnNfYWRkciApCiAgICAgIHJldHVybiBhZHYtPnByZWY7CgogIHJldHVybiBpcmRwLT5QcmVmZXJlbmNlOwp9CgovKiBNYWtlIElDTVAgUm91dGVyIEFkdmVydGlzZW1lbnQgTWVzc2FnZS4gKi8Kc3RhdGljIGludAptYWtlX2FkdmVydGlzZW1lbnRfcGFja2V0IChzdHJ1Y3QgaW50ZXJmYWNlICppZnAsIAoJCQkgICBzdHJ1Y3QgcHJlZml4ICpwLAoJCQkgICBzdHJ1Y3Qgc3RyZWFtICpzKQp7CiAgc3RydWN0IHplYnJhX2lmICp6aT1pZnAtPmluZm87CiAgc3RydWN0IGlyZHBfaW50ZXJmYWNlICppcmRwPSZ6aS0+aXJkcDsKICBpbnQgc2l6ZTsKICBpbnQgcHJlZjsKICB1X2ludDE2X3QgY2hlY2tzdW07CgogIHByZWYgPSAgZ2V0X3ByZWYoaXJkcCwgcCk7CgogIHN0cmVhbV9wdXRjIChzLCBJQ01QX1JPVVRFUkFEVkVSVCk7IC8qIFR5cGUuICovCiAgc3RyZWFtX3B1dGMgKHMsIDApOwkJLyogQ29kZS4gKi8KICBzdHJlYW1fcHV0dyAocywgMCk7CQkvKiBDaGVja3N1bS4gKi8KICBzdHJlYW1fcHV0YyAocywgMSk7CQkvKiBOdW0gYWRkcmVzcy4gKi8KICBzdHJlYW1fcHV0YyAocywgMik7CQkvKiBBZGRyZXNzIEVudHJ5IFNpemUuICovCgogIGlmKGlyZHAtPmZsYWdzICYgSUZfU0hVVERPV04pICAKICAgIHN0cmVhbV9wdXR3IChzLCAwKTsKICBlbHNlIAogICAgc3RyZWFtX3B1dHcgKHMsIGlyZHAtPkxpZmV0aW1lKTsKCiAgc3RyZWFtX3B1dGwgKHMsIGh0b25sKHAtPnUucHJlZml4NC5zX2FkZHIpKTsgLyogUm91dGVyIGFkZHJlc3MuICovCiAgc3RyZWFtX3B1dGwgKHMsIHByZWYpOwoKICAvKiBpbl9ja3N1bSByZXR1cm4gbmV0d29yayBieXRlIG9yZGVyIHZhbHVlICovCiAgc2l6ZSA9IDE2OwogIGNoZWNrc3VtID0gaW5fY2tzdW0gKHMtPmRhdGEsIHNpemUpOwogIHN0cmVhbV9wdXR3X2F0IChzLCAyLCBodG9ucyhjaGVja3N1bSkpOwoKICByZXR1cm4gc2l6ZTsKfQoKc3RhdGljIHZvaWQKaXJkcF9zZW5kKHN0cnVjdCBpbnRlcmZhY2UgKmlmcCwgc3RydWN0IHByZWZpeCAqcCwgc3RydWN0IHN0cmVhbSAqcykKewogIHN0cnVjdCB6ZWJyYV9pZiAqemk9aWZwLT5pbmZvOwogIHN0cnVjdCBpcmRwX2ludGVyZmFjZSAqaXJkcD0memktPmlyZHA7CiAgY2hhciBidWZbUFJFRklYX1NUUkxFTl07CiAgdV9pbnQzMl90IGRzdDsKICB1X2ludDMyX3QgdHRsPTE7CgogIGlmICghIChpZnAtPmZsYWdzICYgSUZGX1VQKSkgcmV0dXJuOyAKCiAgaWYgKGlyZHAtPmZsYWdzICYgSUZfQlJPQURDQVNUKSAKICAgIGRzdCA9SU5BRERSX0JST0FEQ0FTVCA7CiAgZWxzZSAKICAgIGRzdCA9IGh0b25sKElOQUREUl9BTExIT1NUU19HUk9VUCk7CgogIGlmKGlyZHAtPmZsYWdzICYgSUZfREVCVUdfTUVTU0FHRVMpIAogICAgemxvZ19kZWJ1ZygiSVJEUDogVFggQWR2ZXJ0IG9uICVzICVzIEhvbGR0aW1lPSVkIFByZWZlcmVuY2U9JWQiLAoJICAgICAgaWZwLT5uYW1lLAoJICAgICAgcHJlZml4MnN0cihwLCBidWYsIHNpemVvZiBidWYpLAoJICAgICAgaXJkcC0+ZmxhZ3MgJiBJRl9TSFVURE9XTj8gMCA6IGlyZHAtPkxpZmV0aW1lLAoJICAgICAgZ2V0X3ByZWYoaXJkcCwgcCkpOwoKICBzZW5kX3BhY2tldCAoaWZwLCBzLCBkc3QsIHAsIHR0bCk7Cn0KCnN0YXRpYyB2b2lkIGlyZHBfYWR2ZXJ0aXNlbWVudCAoc3RydWN0IGludGVyZmFjZSAqaWZwLCBzdHJ1Y3QgcHJlZml4ICpwKQp7CiAgc3RydWN0IHN0cmVhbSAqczsKICBzID0gc3RyZWFtX25ldyAoMTI4KTsKICBtYWtlX2FkdmVydGlzZW1lbnRfcGFja2V0IChpZnAsIHAsIHMpOwogIGlyZHBfc2VuZChpZnAsIHAsIHMpOwogIHN0cmVhbV9mcmVlIChzKTsKfQoKaW50IGlyZHBfc2VuZF90aHJlYWQoc3RydWN0IHRocmVhZCAqdF9hZHZlcnQpCnsKICB1X2ludDMyX3QgdGltZXIsIHRtcDsKICBzdHJ1Y3QgaW50ZXJmYWNlICppZnAgPSBUSFJFQURfQVJHICh0X2FkdmVydCk7CiAgc3RydWN0IHplYnJhX2lmICp6aT1pZnAtPmluZm87CiAgc3RydWN0IGlyZHBfaW50ZXJmYWNlICppcmRwPSZ6aS0+aXJkcDsKICBzdHJ1Y3QgcHJlZml4ICpwOwogIHN0cnVjdCBsaXN0bm9kZSAqbm9kZSwgKm5ub2RlOwogIHN0cnVjdCBjb25uZWN0ZWQgKmlmYzsKCiAgaXJkcC0+ZmxhZ3MgJj0gfklGX1NPTElDSVQ7CgogIGlmKGlmcC0+Y29ubmVjdGVkKSAKICAgIGZvciAoQUxMX0xJU1RfRUxFTUVOVFMgKGlmcC0+Y29ubmVjdGVkLCBub2RlLCBubm9kZSwgaWZjKSkKICAgICAgewogICAgICAgIHAgPSBpZmMtPmFkZHJlc3M7CiAgICAgICAgCiAgICAgICAgaWYgKHAtPmZhbWlseSAhPSBBRl9JTkVUKQogICAgICAgICAgY29udGludWU7CiAgICAgICAgCiAgICAgICAgaXJkcF9hZHZlcnRpc2VtZW50KGlmcCwgcCk7CiAgICAgICAgaXJkcC0+aXJkcF9zZW50Kys7CiAgICAgIH0KCiAgdG1wID0gaXJkcC0+TWF4QWR2ZXJ0SW50ZXJ2YWwtaXJkcC0+TWluQWR2ZXJ0SW50ZXJ2YWw7CiAgdGltZXIgPSAgKHJhbmRvbSAoKSAlIHRtcCApICsgMTsKICB0aW1lciA9IGlyZHAtPk1pbkFkdmVydEludGVydmFsICsgdGltZXI7CgogIGlmKGlyZHAtPmlyZHBfc2VudCA8ICBNQVhfSU5JVElBTF9BRFZFUlRJU0VNRU5UUyAmJgogICAgIHRpbWVyID4gTUFYX0lOSVRJQUxfQURWRVJUX0lOVEVSVkFMICkgCgkgIHRpbWVyPSBNQVhfSU5JVElBTF9BRFZFUlRfSU5URVJWQUw7CgogIGlmKGlyZHAtPmZsYWdzICYgSUZfREVCVUdfTUlTQykKICAgIHpsb2dfZGVidWcoIklSRFA6IE5ldyB0aW1lciBmb3IgJXMgc2V0IHRvICV1XG4iLCBpZnAtPm5hbWUsIHRpbWVyKTsKCiAgaXJkcC0+dF9hZHZlcnRpc2UgPSB0aHJlYWRfYWRkX3RpbWVyKHplYnJhZC5tYXN0ZXIsIGlyZHBfc2VuZF90aHJlYWQsIGlmcCwgdGltZXIpOwogIHJldHVybiAwOwp9Cgp2b2lkIGlyZHBfYWR2ZXJ0X29mZihzdHJ1Y3QgaW50ZXJmYWNlICppZnApCnsKICBzdHJ1Y3QgemVicmFfaWYgKnppPWlmcC0+aW5mbzsKICBzdHJ1Y3QgaXJkcF9pbnRlcmZhY2UgKmlyZHA9JnppLT5pcmRwOwogIHN0cnVjdCBsaXN0bm9kZSAqbm9kZSwgKm5ub2RlOwogIGludCBpOwogIHN0cnVjdCBjb25uZWN0ZWQgKmlmYzsKICBzdHJ1Y3QgcHJlZml4ICpwOwoKICBpZihpcmRwLT50X2FkdmVydGlzZSkgIHRocmVhZF9jYW5jZWwoaXJkcC0+dF9hZHZlcnRpc2UpOwogIGlyZHAtPnRfYWR2ZXJ0aXNlID0gTlVMTDsKICAKICBpZihpZnAtPmNvbm5lY3RlZCkgCiAgICBmb3IgKEFMTF9MSVNUX0VMRU1FTlRTIChpZnAtPmNvbm5lY3RlZCwgbm9kZSwgbm5vZGUsIGlmYykpCiAgICAgIHsKICAgICAgICBwID0gaWZjLT5hZGRyZXNzOwoKICAgICAgICAvKiBPdXRwdXQgc29tZSBwYWNrZXRzIHdpdGggTGlmZXRpbWUgMCAKICAgICAgICAgICB3ZSBzaG91bGQgYWRkIGEgd2FpdC4uLgogICAgICAgICovCgogICAgICAgIGZvcihpPTA7IGk8IElSRFBfTEFTVF9BRFZFUlRfTUVTU0FHRVM7IGkrKykgCiAgICAgICAgICB7CiAgICAgICAgICAgIGlyZHAtPmlyZHBfc2VudCsrOwogICAgICAgICAgICBpcmRwX2FkdmVydGlzZW1lbnQoaWZwLCBwKTsKICAgICAgICAgIH0KICAgICAgfQp9CgoKdm9pZCBwcm9jZXNzX3NvbGljaXQgKHN0cnVjdCBpbnRlcmZhY2UgKmlmcCkKewogIHN0cnVjdCB6ZWJyYV9pZiAqemk9aWZwLT5pbmZvOwogIHN0cnVjdCBpcmRwX2ludGVyZmFjZSAqaXJkcD0memktPmlyZHA7CiAgdV9pbnQzMl90IHRpbWVyOwoKICAvKiBXaGVuIFNPTElDSVQgaXMgYWN0aXZlIHdlIHJlamVjdCBmdXJ0aGVyIGluY29taW5nIHNvbGljaXRzIAogICAgIHRoaXMga2VlcHMgZG93biB0aGUgYW5zd2VyaW5nIHJhdGUgc28gd2UgZG9uJ3QgaGF2ZSB0aGluawogICAgIGFib3V0IERvUyBhdHRhY2tzIGhlcmUuICovCgogIGlmKCBpcmRwLT5mbGFncyAmIElGX1NPTElDSVQpIHJldHVybjsKCiAgaXJkcC0+ZmxhZ3MgfD0gSUZfU09MSUNJVDsKICBpZihpcmRwLT50X2FkdmVydGlzZSkgIHRocmVhZF9jYW5jZWwoaXJkcC0+dF9hZHZlcnRpc2UpOwogIGlyZHAtPnRfYWR2ZXJ0aXNlID0gTlVMTDsKCiAgdGltZXIgPSAgKHJhbmRvbSAoKSAlIE1BWF9SRVNQT05TRV9ERUxBWSkgKyAxOwoKICBpcmRwLT50X2FkdmVydGlzZSA9IHRocmVhZF9hZGRfdGltZXIoemVicmFkLm1hc3RlciwgCgkJCQkgICAgICAgaXJkcF9zZW5kX3RocmVhZCwgCgkJCQkgICAgICAgaWZwLCAKCQkJCSAgICAgICB0aW1lcik7Cn0KCnZvaWQgaXJkcF9maW5pc2goKQp7CgogIHN0cnVjdCBsaXN0bm9kZSAqbm9kZSwgKm5ub2RlOwogIHN0cnVjdCBpbnRlcmZhY2UgKmlmcDsKICBzdHJ1Y3QgemVicmFfaWYgKnppOwogIHN0cnVjdCBpcmRwX2ludGVyZmFjZSAqaXJkcDsKCiAgemxvZ19pbmZvKCJJUkRQOiBSZWNlaXZlZCBzaHV0ZG93biBub3RpZmljYXRpb24uIik7CiAgCiAgZm9yIChBTExfTElTVF9FTEVNRU5UUyAoaWZsaXN0LCBub2RlLCBubm9kZSwgaWZwKSkKICAgIHsKICAgICAgemkgPSBpZnAtPmluZm87CiAgICAgIAogICAgICBpZiAoIXppKSAKICAgICAgICBjb250aW51ZTsKICAgICAgaXJkcCA9ICZ6aS0+aXJkcDsKICAgICAgaWYgKCFpcmRwKSAKICAgICAgICBjb250aW51ZTsKCiAgICAgIGlmIChpcmRwLT5mbGFncyAmIElGX0FDVElWRSApIAogICAgICAgIHsKCSAgaXJkcC0+ZmxhZ3MgfD0gSUZfU0hVVERPV047CgkgIGlyZHBfYWR2ZXJ0X29mZihpZnApOwogICAgICAgIH0KICAgIH0KfQoKI2VuZGlmIC8qIEhBVkVfSVJEUCAqLwo=