LyoKICogQ29weXJpZ2h0IChDKSAyMDA4IFN0ZWZhbiBE9nNpbmdlcihmb3IgQ29kZVdlYXZlcnMpCiAqCiAqIFRoaXMgbGlicmFyeSBpcyBmcmVlIHNvZnR3YXJlOyB5b3UgY2FuIHJlZGlzdHJpYnV0ZSBpdCBhbmQvb3IKICogbW9kaWZ5IGl0IHVuZGVyIHRoZSB0ZXJtcyBvZiB0aGUgR05VIExlc3NlciBHZW5lcmFsIFB1YmxpYwogKiBMaWNlbnNlIGFzIHB1Ymxpc2hlZCBieSB0aGUgRnJlZSBTb2Z0d2FyZSBGb3VuZGF0aW9uOyBlaXRoZXIKICogdmVyc2lvbiAyLjEgb2YgdGhlIExpY2Vuc2UsIG9yIChhdCB5b3VyIG9wdGlvbikgYW55IGxhdGVyIHZlcnNpb24uCiAqCiAqIFRoaXMgbGlicmFyeSBpcyBkaXN0cmlidXRlZCBpbiB0aGUgaG9wZSB0aGF0IGl0IHdpbGwgYmUgdXNlZnVsLAogKiBidXQgV0lUSE9VVCBBTlkgV0FSUkFOVFk7IHdpdGhvdXQgZXZlbiB0aGUgaW1wbGllZCB3YXJyYW50eSBvZgogKiBNRVJDSEFOVEFCSUxJVFkgb3IgRklUTkVTUyBGT1IgQSBQQVJUSUNVTEFSIFBVUlBPU0UuICBTZWUgdGhlIEdOVQogKiBMZXNzZXIgR2VuZXJhbCBQdWJsaWMgTGljZW5zZSBmb3IgbW9yZSBkZXRhaWxzLgogKgogKiBZb3Ugc2hvdWxkIGhhdmUgcmVjZWl2ZWQgYSBjb3B5IG9mIHRoZSBHTlUgTGVzc2VyIEdlbmVyYWwgUHVibGljCiAqIExpY2Vuc2UgYWxvbmcgd2l0aCB0aGlzIGxpYnJhcnk7IGlmIG5vdCwgd3JpdGUgdG8gdGhlIEZyZWUgU29mdHdhcmUKICogRm91bmRhdGlvbiwgSW5jLiwgNTEgRnJhbmtsaW4gU3QsIEZpZnRoIEZsb29yLCBCb3N0b24sIE1BIDAyMTEwLTEzMDEsIFVTQQogKi8KCi8qIFRoaXMgZmlsZSBjb250YWlucyB0ZXN0cyBzcGVjaWZpYyB0byBJRGlyZWN0M0Q5RXggYW5kIElEaXJlY3QzRERldmljZTlFeCwgbGlrZQogKiBob3cgdG8gb2J0YWluIHRoZW0uIEZvciB0ZXN0aW5nIHJlbmRlcmluZyB3aXRoIGV4dGVuZGVkIGZ1bmN0aW9ucyB1c2UgdmlzdWFsLmMKICovCgojZGVmaW5lIENPQkpNQUNST1MKI2luY2x1ZGUgPGluaXRndWlkLmg+CiNpbmNsdWRlIDxkM2Q5Lmg+CiNpbmNsdWRlICJ3aW5lL3Rlc3QuaCIKCnN0YXRpYyBITU9EVUxFIGQzZDlfaGFuZGxlID0gMDsKCnN0YXRpYyBJRGlyZWN0M0Q5ICogKFdJTkFQSSAqcERpcmVjdDNEQ3JlYXRlOSkoVUlOVCBTREtWZXJzaW9uKTsKc3RhdGljIEhSRVNVTFQgKFdJTkFQSSAqcERpcmVjdDNEQ3JlYXRlOUV4KShVSU5UIFNES1ZlcnNpb24sIElEaXJlY3QzRDlFeCAqKmQzZDlleCk7CgpzdGF0aWMgSFdORCBjcmVhdGVfd2luZG93KHZvaWQpCnsKICAgIFdORENMQVNTIHdjID0gezB9OwogICAgSFdORCByZXQ7CiAgICB3Yy5scGZuV25kUHJvYyA9IERlZldpbmRvd1Byb2M7CiAgICB3Yy5scHN6Q2xhc3NOYW1lID0gImQzZDlfdGVzdF93YyI7CiAgICBSZWdpc3RlckNsYXNzKCZ3Yyk7CgogICAgcmV0ID0gQ3JlYXRlV2luZG93KCJkM2Q5X3Rlc3Rfd2MiLCAiZDNkOV90ZXN0IiwKICAgICAgICAgICAgICAgICAgICAgICAgV1NfTUFYSU1JWkUgfCBXU19WSVNJQkxFIHwgV1NfQ0FQVElPTiAsIDAsIDAsIDY0MCwgNDgwLCAwLCAwLCAwLCAwKTsKICAgIHJldHVybiByZXQ7Cn0KCnN0YXRpYyB1bnNpZ25lZCBsb25nIGdldHJlZihJVW5rbm93biAqb2JqKSB7CiAgICBJVW5rbm93bl9BZGRSZWYob2JqKTsKICAgIHJldHVybiBJVW5rbm93bl9SZWxlYXNlKG9iaik7Cn0KCnN0YXRpYyB2b2lkIHRlc3RfcWlfYmFzZV90b19leCh2b2lkKQp7CiAgICBJRGlyZWN0M0Q5ICpkM2Q5ID0gcERpcmVjdDNEQ3JlYXRlOShEM0RfU0RLX1ZFUlNJT04pOwogICAgSURpcmVjdDNEOUV4ICpkM2Q5ZXggPSAodm9pZCAqKSAweGRlYWRiZWVmOwogICAgSURpcmVjdDNERGV2aWNlOSAqZGV2aWNlOwogICAgSURpcmVjdDNERGV2aWNlOUV4ICpkZXZpY2VFeCA9ICh2b2lkICopIDB4ZGVhZGJlZWY7CiAgICBIUkVTVUxUIGhyOwogICAgSFdORCB3aW5kb3cgPSBjcmVhdGVfd2luZG93KCk7CiAgICBEM0RQUkVTRU5UX1BBUkFNRVRFUlMgcHJlc2VudF9wYXJhbWV0ZXJzOwoKICAgIG9rKCBkM2Q5ICE9IE5VTEwsICJGYWlsZWQgdG8gY3JlYXRlIEQzRDkgb2JqZWN0XG4iICk7CiAgICBpZiAoIWQzZDkpIHJldHVybjsKCiAgICBociA9IElEaXJlY3QzRDlfUXVlcnlJbnRlcmZhY2UoZDNkOSwgJklJRF9JRGlyZWN0M0Q5RXgsICh2b2lkICoqKSAmZDNkOWV4KTsKICAgIG9rKGhyID09IEVfTk9JTlRFUkZBQ0UsCiAgICAgICAiSURpcmVjdDNEOTo6UXVlcnlJbnRlcmZhY2UgZm9yIElJRF9JRGlyZWN0M0Q5RXggcmV0dXJuZWQgJTA4eCwgZXhwZWN0ZWQgRV9OT0lOVEVSRkFDRVxuIiwKICAgICAgIGhyKTsKICAgIG9rKGQzZDlleCA9PSBOVUxMLCAiUXVlcnlJbnRlcmZhY2UgcmV0dXJuZWQgaW50ZXJmYWNlICVwLCBleHBlY3RlZCBOVUxMXG4iLCBkM2Q5ZXgpOwogICAgaWYoZDNkOWV4KSBJRGlyZWN0M0Q5RXhfUmVsZWFzZShkM2Q5ZXgpOwoKICAgIG1lbXNldCgmcHJlc2VudF9wYXJhbWV0ZXJzLCAwLCBzaXplb2YocHJlc2VudF9wYXJhbWV0ZXJzKSk7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuV2luZG93ZWQgPSBUUlVFOwogICAgcHJlc2VudF9wYXJhbWV0ZXJzLmhEZXZpY2VXaW5kb3cgPSB3aW5kb3c7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuU3dhcEVmZmVjdCA9IEQzRFNXQVBFRkZFQ1RfQ09QWTsKICAgIHByZXNlbnRfcGFyYW1ldGVycy5CYWNrQnVmZmVyV2lkdGggPSA2NDA7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuQmFja0J1ZmZlckhlaWdodCA9IDQ4MDsKICAgIHByZXNlbnRfcGFyYW1ldGVycy5FbmFibGVBdXRvRGVwdGhTdGVuY2lsID0gRkFMU0U7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuQXV0b0RlcHRoU3RlbmNpbEZvcm1hdCA9IEQzREZNVF9EMTY7CiAgICBociA9IElEaXJlY3QzRDlfQ3JlYXRlRGV2aWNlKGQzZDksIEQzREFEQVBURVJfREVGQVVMVCwgRDNEREVWVFlQRV9IQUwsIHByZXNlbnRfcGFyYW1ldGVycy5oRGV2aWNlV2luZG93LCBEM0RDUkVBVEVfU09GVFdBUkVfVkVSVEVYUFJPQ0VTU0lORywgJnByZXNlbnRfcGFyYW1ldGVycywgJmRldmljZSk7CiAgICBpZihGQUlMRUQoaHIpKSB7CiAgICAgICAgc2tpcCgiRmFpbGVkIHRvIGNyZWF0ZSBhIHJlZ3VsYXIgRGlyZWN0M0REZXZpY2U5LCBza2lwcGluZyBRSSB0ZXN0c1xuIik7CiAgICAgICAgZ290byBvdXQ7CiAgICB9CgogICAgaHIgPSBJRGlyZWN0M0REZXZpY2U5X1F1ZXJ5SW50ZXJmYWNlKGRldmljZSwgJklJRF9JRGlyZWN0M0REZXZpY2U5RXgsICh2b2lkICoqKSAmZGV2aWNlRXgpOwogICAgb2soaHIgPT0gRV9OT0lOVEVSRkFDRSwKICAgICAgICJJRGlyZWN0M0Q5RGV2aWNlOjpRdWVyeUludGVyZmFjZSBmb3IgSUlEX0lEaXJlY3QzRERldmljZTlFeCByZXR1cm5lZCAlMDh4LCBleHBlY3RlZCBFX05PSU5URVJGQUNFXG4iLAogICAgICAgaHIpOwogICAgb2soZGV2aWNlRXggPT0gTlVMTCwgIlF1ZXJ5SW50ZXJmYWNlIHJldHVybmVkIGludGVyZmFjZSAlcCwgZXhwZWN0ZWQgTlVMTFxuIiwgZGV2aWNlRXgpOwogICAgaWYoZGV2aWNlRXgpIElEaXJlY3QzRERldmljZTlFeF9SZWxlYXNlKGRldmljZUV4KTsKCiAgICBJRGlyZWN0M0REZXZpY2U5X1JlbGVhc2UoZGV2aWNlKTsKCm91dDoKICAgIElEaXJlY3QzRDlfUmVsZWFzZShkM2Q5KTsKICAgIERlc3Ryb3lXaW5kb3cod2luZG93KTsKfQoKc3RhdGljIHZvaWQgdGVzdF9xaV9leF90b19iYXNlKHZvaWQpCnsKICAgIElEaXJlY3QzRDkgKmQzZDkgPSAodm9pZCAqKSAweGRlYWRiZWVmOwogICAgSURpcmVjdDNEOUV4ICpkM2Q5ZXg7CiAgICBJRGlyZWN0M0REZXZpY2U5ICpkZXZpY2U7CiAgICBJRGlyZWN0M0REZXZpY2U5RXggKmRldmljZUV4ID0gKHZvaWQgKikgMHhkZWFkYmVlZjsKICAgIEhSRVNVTFQgaHI7CiAgICBIV05EIHdpbmRvdyA9IGNyZWF0ZV93aW5kb3coKTsKICAgIEQzRFBSRVNFTlRfUEFSQU1FVEVSUyBwcmVzZW50X3BhcmFtZXRlcnM7CiAgICB1bnNpZ25lZCBsb25nIHJlZjsKCiAgICBociA9IHBEaXJlY3QzRENyZWF0ZTlFeChEM0RfU0RLX1ZFUlNJT04sICZkM2Q5ZXgpOwogICAgb2soaHIgPT0gRDNEX09LIHx8IGhyID09IEQzREVSUl9OT1RBVkFJTEFCTEUsICJEaXJlY3QzRENyZWF0ZTlFeCByZXR1cm5lZCAlMDh4XG4iLCBocik7CiAgICBpZihGQUlMRUQoaHIpKSB7CiAgICAgICAgc2tpcCgiRGlyZWN0M0Q5RXggaXMgbm90IGF2YWlsYWJsZVxuIik7CiAgICAgICAgZ290byBvdXQ7CiAgICB9CgogICAgaHIgPSBJRGlyZWN0M0Q5RXhfUXVlcnlJbnRlcmZhY2UoZDNkOWV4LCAmSUlEX0lEaXJlY3QzRDksICh2b2lkICoqKSAmZDNkOSk7CiAgICBvayhociA9PSBEM0RfT0ssCiAgICAgICAiSURpcmVjdDNEOUV4OjpRdWVyeUludGVyZmFjZSBmb3IgSUlEX0lEaXJlY3QzRDkgcmV0dXJuZWQgJTA4eCwgZXhwZWN0ZWQgRDNEX09LXG4iLAogICAgICAgaHIpOwogICAgb2soZDNkOSAhPSBOVUxMICYmIGQzZDkgIT0gKHZvaWQgKikgMHhkZWFkYmVlZiwKICAgICAgICJRdWVyeUludGVyZmFjZSByZXR1cm5lZCBpbnRlcmZhY2UgJXAsIGV4cGVjdGVkICE9IE5VTEwgJiYgIT0gMHhkZWFkYmVlZlxuIiwgZDNkOSk7CiAgICByZWYgPSBnZXRyZWYoKElVbmtub3duICopIGQzZDlleCk7CiAgICBvayhyZWYgPT0gMiwgIklEaXJlY3QzRDlFeCByZWZjb3VudCBpcyAlbGQsIGV4cGVjdGVkIDJcbiIsIHJlZik7CiAgICByZWYgPSBnZXRyZWYoKElVbmtub3duICopIGQzZDkpOwogICAgb2socmVmID09IDIsICJJRGlyZWN0M0Q5IHJlZmNvdW50IGlzICVsZCwgZXhwZWN0ZWQgMlxuIiwgcmVmKTsKCiAgICBtZW1zZXQoJnByZXNlbnRfcGFyYW1ldGVycywgMCwgc2l6ZW9mKHByZXNlbnRfcGFyYW1ldGVycykpOwogICAgcHJlc2VudF9wYXJhbWV0ZXJzLldpbmRvd2VkID0gVFJVRTsKICAgIHByZXNlbnRfcGFyYW1ldGVycy5oRGV2aWNlV2luZG93ID0gd2luZG93OwogICAgcHJlc2VudF9wYXJhbWV0ZXJzLlN3YXBFZmZlY3QgPSBEM0RTV0FQRUZGRUNUX0NPUFk7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuQmFja0J1ZmZlcldpZHRoID0gNjQwOwogICAgcHJlc2VudF9wYXJhbWV0ZXJzLkJhY2tCdWZmZXJIZWlnaHQgPSA0ODA7CiAgICBwcmVzZW50X3BhcmFtZXRlcnMuRW5hYmxlQXV0b0RlcHRoU3RlbmNpbCA9IEZBTFNFOwogICAgcHJlc2VudF9wYXJhbWV0ZXJzLkF1dG9EZXB0aFN0ZW5jaWxGb3JtYXQgPSBEM0RGTVRfRDE2OwoKICAgIC8qIEZpcnN0LCB0cnkgdG8gY3JlYXRlIGEgbm9ybWFsIGRldmljZSB3aXRoIElEaXJlY3QzRDlFeDo6Q3JlYXRlRGV2aWNlIGFuZCBRSSBpdCBmb3IgSURpcmVjdDNERGV2aWNlOUV4ICovCiAgICBociA9IElEaXJlY3QzRDlFeF9DcmVhdGVEZXZpY2UoZDNkOWV4LCBEM0RBREFQVEVSX0RFRkFVTFQsIEQzRERFVlRZUEVfSEFMLCBwcmVzZW50X3BhcmFtZXRlcnMuaERldmljZVdpbmRvdywgRDNEQ1JFQVRFX1NPRlRXQVJFX1ZFUlRFWFBST0NFU1NJTkcsICZwcmVzZW50X3BhcmFtZXRlcnMsICZkZXZpY2UpOwogICAgaWYoRkFJTEVEKGhyKSkgewogICAgICAgIHNraXAoIkZhaWxlZCB0byBjcmVhdGUgYSByZWd1bGFyIERpcmVjdDNERGV2aWNlOSwgc2tpcHBpbmcgUUkgdGVzdHNcbiIpOwogICAgICAgIGdvdG8gb3V0OwogICAgfQoKICAgIGhyID0gSURpcmVjdDNERGV2aWNlOV9RdWVyeUludGVyZmFjZShkZXZpY2UsICZJSURfSURpcmVjdDNERGV2aWNlOUV4LCAodm9pZCAqKikgJmRldmljZUV4KTsKICAgIG9rKGhyID09IEQzRF9PSywKICAgICAgICJJRGlyZWN0M0Q5RGV2aWNlOjpRdWVyeUludGVyZmFjZSBmb3IgSUlEX0lEaXJlY3QzRERldmljZTlFeCByZXR1cm5lZCAlMDh4LCBleHBlY3RlZCBEM0RfT0tcbiIsCiAgICAgICBocik7CiAgICBvayhkZXZpY2VFeCAhPSBOVUxMICYmIGRldmljZUV4ICE9ICh2b2lkICopIDB4ZGVhZGJlZWYsCiAgICAgICAiUXVlcnlJbnRlcmZhY2UgcmV0dXJuZWQgaW50ZXJmYWNlICVwLCBleHBlY3RlZCAhPSBOVUxMICYmICE9IDB4ZGVhZGJlZWZcbiIsIGRldmljZUV4KTsKICAgIHJlZiA9IGdldHJlZigoSVVua25vd24gKikgZGV2aWNlKTsKICAgIG9rKHJlZiA9PSAyLCAiSURpcmVjdDNERGV2aWNlOSByZWZjb3VudCBpcyAlbGQsIGV4cGVjdGVkIDJcbiIsIHJlZik7CiAgICByZWYgPSBnZXRyZWYoKElVbmtub3duICopIGRldmljZUV4KTsKICAgIG9rKHJlZiA9PSAyLCAiSURpcmVjdDNERGV2aWNlOUV4IHJlZmNvdW50IGlzICVsZCwgZXhwZWN0ZWQgMlxuIiwgcmVmKTsKICAgIGlmKGRldmljZUV4KSBJRGlyZWN0M0REZXZpY2U5RXhfUmVsZWFzZShkZXZpY2VFeCk7CiAgICBJRGlyZWN0M0REZXZpY2U5X1JlbGVhc2UoZGV2aWNlKTsKCiAgICAvKiBOZXh0LCB0cnkgdG8gY3JlYXRlIGEgbm9ybWFsIGRldmljZSB3aXRoIElEaXJlY3QzRDk6OkNyZWF0ZURldmljZShub24tZXgpIGFuZCBRSSBpdCAqLwogICAgaHIgPSBJRGlyZWN0M0Q5X0NyZWF0ZURldmljZShkM2Q5LCBEM0RBREFQVEVSX0RFRkFVTFQsIEQzRERFVlRZUEVfSEFMLCBwcmVzZW50X3BhcmFtZXRlcnMuaERldmljZVdpbmRvdywgRDNEQ1JFQVRFX1NPRlRXQVJFX1ZFUlRFWFBST0NFU1NJTkcsICZwcmVzZW50X3BhcmFtZXRlcnMsICZkZXZpY2UpOwogICAgaWYoRkFJTEVEKGhyKSkgewogICAgICAgIHNraXAoIkZhaWxlZCB0byBjcmVhdGUgYSByZWd1bGFyIERpcmVjdDNERGV2aWNlOSwgc2tpcHBpbmcgUUkgdGVzdHNcbiIpOwogICAgICAgIGdvdG8gb3V0OwogICAgfQoKICAgIGhyID0gSURpcmVjdDNERGV2aWNlOV9RdWVyeUludGVyZmFjZShkZXZpY2UsICZJSURfSURpcmVjdDNERGV2aWNlOUV4LCAodm9pZCAqKikgJmRldmljZUV4KTsKICAgIG9rKGhyID09IEQzRF9PSywKICAgICAgICJJRGlyZWN0M0Q5RGV2aWNlOjpRdWVyeUludGVyZmFjZSBmb3IgSUlEX0lEaXJlY3QzRERldmljZTlFeCByZXR1cm5lZCAlMDh4LCBleHBlY3RlZCBEM0RfT0tcbiIsCiAgICAgICBocik7CiAgICBvayhkZXZpY2VFeCAhPSBOVUxMICYmIGRldmljZUV4ICE9ICh2b2lkICopIDB4ZGVhZGJlZWYsCiAgICAgICAiUXVlcnlJbnRlcmZhY2UgcmV0dXJuZWQgaW50ZXJmYWNlICVwLCBleHBlY3RlZCAhPSBOVUxMICYmICE9IDB4ZGVhZGJlZWZcbiIsIGRldmljZUV4KTsKICAgIHJlZiA9IGdldHJlZigoSVVua25vd24gKikgZGV2aWNlKTsKICAgIG9rKHJlZiA9PSAyLCAiSURpcmVjdDNERGV2aWNlOSByZWZjb3VudCBpcyAlbGQsIGV4cGVjdGVkIDJcbiIsIHJlZik7CiAgICByZWYgPSBnZXRyZWYoKElVbmtub3duICopIGRldmljZUV4KTsKICAgIG9rKHJlZiA9PSAyLCAiSURpcmVjdDNERGV2aWNlOUV4IHJlZmNvdW50IGlzICVsZCwgZXhwZWN0ZWQgMlxuIiwgcmVmKTsKICAgIGlmKGRldmljZUV4KSBJRGlyZWN0M0REZXZpY2U5RXhfUmVsZWFzZShkZXZpY2VFeCk7CiAgICBJRGlyZWN0M0REZXZpY2U5X1JlbGVhc2UoZGV2aWNlKTsKCiAgICBJRGlyZWN0M0Q5X1JlbGVhc2UoZDNkOSk7CgpvdXQ6CiAgICBEZXN0cm95V2luZG93KHdpbmRvdyk7Cn0KClNUQVJUX1RFU1QoZDNkOWV4KQp7CiAgICBkM2Q5X2hhbmRsZSA9IExvYWRMaWJyYXJ5QSgiZDNkOS5kbGwiKTsKICAgIGlmICghZDNkOV9oYW5kbGUpCiAgICB7CiAgICAgICAgc2tpcCgiQ291bGQgbm90IGxvYWQgZDNkOS5kbGxcbiIpOwogICAgICAgIHJldHVybjsKICAgIH0KICAgIHBEaXJlY3QzRENyZWF0ZTkgPSAodm9pZCAqKUdldFByb2NBZGRyZXNzKGQzZDlfaGFuZGxlLCAiRGlyZWN0M0RDcmVhdGU5Iik7CiAgICBvayhwRGlyZWN0M0RDcmVhdGU5ICE9IE5VTEwsICJGYWlsZWQgdG8gZ2V0IGFkZHJlc3Mgb2YgRGlyZWN0M0RDcmVhdGU5XG4iKTsKICAgIGlmKCFwRGlyZWN0M0RDcmVhdGU5KSB7CiAgICAgICAgcmV0dXJuOwogICAgfQoKICAgIHBEaXJlY3QzRENyZWF0ZTlFeCA9ICh2b2lkICopR2V0UHJvY0FkZHJlc3MoZDNkOV9oYW5kbGUsICJEaXJlY3QzRENyZWF0ZTlFeCIpOwogICAgaWYgKCFwRGlyZWN0M0RDcmVhdGU5RXgpIHsKICAgICAgICBza2lwKCJGYWlsZWQgdG8gZ2V0IGFkZHJlc3Mgb2YgRGlyZWN0M0RDcmVhdGU5RXhcbiIpOwogICAgICAgIHJldHVybjsKICAgIH0KCiAgICB0ZXN0X3FpX2Jhc2VfdG9fZXgoKTsKICAgIHRlc3RfcWlfZXhfdG9fYmFzZSgpOwp9Cg==